Home
last modified time | relevance | path

Searched refs:quint16 (Results 1 – 25 of 47) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Dtype_traits.h49 struct is_quantized<quint16> : true_type {};
89 class numeric_limits<tensorflow::quint16>
103 struct is_signed<tensorflow::quint16> : public is_signed<tensorflow::uint16> {};
Dregister_types.h85 #define TF_CALL_quint16(m) m(::tensorflow::quint16)
117 #define TF_CALL_quint16(m) m(::tensorflow::quint16)
Dnumeric_types.h41 typedef Eigen::QUInt16 quint16; typedef
Dtypes.cc265 DEFINE_DATATYPETOENUM_VALUE(quint16);
/external/tensorflow/tensorflow/compiler/mlir/xla/tests/
Dlegalize-tf-types.mlir50 func @id_quint16(%arg0: tensor<1x!tf.quint16>) -> tensor<1x!tf.quint16> {
53 return %arg0: tensor<1x!tf.quint16>
/external/tensorflow/tensorflow/core/grappler/
Dutils_test.cc661 TestSetTensorValue<quint16>( in TEST()
664 TestSetTensorValue<quint16>(DT_QUINT16, 0, /*success=*/true, in TEST()
666 TestSetTensorValue<quint16>(DT_QUINT16, 8, /*success=*/true, in TEST()
668 TestSetTensorValue<quint16>(DT_QUINT16, std::numeric_limits<quint16>::min(), in TEST()
670 TestSetTensorValue<quint16>(DT_QUINT16, std::numeric_limits<quint16>::max(), in TEST()
672 TestSetTensorValue<quint16>(DT_QUINT16, kMinInt, /*success=*/false, in TEST()
674 TestSetTensorValue<quint16>(DT_QUINT16, kMaxInt, /*success=*/false, in TEST()
/external/tensorflow/tensorflow/core/kernels/
Ddequantize_op.cc242 .TypeConstraint<quint16>("T")
244 DequantizeOp<CPUDevice, quint16, float>);
268 .TypeConstraint<quint16>("T")
270 DequantizeOp<CPUDevice, quint16, bfloat16>);
Dfill_functor.cc58 DEFINE_SETZERO_CPU(quint16);
109 DEFINE_FILL_CPU(quint16);
Ddequantize_op_test.cc199 RunDequantizeMinCombinedTest<quint16>(0, 255.0f, "Dequantize"); in TEST_F()
212 RunDequantizeBfloat16MinCombinedTest<quint16>(0, 255.0f); in TEST_F()
272 BM_DequantizeMinCombinedCpu<quint16>(state); in BM_DequantizeMinCombinedCpuQuint16()
317 BM_DequantizeBfloat16MinCombinedCpu<quint16>(state); in BM_DequantizeBfloat16MinCombinedCpuQuint16()
Dcwise_op_not_equal_to_1.cc22 uint64, qint8, qint16, quint8, quint16);
Dcwise_op_equal_to_1.cc22 qint8, qint16, quint8, quint16);
Dconcat_lib_cpu.cc116 REGISTER(quint16)
Dcast_op_test.cc100 TEST_CAST(in, quint16);
117 TEST_ALL_CASTS_FROM(quint16) in TEST_ALL_CASTS_FROM()
Dquantize_op.cc312 Name("QuantizeV2").Device(DEVICE_CPU).TypeConstraint<quint16>("T"),
313 QuantizeV2Op<CPUDevice, quint16>);
Dconstant_op.cc115 REGISTER_KERNEL(GPU, quint16);
187 REGISTER_KERNEL(CPU, quint16);
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_QuantizeDownAndShrinkRange.pbtxt46 input_max is 1.0f, and we are dealing with quint16 quantized data, then a 0
51 data. For example, maybe that quint16 input has no values lower than 16,384 and
Dapi_def_Requantize.pbtxt59 `input_max` is 1.0f, and we are dealing with `quint16` quantized data, then a 0
/external/tensorflow/tensorflow/python/framework/
Dconstant_op_test.py47 dtypes.quint16,
Ddtypes.py283 quint16 = DType(types_pb2.DT_QUINT16) variable
336 types_pb2.DT_QUINT16: quint16,
465 _np_quint16: quint16,
580 _QUANTIZED_DTYPES_NO_REF = frozenset([qint8, quint8, qint16, quint16, qint32])
Dtensor_util.py118 dtypes.quint16.as_numpy_dtype:
181 dtypes.quint16.as_numpy_dtype: SlowAppendQIntArrayToTensorProto,
242 dtypes.qint16, dtypes.quint16, dtypes.qint32, dtypes.uint32, dtypes.uint64
300 dtypes.quint16: _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/compiler/tf2xla/
Dliteral_util_test.cc49 std::pair<int16, qint16>, std::pair<uint16, quint16>,
/external/tensorflow/tensorflow/security/fuzzing/
Dpython_fuzzing.py33 tf.quint16, tf.qint32, tf.resource, tf.variant
/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_tensor_coding_test.cc95 DoTest<quint16>(DT_QUINT16); in TEST_F()
/external/tensorflow/tensorflow/python/kernel_tests/
Dbitcast_op_test.py89 datatype = dtypes.quint16
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.dtypes.pbtxt76 name: "quint16"

12