| /external/tensorflow/tensorflow/core/framework/ |
| D | type_traits.h | 47 struct is_quantized<qint16> : true_type {};
86 class numeric_limits<tensorflow::qint16>
101 struct is_signed<tensorflow::qint16> : public is_signed<tensorflow::int16> {};
|
| D | register_types.h | 83 #define TF_CALL_qint16(m) m(::tensorflow::qint16)
115 #define TF_CALL_qint16(m) m(::tensorflow::qint16)
|
| D | numeric_types.h | 40 typedef Eigen::QInt16 qint16; typedef
|
| D | types.cc | 264 DEFINE_DATATYPETOENUM_VALUE(qint16);
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/tests/ |
| D | legalize-tf-types.mlir | 32 func @id_qint16(%arg0: tensor<1x!tf.qint16>) -> tensor<1x!tf.qint16> {
35 return %arg0: tensor<1x!tf.qint16>
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils_test.cc | 649 TestSetTensorValue<qint16>(DT_QINT16, -8, /*success=*/true, /*error_msg=*/""); in TEST()
650 TestSetTensorValue<qint16>(DT_QINT16, 0, /*success=*/true, /*error_msg=*/""); in TEST()
651 TestSetTensorValue<qint16>(DT_QINT16, 8, /*success=*/true, /*error_msg=*/""); in TEST()
652 TestSetTensorValue<qint16>(DT_QINT16, std::numeric_limits<qint16>::min(), in TEST()
654 TestSetTensorValue<qint16>(DT_QINT16, std::numeric_limits<qint16>::max(), in TEST()
656 TestSetTensorValue<qint16>(DT_QINT16, kMinInt, /*success=*/false, in TEST()
658 TestSetTensorValue<qint16>(DT_QINT16, kMaxInt, /*success=*/false, in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | dequantize_op.cc | 247 .TypeConstraint<qint16>("T")
249 DequantizeOp<CPUDevice, qint16, float>);
273 .TypeConstraint<qint16>("T")
275 DequantizeOp<CPUDevice, qint16, bfloat16>);
|
| D | fill_functor.cc | 59 DEFINE_SETZERO_CPU(qint16);
111 DEFINE_FILL_CPU(qint16);
|
| D | dequantize_op_test.cc | 196 RunDequantizeMinCombinedTest<qint16>(0, 255.0f, "Dequantize"); in TEST_F()
209 RunDequantizeBfloat16MinCombinedTest<qint16>(0, 255.0f); in TEST_F()
276 BM_DequantizeMinCombinedCpu<qint16>(state); in BM_DequantizeMinCombinedCpuQint16()
322 BM_DequantizeBfloat16MinCombinedCpu<qint16>(state); in BM_DequantizeBfloat16MinCombinedCpuQint16()
|
| D | cwise_op_not_equal_to_1.cc | 22 uint64, qint8, qint16, quint8, quint16);
|
| D | cwise_op_equal_to_1.cc | 22 qint8, qint16, quint8, quint16);
|
| D | concat_lib_cpu.cc | 117 REGISTER(qint16)
|
| D | quantize_op.cc | 315 Name("QuantizeV2").Device(DEVICE_CPU).TypeConstraint<qint16>("T"),
316 QuantizeV2Op<CPUDevice, qint16>);
|
| D | cast_op_test.cc | 99 TEST_CAST(in, qint16); \
116 TEST_ALL_CASTS_FROM(qint16) in TEST_ALL_CASTS_FROM()
|
| D | constant_op.cc | 114 REGISTER_KERNEL(GPU, qint16);
189 REGISTER_KERNEL(CPU, qint16);
|
| /external/tensorflow/tensorflow/python/framework/ |
| D | constant_op_test.py | 44 dtypes.qint16,
|
| D | dtypes.py | 281 qint16 = DType(types_pb2.DT_QINT16) variable
335 types_pb2.DT_QINT16: qint16,
464 _np_qint16: qint16,
580 _QUANTIZED_DTYPES_NO_REF = frozenset([qint8, quint8, qint16, quint16, qint32])
|
| D | tensor_util.py | 116 dtypes.qint16.as_numpy_dtype:
180 dtypes.qint16.as_numpy_dtype: SlowAppendQIntArrayToTensorProto,
242 dtypes.qint16, dtypes.quint16, dtypes.qint32, dtypes.uint32, dtypes.uint64
297 dtypes.qint16: _check_quantized,
430 dtypes.qint8, dtypes.quint8, dtypes.qint16, dtypes.quint16,
626 dtypes.qint32, dtypes.quint8, dtypes.qint8, dtypes.qint16, dtypes.quint16
|
| /external/tensorflow/tensorflow/python/data/experimental/kernel_tests/ |
| D | unique_test.py | 90 dtypes.float64, dtypes.qint16, dtypes.qint32
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | literal_util_test.cc | 49 std::pair<int16, qint16>, std::pair<uint16, quint16>,
|
| /external/tensorflow/tensorflow/security/fuzzing/ |
| D | python_fuzzing.py | 32 tf.int32, tf.int64, tf.bool, tf.string, tf.qint8, tf.quint8, tf.qint16,
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_tensor_coding_test.cc | 94 DoTest<qint16>(DT_QINT16); in TEST_F()
|
| /external/tensorflow/tensorflow/python/dlpack/ |
| D | dlpack_test.py | 98 tf_tensor = constant_op.constant([[1, 4], [5, 2]], dtype=dtypes.qint16)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.dtypes.pbtxt | 64 name: "qint16"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_types.def | 23 HANDLE_TF_TYPE(Qint16, QINT16, "qint16")
|