Home
last modified time | relevance | path

Searched refs:quint16 (Results 1 – 25 of 42) 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.h42 typedef Eigen::QUInt16 quint16; typedef
Dtypes.cc273 DEFINE_DATATYPETOENUM_VALUE(quint16);
/external/tensorflow/tensorflow/core/grappler/
Dutils_test.cc588 TestSetTensorValue<quint16>( in TEST()
591 TestSetTensorValue<quint16>(DT_QUINT16, 0, /*success=*/true, in TEST()
593 TestSetTensorValue<quint16>(DT_QUINT16, 8, /*success=*/true, in TEST()
595 TestSetTensorValue<quint16>(DT_QUINT16, std::numeric_limits<quint16>::min(), in TEST()
597 TestSetTensorValue<quint16>(DT_QUINT16, std::numeric_limits<quint16>::max(), in TEST()
599 TestSetTensorValue<quint16>(DT_QUINT16, kMinInt, /*success=*/false, in TEST()
601 TestSetTensorValue<quint16>(DT_QUINT16, kMaxInt, /*success=*/false, in TEST()
/external/tensorflow/tensorflow/core/kernels/
Ddequantize_op.cc236 .TypeConstraint<quint16>("T")
238 DequantizeOp<CPUDevice, quint16, float>);
262 .TypeConstraint<quint16>("T")
264 DequantizeOp<CPUDevice, quint16, bfloat16>);
Ddequantize_op_test.cc199 RunDequantizeMinCombinedTest<quint16>(0, 255.0f, "Dequantize"); in TEST_F()
212 RunDequantizeBfloat16MinCombinedTest<quint16>(0, 255.0f); in TEST_F()
271 BM_DequantizeMinCombinedCpu<quint16>(iters); in BM_DequantizeMinCombinedCpuQuint16()
313 BM_DequantizeBfloat16MinCombinedCpu<quint16>(iters); in BM_DequantizeBfloat16MinCombinedCpuQuint16()
Dfill_functor.cc56 DEFINE_SETZERO_CPU(quint16);
139 DEFINE_FILL_CPU(quint16);
Dconstant_op.cc115 REGISTER_KERNEL(GPU, quint16);
140 REGISTER_SYCL_KERNEL(SYCL, quint16);
211 REGISTER_KERNEL(CPU, quint16);
Dcast_op_test.cc98 TEST_CAST(in, quint16);
115 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>);
Dconcat_lib_cpu.cc116 REGISTER(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/
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])
Dconstant_op_test.py47 dtypes.quint16,
Dtensor_util.py116 dtypes.quint16.as_numpy_dtype:
179 dtypes.quint16.as_numpy_dtype: SlowAppendQIntArrayToTensorProto,
240 dtypes.quint16, dtypes.qint32, dtypes.uint32, dtypes.uint64
294 dtypes.quint16: _check_quantized,
424 dtypes.qint8, dtypes.quint8, dtypes.qint16, dtypes.quint16,
620 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/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.py85 datatype = dtypes.quint16
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.dtypes.pbtxt76 name: "quint16"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.dtypes.pbtxt76 name: "quint16"
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_types.def30 HANDLE_TF_TYPE(Quint16, QUINT16, "quint16")
/external/tensorflow/tensorflow/python/ops/
Ddequantize_op_test.py46 dtypes.quint16: np.uint16,
/external/tensorflow/tensorflow/core/distributed_runtime/
Dtensor_coding_test.cc151 DoTest<quint16>(DT_QUINT16); in TEST_F()

12