Home
last modified time | relevance | path

Searched refs:tensor_dtype (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dpad_test.cc346 template <typename integer_type, TensorType tensor_dtype>
351 PadOpConstModel m({tensor_dtype, {1, 2, 2, 1}, 1.0, 2.0}, {4, 2}, in ZeroNotInQuantizationRange()
352 {0, 0, 1, 1, 1, 1, 0, 0}, {tensor_dtype, {}, 1.0, 2.0}), in ZeroNotInQuantizationRange()
367 template <typename integer_type, TensorType tensor_dtype>
373 const float kMax = tensor_dtype == TensorType_INT16 ? 32767.f / 32768.f : 1.f; in SimpleConstTest()
375 PadOpConstModel m({tensor_dtype, {1, 2, 2, 1}, kMin, kMax}, {4, 2}, in SimpleConstTest()
376 {0, 0, 1, 1, 1, 1, 0, 0}, {tensor_dtype, {}, kMin, kMax}); in SimpleConstTest()
396 template <typename integer_type, TensorType tensor_dtype>
399 const float kMax = tensor_dtype == TensorType_INT16 ? 32767.f / 32768.f : 1.f; in SimpleDynamicTest()
401 PadOpDynamicModel m({tensor_dtype, {1, 2, 2, 1}, kMin, kMax}, {4, 2}, in SimpleDynamicTest()
[all …]
Dreduce_test.cc431 template <typename integer_type, TensorType tensor_dtype>
435 MeanOpConstModel m({tensor_dtype, {1, 1, 2, 4}, -255.0, 255.0}, in MeanOpConstModelTest()
436 {tensor_dtype, {1, 2, 4}, -255, 255.0}, {1}, {1}, false); in MeanOpConstModelTest()
454 template <typename integer_type, TensorType tensor_dtype>
458 MeanOpConstModel m({tensor_dtype, {1, 1, 2, 4}, -1.0, 1.0}, in ConstMeanOpTestNonSameScale()
459 {tensor_dtype, {1, 2}, -5.0, 5.0}, {2}, {1, 3}, false); in ConstMeanOpTestNonSameScale()
478 template <typename integer_type, TensorType tensor_dtype>
485 MeanOpConstModel m({tensor_dtype, {1, 2, 2, 9}, -1.0, 1.0}, in MeanOpTestQuantizedSameScale()
486 {tensor_dtype, {2}, -1.0, 1.0}, {2}, {1, 2}, true); in MeanOpTestQuantizedSameScale()
506 template <typename integer_type, TensorType tensor_dtype>
[all …]
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Darg_min_max_test.cc77 template <typename integer_type, TensorType tensor_dtype>
79 ArgMinOpModel model({1, 1, 1, 4}, tensor_dtype); in ArgMinTestImpl()
83 if (tensor_dtype == TensorType_UINT8) { in ArgMinTestImpl()
93 template <typename integer_type, TensorType tensor_dtype>
95 ArgMinOpModel model({1, 1, 2, 4}, tensor_dtype); in ArgMinNegativeTestImpl()
99 if (tensor_dtype == TensorType_UINT8) { in ArgMinNegativeTestImpl()
110 template <typename integer_type, TensorType tensor_dtype>
112 ArgMaxOpModel model({1, 1, 1, 4}, tensor_dtype); in ArgMaxTestImpl()
116 if (tensor_dtype == TensorType_UINT8) { in ArgMaxTestImpl()
Dactivations_test.cc81 template <typename integer_type, TensorType tensor_dtype>
86 /*input=*/{tensor_dtype, {1, 3}, kMin, kMax}, in ReluTestImpl()
87 /*output=*/{tensor_dtype, {1, 3}, kMin, kMax}); in ReluTestImpl()
97 template <typename integer_type, TensorType tensor_dtype>
102 /*input=*/{tensor_dtype, {1, 3}, kMin, kMax}, in Relu6TestImpl()
103 /*output=*/{tensor_dtype, {1, 3}, kMin, kMax}); in Relu6TestImpl()
113 template <typename integer_type, TensorType tensor_dtype>
119 /*input=*/{tensor_dtype, {1, 3}, 8 * kMin, 8 * kMax}, in TanhTestImpl()
120 /*output=*/{tensor_dtype, {1, 3}, kMin, kMax}); in TanhTestImpl()
130 template <typename integer_type, TensorType tensor_dtype>
[all …]
Dsplit_test.cc57 template <typename integer_type, TensorType tensor_dtype>
72 SplitOpModel const_m({tensor_dtype, input_shape, kMin, kMax}, in CheckSplitBehavior()
73 {tensor_dtype, output_shape, kMin, kMax}, num_splits, in CheckSplitBehavior()
87 template <typename integer_type, TensorType tensor_dtype>
89 CheckSplitBehavior<integer_type, tensor_dtype>( in CheckFourDimSplitImpl()
96 CheckSplitBehavior<integer_type, tensor_dtype>( in CheckFourDimSplitImpl()
103 CheckSplitBehavior<integer_type, tensor_dtype>( in CheckFourDimSplitImpl()
110 CheckSplitBehavior<integer_type, tensor_dtype>( in CheckFourDimSplitImpl()
Dconcat_test.cc85 template <typename integer_type, TensorType tensor_dtype>
87 QuantizedConcatenationOpModel m0({{tensor_dtype, {2, 1, 1, 2}, -12.7, 12.8}, in FourInputsQuantizedSameRangeImpl()
88 {tensor_dtype, {2, 1, 1, 2}, -12.7, 12.8}, in FourInputsQuantizedSameRangeImpl()
89 {tensor_dtype, {2, 1, 1, 2}, -12.7, 12.8}, in FourInputsQuantizedSameRangeImpl()
90 {tensor_dtype, {2, 1, 1, 2}, -12.7, 12.8}}, in FourInputsQuantizedSameRangeImpl()
91 /*axis=*/3, {tensor_dtype, {}, -12.7, 12.8}); in FourInputsQuantizedSameRangeImpl()
115 template <typename integer_type, TensorType tensor_dtype>
119 QuantizedConcatenationOpModel m0({{tensor_dtype, in TwoInputsNegativeAxisImpl()
123 {tensor_dtype, in TwoInputsNegativeAxisImpl()
128 {tensor_dtype, in TwoInputsNegativeAxisImpl()
Dpad_test.cc61 template <typename integer_type, TensorType tensor_dtype>
66 PadOpConstModel m({tensor_dtype, {1, 2, 2, 1}, -1.0, 1.0}, {4, 2}, in SimpleConstTestImpl()
67 {0, 0, 1, 1, 1, 1, 0, 0}, {tensor_dtype, {}, -1.0, 1.0}); in SimpleConstTestImpl()
77 template <typename integer_type, TensorType tensor_dtype>
80 PadOpConstModel m({tensor_dtype, {1, 2, 3, 1}, -1.0, 1.0}, {4, 2}, in AdvancedConstTestImpl()
81 {0, 0, 0, 2, 1, 3, 0, 0}, {tensor_dtype, {}, -1.0, 1.0}); in AdvancedConstTestImpl()
/external/tensorflow/tensorflow/python/framework/
Dtensor_util.py594 tensor_dtype = dtypes.as_dtype(tensor.dtype)
595 dtype = tensor_dtype.as_numpy_dtype
601 if tensor_dtype == dtypes.string:
610 if tensor_dtype == dtypes.float16 or tensor_dtype == dtypes.bfloat16:
614 values.dtype = tensor_dtype.as_numpy_dtype
615 elif tensor_dtype == dtypes.float32:
617 elif tensor_dtype == dtypes.float64:
619 elif tensor_dtype in [
624 elif tensor_dtype == dtypes.int64:
626 elif tensor_dtype == dtypes.uint32:
[all …]
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc5237 template <typename integer_type, TensorType tensor_dtype>
5242 {tensor_dtype, {1, 2, 2, 1}, -1.0, 1.0}, {4, 2}, {0, 0, 1, 1, 1, 1, 0, 0}, in SimpleConstTestV2()
5243 {tensor_dtype, {1}, -1.0, 1.0}, {tensor_dtype, {}, -1.0, 1.0}); in SimpleConstTestV2()
5261 template <typename integer_type, TensorType tensor_dtype>
5263 PadV2OpDynamicModel<integer_type> m({tensor_dtype, {1, 2, 2, 1}, -1.0, 1.0}, in SimpleDynamicTestV2()
5264 {4, 2}, {tensor_dtype, {1}, -1.0, 1.0}, in SimpleDynamicTestV2()
5265 {tensor_dtype, {}, -1.0, 1.0}); in SimpleDynamicTestV2()
5284 template <typename integer_type, TensorType tensor_dtype>
5287 {tensor_dtype, {1, 2, 3, 1}, -1.0, 1.0}, {4, 2}, {0, 0, 0, 2, 1, 3, 0, 0}, in AdvancedConstTestV2()
5288 {tensor_dtype, {1}, -1.0, 1.0}, {tensor_dtype, {}, -1.0, 1.0}); in AdvancedConstTestV2()
[all …]
/external/tensorflow/tensorflow/core/debug/
Ddebug_io_utils.cc64 const DataType& tensor_dtype, in PrepareChunkEventProto() argument
105 value->mutable_tensor()->set_dtype(tensor_dtype); in PrepareChunkEventProto()
/external/tensorflow/tensorflow/python/ops/
Darray_ops.py3128 tensor_dtype = tensor.dtype
3131 if dtype is not None and dtype != tensor_dtype:
3140 tensor_dtype != dtypes.variant):
3143 return zeros(tensor_shape, dtype=dtype or tensor_dtype, name=name)
3145 if dtype is not None and dtype != tensor_dtype and dtype != dtypes.variant:
Dmath_ops.py4071 def accumulate_n(inputs, shape=None, tensor_dtype=None, name=None): argument
4141 if tensor_dtype is not None and tensor_dtype != inputs[0].dtype:
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.math.pbtxt13 …argspec: "args=[\'inputs\', \'shape\', \'tensor_dtype\', \'name\'], varargs=None, keywords=None, d…
Dtensorflow.pbtxt733 …argspec: "args=[\'inputs\', \'shape\', \'tensor_dtype\', \'name\'], varargs=None, keywords=None, d…
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.math.pbtxt13 …argspec: "args=[\'inputs\', \'shape\', \'tensor_dtype\', \'name\'], varargs=None, keywords=None, d…