Home
last modified time | relevance | path

Searched refs:output_dims_count (Results 1 – 25 of 43) sorted by relevance

12

/external/tensorflow/tensorflow/lite/micro/kernels/
Dadd_test.cc152 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
157 float output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
164 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
170 float output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
177 const int output_dims_count = 6; in TF_LITE_MICRO_TEST() local
178 float output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
201 const int output_dims_count = 6; in TF_LITE_MICRO_TEST() local
202 float output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
228 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
234 uint8_t input1_quantized[output_dims_count]; in TF_LITE_MICRO_TEST()
[all …]
Dsub_test.cc151 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
156 float output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
163 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
169 float output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
176 const int output_dims_count = 6; in TF_LITE_MICRO_TEST() local
177 float output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
200 const int output_dims_count = 6; in TF_LITE_MICRO_TEST() local
201 float output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
227 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
233 uint8_t input1_quantized[output_dims_count]; in TF_LITE_MICRO_TEST()
[all …]
Delementwise_test.cc33 const int output_dims_count = ElementCount(*output_dims); in TestElementwiseFloat() local
42 for (int i = 0; i < output_dims_count; ++i) { in TestElementwiseFloat()
58 for (int i = 0; i < output_dims_count; ++i) { in TestElementwiseFloat()
69 const int output_dims_count = ElementCount(*output_dims); in TestElementwiseBool() local
78 for (int i = 0; i < output_dims_count; ++i) { in TestElementwiseBool()
94 for (int i = 0; i < output_dims_count; ++i) { in TestElementwiseBool()
105 constexpr int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
109 float output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
116 constexpr int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
120 float output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
[all …]
Dpack_test.cc58 const int output_dims_count = ElementCount(*output_dims); in TestPackTwoInputsFloat() local
77 outputs_array, expected_output_data, output_dims_count, in TestPackTwoInputsFloat()
91 const int output_dims_count = ElementCount(*output_dims); in TestPackThreeInputsFloat() local
111 outputs_array, expected_output_data, output_dims_count, in TestPackThreeInputsFloat()
125 const int output_dims_count = ElementCount(*output_dims); in TestPackTwoInputsQuantized() local
148 outputs_array, expected_output_data, output_dims_count, in TestPackTwoInputsQuantized()
162 const int output_dims_count = ElementCount(*output_dims); in TestPackTwoInputsQuantized32() local
181 outputs_array, expected_output_data, output_dims_count, in TestPackTwoInputsQuantized32()
198 constexpr int output_dims_count = 6; in TF_LITE_MICRO_TEST() local
199 float output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
[all …]
Dconv_test.cc82 const int output_dims_count = 2; in TF_LITE_MICRO_TEST() local
83 float output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
102 const int output_dims_count = 12; in TF_LITE_MICRO_TEST() local
103 uint8_t output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
136 const int output_dims_count = tflite::ElementCount(*output_dims); in TF_LITE_MICRO_TEST() local
151 tflite::testing::InvokeConv(tensors, tensors_size, output_dims_count, in TF_LITE_MICRO_TEST()
165 const int output_dims_count = tflite::ElementCount(*output_dims); in TF_LITE_MICRO_TEST() local
182 tflite::testing::InvokeConv(tensors, tensors_size, output_dims_count, in TF_LITE_MICRO_TEST()
188 const int output_dims_count = 24; in TF_LITE_MICRO_TEST() local
189 uint8_t output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
[all …]
Dprelu_test.cc55 const int output_dims_count = ElementCount(*output_dims); in TestPreluFloat() local
66 output_dims_count, output_data); in TestPreluFloat()
83 const int output_dims_count = ElementCount(*output_dims); in TestPreluQuantized() local
96 Quantize(golden, golden_quantized, output_dims_count, output_scale, in TestPreluQuantized()
100 output_dims_count, output_data); in TestPreluQuantized()
125 const int output_dims_count = 12; in TF_LITE_MICRO_TEST() local
126 float output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
Dzeros_like_test.cc31 const int output_dims_count = ElementCount(*output_dims); in TestZerosLikeFloat() local
53 for (int i = 0; i < output_dims_count; ++i) { in TestZerosLikeFloat()
63 const int output_dims_count = ElementCount(*output_dims); in TestZerosLikeInt32() local
85 for (int i = 0; i < output_dims_count; ++i) { in TestZerosLikeInt32()
95 const int output_dims_count = ElementCount(*output_dims); in TestZerosLikeInt64() local
117 for (int i = 0; i < output_dims_count; ++i) { in TestZerosLikeInt64()
Ddepthwise_conv_test.cc96 const int output_dims_count = ElementCount(*output_dims); in TestDepthwiseConvFloat() local
108 ValidateDepthwiseConvGoldens(expected_output_data, output_dims_count, in TestDepthwiseConvFloat()
125 const int output_dims_count = ElementCount(*output_dims); in TestDepthwiseConvQuantizedPerLayer() local
157 Quantize(golden, golden_quantized, output_dims_count, output_scale, in TestDepthwiseConvQuantizedPerLayer()
159 ValidateDepthwiseConvGoldens(golden_quantized, output_dims_count, conv_params, in TestDepthwiseConvQuantizedPerLayer()
177 const int output_dims_count = ElementCount(*output_dims); in TestDepthwiseConvQuantizedPerChannel() local
224 output_dims_count, output_scale, output_zero_point); in TestDepthwiseConvQuantizedPerChannel()
228 output_dims_count, conv_params, in TestDepthwiseConvQuantizedPerChannel()
255 const int output_dims_count = 8; in TF_LITE_MICRO_TEST() local
256 float output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
[all …]
Dconv_test_common.cc126 const int output_dims_count = ElementCount(*output_dims); in TestConvFloat() local
138 output_dims_count, conv_params, registration, in TestConvFloat()
155 const int output_dims_count = ElementCount(*output_dims); in TestConvQuantizedPerLayer() local
158 output_dims_count, output_scale, 128); in TestConvQuantizedPerLayer()
180 output_dims_count, conv_params, registration, in TestConvQuantizedPerLayer()
198 const int output_dims_count = ElementCount(*output_dims); in TestConvQuantizedPerChannel() local
240 output_dims_count, output_scale, output_zero_point); in TestConvQuantizedPerChannel()
242 tensors, tensors_size, expected_output_data_quantized, output_dims_count, in TestConvQuantizedPerChannel()
Dtranspose_conv_test.cc109 const int output_dims_count = ElementCount(*output_dims); in TestTransposeConvFloat() local
127 expected_output_data, output_dims_count, in TestTransposeConvFloat()
145 const int output_dims_count = ElementCount(*output_dims); in TestTransposeConvQuantized() local
154 output_dims_count, output_scale, 0); in TestTransposeConvQuantized()
173 tensors, tensors_size, expected_output_quantized, output_dims_count, in TestTransposeConvQuantized()
233 const int output_dims_count = tflite::ElementCount(*output_dims); in TF_LITE_MICRO_TEST() local
254 tensors, tensors_size, output_dims_count, in TF_LITE_MICRO_TEST()
267 const int output_dims_count = tflite::ElementCount(*output_dims); in TF_LITE_MICRO_TEST() local
292 tensors, tensors_size, output_dims_count, in TF_LITE_MICRO_TEST()
Dpad_test.cc100 const int output_dims_count = ElementCount(*output_dims); in TestPadFloat() local
113 output_data, output_dims_count)); in TestPadFloat()
127 const int output_dims_count = ElementCount(*output_dims); in TestPadV2Float() local
141 output_data, output_dims_count)); in TestPadV2Float()
155 const int output_dims_count = ElementCount(*output_dims); in TestPadQuantized() local
169 tflite::Quantize(golden, golden_quantized, output_dims_count, output_scale, in TestPadQuantized()
174 output_dims_count)); in TestPadQuantized()
194 const int output_dims_count = ElementCount(*output_dims); in TestPadV2Quantized() local
212 tflite::Quantize(golden, golden_quantized, output_dims_count, output_scale, in TestPadV2Quantized()
217 output_dims_count)); in TestPadV2Quantized()
Dsvdf_test.cc687 const int output_dims_count = batch_size * num_units; in TF_LITE_MICRO_TEST() local
688 float output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
714 const int output_dims_count = batch_size * num_units; in TF_LITE_MICRO_TEST() local
715 int8_t output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
768 constexpr int output_dims_count = batch_size * num_units; in TF_LITE_MICRO_TEST() local
772 float output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
799 constexpr int output_dims_count = batch_size * num_units; in TF_LITE_MICRO_TEST() local
803 float output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
831 constexpr int output_dims_count = batch_size * num_units; in TF_LITE_MICRO_TEST() local
834 int8_t output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
[all …]
Dcast_test.cc32 const int output_dims_count = ElementCount(*output_dims); in TestCastFloatToInt8() local
54 for (int i = 0; i < output_dims_count; ++i) { in TestCastFloatToInt8()
64 const int output_dims_count = ElementCount(*output_dims); in TestCastInt8ToFloat() local
86 for (int i = 0; i < output_dims_count; ++i) { in TestCastInt8ToFloat()
Dshape_test.cc29 int output_dims_count) { in ValidateShape() argument
42 for (int i = 0; i < output_dims_count; ++i) { in ValidateShape()
52 const int output_dims_count = ElementCount(*output_dims); in TestShape() local
63 output_dims_count); in TestShape()
Dmaximum_minimum_test.cc35 const int output_dims_count = ElementCount(*output_dims); in TestMaxMinFloat() local
58 for (int i = 0; i < output_dims_count; ++i) { in TestMaxMinFloat()
76 const int output_dims_count = ElementCount(*output_dims); in TestMaxMinQuantized() local
102 for (int i = 0; i < output_dims_count; ++i) { in TestMaxMinQuantized()
115 const int output_dims_count = ElementCount(*output_dims); in TestMaxMinQuantizedInt32() local
138 for (int i = 0; i < output_dims_count; ++i) { in TestMaxMinQuantizedInt32()
Dcircular_buffer_test.cc57 const int output_dims_count = tflite::ElementCount(*output_tensor_dims); in TF_LITE_MICRO_TEST() local
98 for (int j = 0; j < output_dims_count; ++j) { in TF_LITE_MICRO_TEST()
125 const int output_dims_count = tflite::ElementCount(*output_tensor_dims); in TF_LITE_MICRO_TEST() local
169 for (int j = 0; j < output_dims_count; ++j) { in TF_LITE_MICRO_TEST()
Dconcatenation_test.cc64 const int output_dims_count = ElementCount(*output_dims); in TestConcatenateTwoInputs() local
65 for (int i = 0; i < output_dims_count; ++i) { in TestConcatenateTwoInputs()
111 const int output_dims_count = ElementCount(*output_dims); in TestConcatenateQuantizedTwoInputs() local
112 for (int i = 0; i < output_dims_count; ++i) { in TestConcatenateQuantizedTwoInputs()
Dreduce_test.cc100 const int output_dims_count = ElementCount(*output_dims); in TestMeanFloatInput4D() local
117 output_dims_count, registration, params, tolerance)); in TestMeanFloatInput4D()
129 const int output_dims_count = ElementCount(*output_dims); in TestReduceOpFloat() local
144 output_dims_count, registration, params, tolerance)); in TestReduceOpFloat()
161 const int output_dims_count = ElementCount(*output_dims); in TestReduceOpQuantized() local
175 output_dims_count, output_scale, output_zero_point); in TestReduceOpQuantized()
180 output_data_quant, output_dims_count, registration, in TestReduceOpQuantized()
199 const int output_dims_count = ElementCount(*output_dims); in TestMeanOpQuantized() local
213 output_dims_count, output_scale, output_zero_point); in TestMeanOpQuantized()
218 output_data_quant, output_dims_count, in TestMeanOpQuantized()
Dlogical_test.cc35 const int output_dims_count = ElementCount(*output_dims); in TestLogicalOp() local
58 TF_LITE_MICRO_EXPECT_EQ(output_dims_count, 4); in TestLogicalOp()
59 for (int i = 0; i < output_dims_count; ++i) { in TestLogicalOp()
Dmul_test.cc77 const int output_dims_count = ElementCount(*output_dims); in TestMulFloat() local
89 output_dims_count, 1e-5, output_data); in TestMulFloat()
104 const int output_dims_count = ElementCount(*output_dims); in TestMulQuantized() local
117 Quantize(golden, golden_quantized, output_dims_count, output_scale, in TestMulQuantized()
121 output_dims_count, 1.0f, output_data); in TestMulQuantized()
Dsoftmax_test.cc256 int output_dims_count, float tolerance) { in ValidateSoftmaxGoldens() argument
271 for (int i = 0; i < output_dims_count; ++i) { in ValidateSoftmaxGoldens()
282 const int output_dims_count = ElementCount(*output_dims); in TestSoftmaxFloat() local
293 expected_output_data, output_dims_count, 1e-5); in TestSoftmaxFloat()
306 const int output_dims_count = ElementCount(*output_dims); in TestSoftmaxQuantized() local
318 Quantize(golden, golden_quantized, output_dims_count, output_scale, in TestSoftmaxQuantized()
322 output_dims_count, tolerance); in TestSoftmaxQuantized()
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Ddepthwise_conv_slicing_test.cc132 const int output_dims_count = ElementCount(*output_dims); in TestDepthwiseConvQuantizedPerChannel() local
179 output_dims_count, output_scale, output_zero_point); in TestDepthwiseConvQuantizedPerChannel()
183 output_dims_count, activation, in TestDepthwiseConvQuantizedPerChannel()
211 const int output_dims_count = 20; in TF_LITE_MICRO_TEST() local
212 int8_t output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
242 const int output_dims_count = 20; in TF_LITE_MICRO_TEST() local
251 int8_t output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
301 const int output_dims_count = 80; in TF_LITE_MICRO_TEST() local
302 int8_t output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
332 const int output_dims_count = 80; in TF_LITE_MICRO_TEST() local
[all …]
Dconv_slicing_test.cc205 const int output_dims_count = ElementCount(*output_dims); in TestConvQuantizedPerChannel() local
264 expected_output_data_quantized, output_dims_count, in TestConvQuantizedPerChannel()
269 output_data, output_dims_count, conv_params, in TestConvQuantizedPerChannel()
281 const int output_dims_count = 20; in TF_LITE_MICRO_TEST() local
291 int8_t output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
308 const int output_dims_count = 20; in TF_LITE_MICRO_TEST() local
318 static int8_t output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
338 const int output_dims_count = 80; in TF_LITE_MICRO_TEST() local
348 int8_t output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
365 const int output_dims_count = 80; in TF_LITE_MICRO_TEST() local
[all …]
Dfully_connected_slicing_test.cc50 const int output_dims_count = ElementCount(*output_dims); in TestFullyConnectedQuantized() local
113 for (int i = 0; i < output_dims_count; ++i) { in TestFullyConnectedQuantized()
145 const int output_dims_count = 6; in TF_LITE_MICRO_TEST() local
146 int8_t output_data[output_dims_count]; in TF_LITE_MICRO_TEST()
168 const int output_dims_count = 6; in TF_LITE_MICRO_TEST() local
177 int8_t output_data_local[output_dims_count]; in TF_LITE_MICRO_TEST()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/sparse_ops/
Dfully_connected.h42 const int output_dims_count = output_shape.DimensionsCount(); in FullyConnectedSparseWeight() local
44 const int batches = FlatSizeSkipDim(output_shape, output_dims_count - 1); in FullyConnectedSparseWeight()
46 output_shape, output_dims_count - 1); in FullyConnectedSparseWeight()
89 const int output_dims_count = output_shape.DimensionsCount(); in FullyConnectedSparseWeight1x4Impl() local
95 output_shape, output_dims_count - 1); in FullyConnectedSparseWeight1x4Impl()

12