Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/lite/micro/kernels/
Delementwise_test.cc33 const int output_dims_count = ElementCount(*output_dims); in TestElementwiseFloat() local
69 const int output_dims_count = ElementCount(*output_dims); in TestElementwiseBool() local
105 constexpr int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
116 constexpr int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
127 constexpr int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
138 constexpr int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
149 constexpr int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
160 constexpr int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
171 constexpr int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
182 constexpr int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
Dpack_test.cc58 const int output_dims_count = ElementCount(*output_dims); in TestPackTwoInputsFloat() local
91 const int output_dims_count = ElementCount(*output_dims); in TestPackThreeInputsFloat() local
125 const int output_dims_count = ElementCount(*output_dims); in TestPackTwoInputsQuantized() local
162 const int output_dims_count = ElementCount(*output_dims); in TestPackTwoInputsQuantized32() local
198 constexpr int output_dims_count = 6; in TF_LITE_MICRO_TEST() local
214 constexpr int output_dims_count = 6; in TF_LITE_MICRO_TEST() local
230 constexpr int output_dims_count = 6; in TF_LITE_MICRO_TEST() local
245 constexpr int output_dims_count = 12; in TF_LITE_MICRO_TEST() local
260 constexpr int output_dims_count = 12; in TF_LITE_MICRO_TEST() local
275 constexpr int output_dims_count = 12; in TF_LITE_MICRO_TEST() local
Dadd_test.cc152 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
164 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
177 const int output_dims_count = 6; in TF_LITE_MICRO_TEST() local
201 const int output_dims_count = 6; in TF_LITE_MICRO_TEST() local
228 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
249 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
270 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
291 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
312 const int output_dims_count = 6; in TF_LITE_MICRO_TEST() local
344 const int output_dims_count = 6; in TF_LITE_MICRO_TEST() local
[all …]
Dsub_test.cc151 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
163 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
176 const int output_dims_count = 6; in TF_LITE_MICRO_TEST() local
200 const int output_dims_count = 6; in TF_LITE_MICRO_TEST() local
227 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
248 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
269 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
290 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
311 const int output_dims_count = 6; in TF_LITE_MICRO_TEST() local
343 const int output_dims_count = 6; in TF_LITE_MICRO_TEST() local
[all …]
Dconv_test.cc82 const int output_dims_count = 2; in TF_LITE_MICRO_TEST() local
102 const int output_dims_count = 12; in TF_LITE_MICRO_TEST() local
136 const int output_dims_count = tflite::ElementCount(*output_dims); in TF_LITE_MICRO_TEST() local
165 const int output_dims_count = tflite::ElementCount(*output_dims); in TF_LITE_MICRO_TEST() local
188 const int output_dims_count = 24; in TF_LITE_MICRO_TEST() local
228 const int output_dims_count = 12; in TF_LITE_MICRO_TEST() local
258 const int output_dims_count = 24; in TF_LITE_MICRO_TEST() local
302 const int output_dims_count = 12; in TF_LITE_MICRO_TEST() local
436 const int output_dims_count = 12; in TF_LITE_MICRO_TEST() local
525 const int output_dims_count = 12; in TF_LITE_MICRO_TEST() local
[all …]
Dzeros_like_test.cc31 const int output_dims_count = ElementCount(*output_dims); in TestZerosLikeFloat() local
63 const int output_dims_count = ElementCount(*output_dims); in TestZerosLikeInt32() local
95 const int output_dims_count = ElementCount(*output_dims); in TestZerosLikeInt64() local
Dcast_test.cc32 const int output_dims_count = ElementCount(*output_dims); in TestCastFloatToInt8() local
64 const int output_dims_count = ElementCount(*output_dims); in TestCastInt8ToFloat() local
Dprelu_test.cc55 const int output_dims_count = ElementCount(*output_dims); in TestPreluFloat() local
83 const int output_dims_count = ElementCount(*output_dims); in TestPreluQuantized() local
125 const int output_dims_count = 12; in TF_LITE_MICRO_TEST() local
Dcircular_buffer_test.cc57 const int output_dims_count = tflite::ElementCount(*output_tensor_dims); in TF_LITE_MICRO_TEST() local
125 const int output_dims_count = tflite::ElementCount(*output_tensor_dims); in TF_LITE_MICRO_TEST() local
Dtranspose_conv_test.cc109 const int output_dims_count = ElementCount(*output_dims); in TestTransposeConvFloat() local
145 const int output_dims_count = ElementCount(*output_dims); in TestTransposeConvQuantized() local
233 const int output_dims_count = tflite::ElementCount(*output_dims); in TF_LITE_MICRO_TEST() local
267 const int output_dims_count = tflite::ElementCount(*output_dims); in TF_LITE_MICRO_TEST() local
Dsvdf_test.cc687 const int output_dims_count = batch_size * num_units; in TF_LITE_MICRO_TEST() local
714 const int output_dims_count = batch_size * num_units; in TF_LITE_MICRO_TEST() local
768 constexpr int output_dims_count = batch_size * num_units; in TF_LITE_MICRO_TEST() local
799 constexpr int output_dims_count = batch_size * num_units; in TF_LITE_MICRO_TEST() local
831 constexpr int output_dims_count = batch_size * num_units; in TF_LITE_MICRO_TEST() local
884 constexpr int output_dims_count = batch_size * num_units; in TF_LITE_MICRO_TEST() local
Dmaximum_minimum_test.cc35 const int output_dims_count = ElementCount(*output_dims); in TestMaxMinFloat() local
76 const int output_dims_count = ElementCount(*output_dims); in TestMaxMinQuantized() local
115 const int output_dims_count = ElementCount(*output_dims); in TestMaxMinQuantizedInt32() local
Dshape_test.cc29 int output_dims_count) { in ValidateShape()
52 const int output_dims_count = ElementCount(*output_dims); in TestShape() local
Dconcatenation_test.cc64 const int output_dims_count = ElementCount(*output_dims); in TestConcatenateTwoInputs() local
111 const int output_dims_count = ElementCount(*output_dims); in TestConcatenateQuantizedTwoInputs() local
Ddepthwise_conv_test.cc96 const int output_dims_count = ElementCount(*output_dims); in TestDepthwiseConvFloat() local
125 const int output_dims_count = ElementCount(*output_dims); in TestDepthwiseConvQuantizedPerLayer() local
177 const int output_dims_count = ElementCount(*output_dims); in TestDepthwiseConvQuantizedPerChannel() local
255 const int output_dims_count = 8; in TF_LITE_MICRO_TEST() local
367 const int output_dims_count = 8; in TF_LITE_MICRO_TEST() local
430 const int output_dims_count = 9; in TF_LITE_MICRO_TEST() local
481 const int output_dims_count = 8; in TF_LITE_MICRO_TEST() local
524 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
696 const int output_dims_count = 8; in TF_LITE_MICRO_TEST() local
Dexp_test.cc32 const int output_dims_count = ElementCount(*output_dims); in TestExp() local
Dround_test.cc31 const int output_dims_count = ElementCount(*output_dims); in TestRound() local
Dceil_test.cc31 const int output_dims_count = ElementCount(*output_dims); in TestCeil() local
Dfloor_test.cc32 const int output_dims_count = ElementCount(*output_dims); in TestFloor() local
Dneg_test.cc32 const int output_dims_count = ElementCount(*output_dims); in TestNegFloat() local
Dsoftmax_test.cc256 int output_dims_count, float tolerance) { in ValidateSoftmaxGoldens()
282 const int output_dims_count = ElementCount(*output_dims); in TestSoftmaxFloat() local
306 const int output_dims_count = ElementCount(*output_dims); in TestSoftmaxQuantized() local
Dconv_test_common.cc126 const int output_dims_count = ElementCount(*output_dims); in TestConvFloat() local
155 const int output_dims_count = ElementCount(*output_dims); in TestConvQuantizedPerLayer() local
198 const int output_dims_count = ElementCount(*output_dims); in TestConvQuantizedPerChannel() local
Dmul_test.cc77 const int output_dims_count = ElementCount(*output_dims); in TestMulFloat() local
104 const int output_dims_count = ElementCount(*output_dims); in TestMulQuantized() local
/external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/
Dconv_slicing_test.cc205 const int output_dims_count = ElementCount(*output_dims); in TestConvQuantizedPerChannel() local
281 const int output_dims_count = 20; in TF_LITE_MICRO_TEST() local
308 const int output_dims_count = 20; in TF_LITE_MICRO_TEST() local
338 const int output_dims_count = 80; in TF_LITE_MICRO_TEST() local
365 const int output_dims_count = 80; in TF_LITE_MICRO_TEST() local
395 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
422 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
452 const int output_dims_count = 8; in TF_LITE_MICRO_TEST() local
479 const int output_dims_count = 8; in TF_LITE_MICRO_TEST() local
Ddepthwise_conv_slicing_test.cc132 const int output_dims_count = ElementCount(*output_dims); in TestDepthwiseConvQuantizedPerChannel() local
211 const int output_dims_count = 20; in TF_LITE_MICRO_TEST() local
242 const int output_dims_count = 20; in TF_LITE_MICRO_TEST() local
301 const int output_dims_count = 80; in TF_LITE_MICRO_TEST() local
332 const int output_dims_count = 80; in TF_LITE_MICRO_TEST() local
393 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
424 const int output_dims_count = 4; in TF_LITE_MICRO_TEST() local
482 const int output_dims_count = 8; in TF_LITE_MICRO_TEST() local
513 const int output_dims_count = 8; in TF_LITE_MICRO_TEST() local

12