| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | conv_test_common.cc | 22 TfLiteStatus InvokeConv(TfLiteTensor* tensors, int tensors_size, in InvokeConv() argument
30 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in InvokeConv()
42 TfLiteStatus ValidateConvGoldens(TfLiteTensor* tensors, int tensors_size, in ValidateConvGoldens() argument
48 TfLiteStatus status = InvokeConv(tensors, tensors_size, output_length, in ValidateConvGoldens()
60 TfLiteStatus InvokeConv(TfLiteTensor* tensors, int tensors_size, in InvokeConv() argument
63 return InvokeConv<float>(tensors, tensors_size, output_length, conv_params, in InvokeConv()
67 TfLiteStatus InvokeConv(TfLiteTensor* tensors, int tensors_size, in InvokeConv() argument
70 return InvokeConv<int8_t>(tensors, tensors_size, output_length, conv_params, in InvokeConv()
74 TfLiteStatus InvokeConv(TfLiteTensor* tensors, int tensors_size, in InvokeConv() argument
77 return InvokeConv<uint8_t>(tensors, tensors_size, output_length, conv_params, in InvokeConv()
[all …]
|
| D | activations_test.cc | 36 constexpr int tensors_size = inputs_size + outputs_size; in TestReluFloat() local
37 TfLiteTensor tensors[tensors_size] = { in TestReluFloat()
48 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestReluFloat()
69 constexpr int tensors_size = inputs_size + outputs_size; in TestRelu6Float() local
70 TfLiteTensor tensors[tensors_size] = { in TestRelu6Float()
81 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestRelu6Float()
105 constexpr int tensors_size = inputs_size + outputs_size; in TestReluUint8() local
106 TfLiteTensor tensors[tensors_size] = { in TestReluUint8()
119 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestReluUint8()
146 constexpr int tensors_size = inputs_size + outputs_size; in TestRelu6Uint8() local
[all …]
|
| D | transpose_conv_test.cc | 59 TfLiteStatus InvokeTransposeConv(TfLiteTensor* tensors, int tensors_size, in InvokeTransposeConv() argument
69 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in InvokeTransposeConv()
82 int tensors_size, in ValidateTransposeConvGoldens() argument
88 tensors, tensors_size, output_length, conv_params, output_data); in ValidateTransposeConvGoldens()
117 constexpr int tensors_size = inputs_size + outputs_size; in TestTransposeConvFloat() local
118 TfLiteTensor tensors[tensors_size] = { in TestTransposeConvFloat()
126 return ValidateTransposeConvGoldens(tensors, tensors_size, in TestTransposeConvFloat()
162 constexpr int tensors_size = inputs_size + outputs_size; in TestTransposeConvQuantized() local
163 TfLiteTensor tensors[tensors_size] = { in TestTransposeConvQuantized()
173 tensors, tensors_size, expected_output_quantized, output_dims_count, in TestTransposeConvQuantized()
[all …]
|
| D | pack_test.cc | 27 void ValidatePackGoldens(TfLiteTensor* tensors, int tensors_size, in ValidatePackGoldens() argument
37 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in ValidatePackGoldens()
62 constexpr int tensors_size = input_size + output_size; in TestPackTwoInputsFloat() local
63 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestPackTwoInputsFloat()
76 ValidatePackGoldens(tensors, tensors_size, builtin_data, inputs_array, in TestPackTwoInputsFloat()
95 constexpr int tensors_size = input_size + output_size; in TestPackThreeInputsFloat() local
96 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestPackThreeInputsFloat()
110 ValidatePackGoldens(tensors, tensors_size, builtin_data, inputs_array, in TestPackThreeInputsFloat()
129 constexpr int tensors_size = input_size + output_size; in TestPackTwoInputsQuantized() local
130 TfLiteTensor tensors[tensors_size] = { in TestPackTwoInputsQuantized()
[all …]
|
| D | zeros_like_test.cc | 34 constexpr int tensors_size = inputs_size + outputs_size; in TestZerosLikeFloat() local
35 TfLiteTensor tensors[tensors_size] = { in TestZerosLikeFloat()
46 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestZerosLikeFloat()
66 constexpr int tensors_size = inputs_size + outputs_size; in TestZerosLikeInt32() local
67 TfLiteTensor tensors[tensors_size] = { in TestZerosLikeInt32()
78 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestZerosLikeInt32()
98 constexpr int tensors_size = inputs_size + outputs_size; in TestZerosLikeInt64() local
99 TfLiteTensor tensors[tensors_size] = { in TestZerosLikeInt64()
110 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestZerosLikeInt64()
|
| D | dequantize_test.cc | 27 void ValidateDequantizeGoldens(TfLiteTensor* tensors, int tensors_size, in ValidateDequantizeGoldens() argument
37 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in ValidateDequantizeGoldens()
60 const int tensors_size = 2; in TestDequantizeToFloat() local
61 TfLiteTensor tensors[tensors_size] = { in TestDequantizeToFloat()
67 ValidateDequantizeGoldens(tensors, tensors_size, expected_output_data, in TestDequantizeToFloat()
83 const int tensors_size = 2; in TestDequantizeToInt32() local
84 TfLiteTensor tensors[tensors_size] = { in TestDequantizeToInt32()
93 ValidateDequantizeGoldens(tensors, tensors_size, expected_output_data, in TestDequantizeToInt32()
|
| D | pad_test.cc | 28 TfLiteStatus ValidatePadGoldens(TfLiteTensor* tensors, int tensors_size, in ValidatePadGoldens() argument
37 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in ValidatePadGoldens()
60 TfLiteStatus ValidatePadV2Goldens(TfLiteTensor* tensors, int tensors_size, in ValidatePadV2Goldens() argument
69 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in ValidatePadV2Goldens()
103 constexpr int tensors_size = inputs_size + outputs_size; in TestPadFloat() local
104 TfLiteTensor tensors[tensors_size] = {CreateTensor(input_data, input_dims), in TestPadFloat()
112 ValidatePadGoldens(tensors, tensors_size, golden, in TestPadFloat()
130 constexpr int tensors_size = inputs_size + outputs_size; in TestPadV2Float() local
131 TfLiteTensor tensors[tensors_size] = { in TestPadV2Float()
140 ValidatePadV2Goldens(tensors, tensors_size, golden, in TestPadV2Float()
[all …]
|
| D | conv_test.h | 29 TfLiteStatus InvokeConv(TfLiteTensor* tensors, int tensors_size,
33 TfLiteStatus InvokeConv(TfLiteTensor* tensors, int tensors_size,
37 TfLiteStatus InvokeConv(TfLiteTensor* tensors, int tensors_size,
41 TfLiteStatus ValidateConvGoldens(TfLiteTensor* tensors, int tensors_size,
48 TfLiteStatus ValidateConvGoldens(TfLiteTensor* tensors, int tensors_size,
55 TfLiteStatus ValidateConvGoldens(TfLiteTensor* tensors, int tensors_size,
|
| D | prelu_test.cc | 27 void ValidatePreluGoldens(TfLiteTensor* tensors, int tensors_size, in ValidatePreluGoldens() argument
36 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in ValidatePreluGoldens()
58 constexpr int tensors_size = inputs_size + outputs_size; in TestPreluFloat() local
59 TfLiteTensor tensors[tensors_size] = { in TestPreluFloat()
65 ValidatePreluGoldens(tensors, tensors_size, expected_output_data, in TestPreluFloat()
86 constexpr int tensors_size = inputs_size + outputs_size; in TestPreluQuantized() local
87 TfLiteTensor tensors[tensors_size] = { in TestPreluQuantized()
99 ValidatePreluGoldens(tensors, tensors_size, golden_quantized, in TestPreluQuantized()
|
| D | cast_test.cc | 35 constexpr int tensors_size = inputs_size + outputs_size; in TestCastFloatToInt8() local
36 TfLiteTensor tensors[tensors_size] = { in TestCastFloatToInt8()
47 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestCastFloatToInt8()
67 constexpr int tensors_size = inputs_size + outputs_size; in TestCastInt8ToFloat() local
68 TfLiteTensor tensors[tensors_size] = { in TestCastInt8ToFloat()
79 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestCastInt8ToFloat()
|
| D | batch_to_space_nd_test.cc | 42 int tensors_size, const T* golden, in ValidateBatchToSpaceNdGoldens() argument
50 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in ValidateBatchToSpaceNdGoldens()
78 constexpr int tensors_size = inputs_size + outputs_size; in TestBatchToSpaceNdFloat() local
79 TfLiteTensor tensors[tensors_size] = { in TestBatchToSpaceNdFloat()
86 return ValidateBatchToSpaceNdGoldens(tensors, tensors_size, golden, in TestBatchToSpaceNdFloat()
105 constexpr int tensors_size = inputs_size + outputs_size; in TestBatchToSpaceNdQuantized() local
106 TfLiteTensor tensors[tensors_size] = { in TestBatchToSpaceNdQuantized()
118 return ValidateBatchToSpaceNdGoldens(tensors, tensors_size, golden_quantized, in TestBatchToSpaceNdQuantized()
|
| D | space_to_batch_nd_test.cc | 42 int tensors_size, const T* golden, in ValidateSpaceToBatchNdGoldens() argument
50 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in ValidateSpaceToBatchNdGoldens()
78 constexpr int tensors_size = inputs_size + outputs_size; in TestSpaceToBatchNdFloat() local
79 TfLiteTensor tensors[tensors_size] = { in TestSpaceToBatchNdFloat()
86 return ValidateSpaceToBatchNdGoldens(tensors, tensors_size, golden, in TestSpaceToBatchNdFloat()
105 constexpr int tensors_size = inputs_size + outputs_size; in TestSpaceToBatchNdQuantized() local
106 TfLiteTensor tensors[tensors_size] = { in TestSpaceToBatchNdQuantized()
118 return ValidateSpaceToBatchNdGoldens(tensors, tensors_size, golden_quantized, in TestSpaceToBatchNdQuantized()
|
| D | maximum_minimum_test.cc | 39 constexpr int tensors_size = inputs_size + outputs_size; in TestMaxMinFloat() local
40 TfLiteTensor tensors[tensors_size] = { in TestMaxMinFloat()
51 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestMaxMinFloat()
80 constexpr int tensors_size = inputs_size + outputs_size; in TestMaxMinQuantized() local
81 TfLiteTensor tensors[tensors_size] = { in TestMaxMinQuantized()
95 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestMaxMinQuantized()
119 constexpr int tensors_size = inputs_size + outputs_size; in TestMaxMinQuantizedInt32() local
120 TfLiteTensor tensors[tensors_size] = { in TestMaxMinQuantizedInt32()
131 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestMaxMinQuantizedInt32()
|
| D | unpack_test.cc | 42 constexpr int tensors_size = input_size + output_size; in TestUnpackThreeOutputsFloat() local
43 TfLiteTensor tensors[tensors_size] = { in TestUnpackThreeOutputsFloat()
73 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestUnpackThreeOutputsFloat()
104 constexpr int tensors_size = input_size + output_size; in TestUnpackOneOutputFloat() local
105 TfLiteTensor tensors[tensors_size] = {CreateTensor(input_data, input_dims), in TestUnpackOneOutputFloat()
124 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestUnpackOneOutputFloat()
152 constexpr int tensors_size = input_size + output_size; in TestUnpackThreeOutputsQuantized() local
153 TfLiteTensor tensors[tensors_size] = { in TestUnpackThreeOutputsQuantized()
186 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestUnpackThreeOutputsQuantized()
222 constexpr int tensors_size = input_size + output_size; in TestUnpackThreeOutputsQuantized32() local
[all …]
|
| D | circular_buffer_test.cc | 61 constexpr int tensors_size = inputs_size + outputs_size; in TF_LITE_MICRO_TEST() local
62 TfLiteTensor tensors[tensors_size] = { in TF_LITE_MICRO_TEST()
81 *registration, tensors, tensors_size, inputs_array, outputs_array, in TF_LITE_MICRO_TEST()
129 constexpr int tensors_size = inputs_size + outputs_size; in TF_LITE_MICRO_TEST() local
130 TfLiteTensor tensors[tensors_size] = { in TF_LITE_MICRO_TEST()
149 *registration, tensors, tensors_size, inputs_array, outputs_array, in TF_LITE_MICRO_TEST()
|
| D | arg_min_max_test.cc | 27 void ValidateArgMinMaxGoldens(TfLiteTensor* tensors, int tensors_size, in ValidateArgMinMaxGoldens() argument
38 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in ValidateArgMinMaxGoldens()
61 constexpr int tensors_size = inputs_size + outputs_size; in TestArgMinMaxFloat() local
62 TfLiteTensor tensors[tensors_size] = { in TestArgMinMaxFloat()
68 ValidateArgMinMaxGoldens(tensors, tensors_size, goldens, output, in TestArgMinMaxFloat()
87 constexpr int tensors_size = inputs_size + outputs_size; in TestArgMinMaxQuantized() local
88 TfLiteTensor tensors[tensors_size] = { in TestArgMinMaxQuantized()
95 ValidateArgMinMaxGoldens(tensors, tensors_size, goldens, output, in TestArgMinMaxQuantized()
|
| D | logistic_test.cc | 80 constexpr int tensors_size = inputs_size + outputs_size; in TestLogisticFloat() local
81 TfLiteTensor tensors[tensors_size] = { in TestLogisticFloat()
86 ValidateLogisticGoldens(tensors, tensors_size, output_data, golden, in TestLogisticFloat()
103 constexpr int tensors_size = inputs_size + outputs_size; in TestLogisticQuantized() local
104 TfLiteTensor tensors[tensors_size] = { in TestLogisticQuantized()
113 ValidateLogisticGoldens(tensors, tensors_size, output_data, golden_quantized, in TestLogisticQuantized()
|
| D | concatenation_test.cc | 40 constexpr int tensors_size = input_size + output_size; in TestConcatenateTwoInputs() local
41 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestConcatenateTwoInputs()
57 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestConcatenateTwoInputs()
83 constexpr int tensors_size = input_size + output_size; in TestConcatenateQuantizedTwoInputs() local
84 TfLiteTensor tensors[tensors_size] = { in TestConcatenateQuantizedTwoInputs()
104 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestConcatenateQuantizedTwoInputs()
|
| D | tanh_test.cc | 78 constexpr int tensors_size = inputs_size + outputs_size; in TestTanhFloat() local
79 TfLiteTensor tensors[tensors_size] = { in TestTanhFloat()
90 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestTanhFloat()
120 constexpr int tensors_size = inputs_size + outputs_size; in TestTanhQuantized() local
121 TfLiteTensor tensors[tensors_size] = { in TestTanhQuantized()
133 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestTanhQuantized()
|
| D | elementwise_test.cc | 37 constexpr int tensors_size = input_size + output_size; in TestElementwiseFloat() local
38 TfLiteTensor tensors[tensors_size] = {CreateTensor(input_data, input_dims), in TestElementwiseFloat()
51 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestElementwiseFloat()
73 constexpr int tensors_size = input_size + output_size; in TestElementwiseBool() local
74 TfLiteTensor tensors[tensors_size] = {CreateTensor(input_data, input_dims), in TestElementwiseBool()
87 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestElementwiseBool()
|
| D | split_test.cc | 43 constexpr int tensors_size = input_size + output_size + axis_size; in TestSplitTwoOutputsFloat() local
44 TfLiteTensor tensors[tensors_size] = { in TestSplitTwoOutputsFloat()
66 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestSplitTwoOutputsFloat()
104 constexpr int tensors_size = input_size + output_size + axis_size; in TestSplitFourOutputsFloat() local
105 TfLiteTensor tensors[tensors_size] = { in TestSplitFourOutputsFloat()
135 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestSplitFourOutputsFloat()
171 constexpr int tensors_size = input_size + output_size + axis_size; in TestSplitTwoOutputsQuantized() local
172 TfLiteTensor tensors[tensors_size] = { in TestSplitTwoOutputsQuantized()
196 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestSplitTwoOutputsQuantized()
227 constexpr int tensors_size = input_size + output_size + axis_size; in TestSplitTwoOutputsQuantized32() local
[all …]
|
| D | quantize_test.cc | 27 void ValidateQuantizeGoldens(TfLiteTensor* tensors, int tensors_size, in ValidateQuantizeGoldens() argument
38 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in ValidateQuantizeGoldens()
73 constexpr int tensors_size = 2; in TestQuantizeFloat() local
74 TfLiteTensor tensors[tensors_size] = { in TestQuantizeFloat()
79 ValidateQuantizeGoldens(tensors, tensors_size, golden, golden_quantized, in TestQuantizeFloat()
106 constexpr int tensors_size = 2; in TestRequantize() local
107 TfLiteTensor tensors[tensors_size] = { in TestRequantize()
113 ValidateQuantizeGoldens(tensors, tensors_size, golden, golden_quantized, in TestRequantize()
|
| D | mul_test.cc | 45 void ValidateMulGoldens(TfLiteTensor* tensors, int tensors_size, in ValidateMulGoldens() argument
58 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in ValidateMulGoldens()
81 constexpr int tensors_size = inputs_size + outputs_size; in TestMulFloat() local
82 TfLiteTensor tensors[tensors_size] = { in TestMulFloat()
88 ValidateMulGoldens(tensors, tensors_size, activation, golden, in TestMulFloat()
108 constexpr int tensors_size = inputs_size + outputs_size; in TestMulQuantized() local
109 TfLiteTensor tensors[tensors_size] = { in TestMulQuantized()
120 ValidateMulGoldens(tensors, tensors_size, activation, golden_quantized, in TestMulQuantized()
|
| D | hard_swish_test.cc | 95 constexpr int tensors_size = inputs_size + outputs_size; in TestHardSwishQuantized() local
96 TfLiteTensor tensors[tensors_size] = { in TestHardSwishQuantized()
110 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestHardSwishQuantized()
174 constexpr int tensors_size = inputs_size + outputs_size; in TestHardSwishQuantizedBias() local
175 TfLiteTensor tensors[tensors_size] = { in TestHardSwishQuantizedBias()
189 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestHardSwishQuantizedBias()
228 constexpr int tensors_size = inputs_size + outputs_size; in TestHardSwishFloat() local
229 TfLiteTensor tensors[tensors_size] = { in TestHardSwishFloat()
241 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestHardSwishFloat()
|
| D | pooling_test.cc | 29 void ValidatePoolingGoldens(TfLiteTensor* tensors, int tensors_size, in ValidatePoolingGoldens() argument
49 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in ValidatePoolingGoldens()
74 constexpr int tensors_size = inputs_size + outputs_size; in TestAveragePoolFloat() local
75 TfLiteTensor tensors[tensors_size] = { in TestAveragePoolFloat()
83 ValidatePoolingGoldens(tensors, tensors_size, registration, filter_height, in TestAveragePoolFloat()
105 constexpr int tensors_size = inputs_size + outputs_size; in TestAveragePoolQuantized() local
106 TfLiteTensor tensors[tensors_size] = { in TestAveragePoolQuantized()
115 ValidatePoolingGoldens(tensors, tensors_size, registration, filter_height, in TestAveragePoolQuantized()
132 constexpr int tensors_size = inputs_size + outputs_size; in TestMaxPoolFloat() local
133 TfLiteTensor tensors[tensors_size] = { in TestMaxPoolFloat()
[all …]
|