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.cc375 template <typename integer_type, TensorType tensor_dtype>
379 MeanOpConstModel m({tensor_dtype, {1, 1, 2, 4}, -255.0, 255.0}, in MeanOpConstModelTest()
380 {tensor_dtype, {1, 2, 4}, -255, 255.0}, {1}, {1}, false); in MeanOpConstModelTest()
398 template <typename integer_type, TensorType tensor_dtype>
402 MeanOpConstModel m({tensor_dtype, {1, 1, 2, 4}, -1.0, 1.0}, in ConstMeanOpTestNonSameScale()
403 {tensor_dtype, {1, 2}, -5.0, 5.0}, {2}, {1, 3}, false); in ConstMeanOpTestNonSameScale()
422 template <typename integer_type, TensorType tensor_dtype>
429 MeanOpConstModel m({tensor_dtype, {1, 2, 2, 9}, -1.0, 1.0}, in MeanOpTestQuantizedSameScale()
430 {tensor_dtype, {2}, -1.0, 1.0}, {2}, {1, 2}, true); in MeanOpTestQuantizedSameScale()
450 template <typename integer_type, TensorType tensor_dtype>
[all …]
/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/tests/
Darg_min_max_test.cc75 template <typename integer_type, TensorType tensor_dtype>
77 ArgMinOpModel model({1, 1, 1, 4}, tensor_dtype); in ArgMinTestImpl()
81 if (tensor_dtype == TensorType_UINT8) { in ArgMinTestImpl()
91 template <typename integer_type, TensorType tensor_dtype>
93 ArgMinOpModel model({1, 1, 2, 4}, tensor_dtype); in ArgMinNegativeTestImpl()
97 if (tensor_dtype == TensorType_UINT8) { in ArgMinNegativeTestImpl()
108 template <typename integer_type, TensorType tensor_dtype>
110 ArgMaxOpModel model({1, 1, 1, 4}, tensor_dtype); in ArgMaxTestImpl()
114 if (tensor_dtype == TensorType_UINT8) { in ArgMaxTestImpl()
Dactivations_test.cc80 template <typename integer_type, TensorType tensor_dtype>
85 /*input=*/{tensor_dtype, {1, 3}, kMin, kMax}, in ReluTestImpl()
86 /*output=*/{tensor_dtype, {1, 3}, kMin, kMax}); in ReluTestImpl()
96 template <typename integer_type, TensorType tensor_dtype>
101 /*input=*/{tensor_dtype, {1, 3}, kMin, kMax}, in Relu6TestImpl()
102 /*output=*/{tensor_dtype, {1, 3}, kMin, kMax}); in Relu6TestImpl()
112 template <typename integer_type, TensorType tensor_dtype>
118 /*input=*/{tensor_dtype, {1, 3}, 8 * kMin, 8 * kMax}, in TanhTestImpl()
119 /*output=*/{tensor_dtype, {1, 3}, kMin, kMax}); in TanhTestImpl()
129 template <typename integer_type, TensorType tensor_dtype>
[all …]
Dsplit_test.cc54 template <typename integer_type, TensorType tensor_dtype>
69 SplitOpModel const_m({tensor_dtype, input_shape, kMin, kMax}, in CheckSplitBehavior()
70 {tensor_dtype, output_shape, kMin, kMax}, num_splits, in CheckSplitBehavior()
84 template <typename integer_type, TensorType tensor_dtype>
86 CheckSplitBehavior<integer_type, tensor_dtype>( in CheckFourDimSplitImpl()
93 CheckSplitBehavior<integer_type, tensor_dtype>( in CheckFourDimSplitImpl()
100 CheckSplitBehavior<integer_type, tensor_dtype>( in CheckFourDimSplitImpl()
107 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.cc59 template <typename integer_type, TensorType tensor_dtype>
64 PadOpConstModel m({tensor_dtype, {1, 2, 2, 1}, -1.0, 1.0}, {4, 2}, in SimpleConstTestImpl()
65 {0, 0, 1, 1, 1, 1, 0, 0}, {tensor_dtype, {}, -1.0, 1.0}); in SimpleConstTestImpl()
75 template <typename integer_type, TensorType tensor_dtype>
78 PadOpConstModel m({tensor_dtype, {1, 2, 3, 1}, -1.0, 1.0}, {4, 2}, in AdvancedConstTestImpl()
79 {0, 0, 0, 2, 1, 3, 0, 0}, {tensor_dtype, {}, -1.0, 1.0}); in AdvancedConstTestImpl()
/external/tensorflow/tensorflow/python/framework/
Dtensor_util.py599 tensor_dtype = dtypes.as_dtype(tensor.dtype)
600 dtype = tensor_dtype.as_numpy_dtype
606 if tensor_dtype == dtypes.string:
615 if tensor_dtype == dtypes.float16 or tensor_dtype == dtypes.bfloat16:
619 values.dtype = tensor_dtype.as_numpy_dtype
620 elif tensor_dtype == dtypes.float32:
622 elif tensor_dtype == dtypes.float64:
624 elif tensor_dtype in [
629 elif tensor_dtype == dtypes.int64:
631 elif tensor_dtype == dtypes.uint32:
[all …]
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate_test.cc5160 template <typename integer_type, TensorType tensor_dtype>
5165 {tensor_dtype, {1, 2, 2, 1}, -1.0, 1.0}, {4, 2}, {0, 0, 1, 1, 1, 1, 0, 0}, in SimpleConstTestV2()
5166 {tensor_dtype, {1}, -1.0, 1.0}, {tensor_dtype, {}, -1.0, 1.0}); in SimpleConstTestV2()
5184 template <typename integer_type, TensorType tensor_dtype>
5186 PadV2OpDynamicModel<integer_type> m({tensor_dtype, {1, 2, 2, 1}, -1.0, 1.0}, in SimpleDynamicTestV2()
5187 {4, 2}, {tensor_dtype, {1}, -1.0, 1.0}, in SimpleDynamicTestV2()
5188 {tensor_dtype, {}, -1.0, 1.0}); in SimpleDynamicTestV2()
5207 template <typename integer_type, TensorType tensor_dtype>
5210 {tensor_dtype, {1, 2, 3, 1}, -1.0, 1.0}, {4, 2}, {0, 0, 0, 2, 1, 3, 0, 0}, in AdvancedConstTestV2()
5211 {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.py3072 tensor_dtype = tensor.dtype
3075 if dtype is not None and dtype != tensor_dtype:
3084 tensor_dtype != dtypes.variant):
3087 return zeros(tensor_shape, dtype=dtype or tensor_dtype, name=name)
3089 if dtype is not None and dtype != tensor_dtype and dtype != dtypes.variant:
Dmath_ops.py4028 def accumulate_n(inputs, shape=None, tensor_dtype=None, name=None): argument
4086 if tensor_dtype is not None and tensor_dtype != inputs[0].dtype:
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.math.pbtxt13 …argspec: "args=[\'inputs\', \'shape\', \'tensor_dtype\', \'name\'], varargs=None, keywords=None, d…
/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…