| /tflite-micro-3.4.0-2.7.6/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 ValidateConvGoldens(TfLiteTensor* tensors, int tensors_size, in ValidateConvGoldens() argument
80 return ValidateConvGoldens<float>(tensors, tensors_size, expected_output_data, in ValidateConvGoldens()
[all …]
|
| 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 | 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()
104 constexpr int tensors_size = inputs_size + outputs_size; in TestReluInt8() local
105 TfLiteTensor tensors[tensors_size] = { in TestReluInt8()
118 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestReluInt8()
144 constexpr int tensors_size = inputs_size + outputs_size; in TestRelu6Int8() local
[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()
60 constexpr int tensors_size = input_size + output_size; in TestPackTwoInputsFloat() local
61 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestPackTwoInputsFloat()
74 ValidatePackGoldens(tensors, tensors_size, builtin_data, inputs_array, in TestPackTwoInputsFloat()
93 constexpr int tensors_size = input_size + output_size; in TestPackThreeInputsFloat() local
94 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestPackThreeInputsFloat()
108 ValidatePackGoldens(tensors, tensors_size, builtin_data, inputs_array, in TestPackThreeInputsFloat()
124 constexpr int tensors_size = input_size + output_size; in TestPackTwoInputsQuantized() local
125 TfLiteTensor tensors[tensors_size] = { in TestPackTwoInputsQuantized()
[all …]
|
| 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 | 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 | 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 | 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 | 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 | 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()
116 constexpr int tensors_size = inputs_size + outputs_size; in TestTransposeConvFloat() local
117 TfLiteTensor tensors[tensors_size] = { in TestTransposeConvFloat()
125 return ValidateTransposeConvGoldens(tensors, tensors_size, in TestTransposeConvFloat()
160 constexpr int tensors_size = inputs_size + outputs_size; in TestTransposeConvQuantized() local
161 TfLiteTensor tensors[tensors_size] = { in TestTransposeConvQuantized()
171 tensors, tensors_size, expected_output_quantized, output_dims_count, in TestTransposeConvQuantized()
[all …]
|
| 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 | circular_buffer_test.cc | 62 constexpr int tensors_size = inputs_size + outputs_size; in TF_LITE_MICRO_TEST() local
63 TfLiteTensor tensors[tensors_size] = { in TF_LITE_MICRO_TEST()
82 *registration, tensors, tensors_size, inputs_array, outputs_array, in TF_LITE_MICRO_TEST()
134 constexpr int tensors_size = inputs_size + outputs_size; in TF_LITE_MICRO_TEST() local
135 TfLiteTensor tensors[tensors_size] = { in TF_LITE_MICRO_TEST()
154 *registration, tensors, tensors_size, inputs_array, outputs_array, in TF_LITE_MICRO_TEST()
200 constexpr int tensors_size = inputs_size + outputs_size; in TF_LITE_MICRO_TEST() local
201 TfLiteTensor tensors[tensors_size] = { in TF_LITE_MICRO_TEST()
220 *registration, tensors, tensors_size, inputs_array, outputs_array, in TF_LITE_MICRO_TEST()
|
| 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()
78 constexpr int tensors_size = inputs_size + outputs_size; in TestMaxMinQuantized() local
79 TfLiteTensor tensors[tensors_size] = { in TestMaxMinQuantized()
93 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestMaxMinQuantized()
117 constexpr int tensors_size = inputs_size + outputs_size; in TestMaxMinQuantizedInt32() local
118 TfLiteTensor tensors[tensors_size] = { in TestMaxMinQuantizedInt32()
129 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()
103 constexpr int tensors_size = input_size + output_size; in TestUnpackOneOutputFloat() local
104 TfLiteTensor tensors[tensors_size] = {CreateTensor(input_data, input_dims), in TestUnpackOneOutputFloat()
123 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestUnpackOneOutputFloat()
151 constexpr int tensors_size = input_size + output_size; in TestUnpackThreeOutputsQuantized() local
152 TfLiteTensor tensors[tensors_size] = { in TestUnpackThreeOutputsQuantized()
185 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestUnpackThreeOutputsQuantized()
221 constexpr int tensors_size = input_size + output_size; in TestUnpackThreeOutputsQuantized32() local
[all …]
|
| 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 | logistic_test.cc | 79 constexpr int tensors_size = inputs_size + outputs_size; in TestLogisticFloat() local
80 TfLiteTensor tensors[tensors_size] = { in TestLogisticFloat()
85 ValidateLogisticGoldens(tensors, tensors_size, output_data, golden, in TestLogisticFloat()
102 constexpr int tensors_size = inputs_size + outputs_size; in TestLogisticQuantized() local
103 TfLiteTensor tensors[tensors_size] = { in TestLogisticQuantized()
112 ValidateLogisticGoldens(tensors, tensors_size, output_data, golden_quantized, in TestLogisticQuantized()
|
| D | tanh_test.cc | 77 constexpr int tensors_size = inputs_size + outputs_size; in TestTanhFloat() local
78 TfLiteTensor tensors[tensors_size] = { in TestTanhFloat()
89 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestTanhFloat()
118 constexpr int tensors_size = inputs_size + outputs_size; in TestTanhQuantized() local
119 TfLiteTensor tensors[tensors_size] = { in TestTanhQuantized()
131 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestTanhQuantized()
|
| 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()
86 constexpr int tensors_size = inputs_size + outputs_size; in TestArgMinMaxQuantized() local
87 TfLiteTensor tensors[tensors_size] = { in TestArgMinMaxQuantized()
94 ValidateArgMinMaxGoldens(tensors, tensors_size, goldens, output, in TestArgMinMaxQuantized()
|
| 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()
82 ValidatePoolingGoldens(tensors, tensors_size, registration, filter_height, in TestAveragePoolFloat()
104 constexpr int tensors_size = inputs_size + outputs_size; in TestAveragePoolQuantized() local
105 TfLiteTensor tensors[tensors_size] = { in TestAveragePoolQuantized()
113 ValidatePoolingGoldens(tensors, tensors_size, registration, filter_height, in TestAveragePoolQuantized()
130 constexpr int tensors_size = inputs_size + outputs_size; in TestMaxPoolFloat() local
131 TfLiteTensor tensors[tensors_size] = { in TestMaxPoolFloat()
[all …]
|
| 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 | hard_swish_test.cc | 95 constexpr int tensors_size = inputs_size + outputs_size; in TestHardSwishQuantized() local
96 TfLiteTensor tensors[tensors_size] = { in TestHardSwishQuantized()
109 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestHardSwishQuantized()
173 constexpr int tensors_size = inputs_size + outputs_size; in TestHardSwishQuantizedBias() local
174 TfLiteTensor tensors[tensors_size] = { in TestHardSwishQuantizedBias()
187 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestHardSwishQuantizedBias()
226 constexpr int tensors_size = inputs_size + outputs_size; in TestHardSwishFloat() local
227 TfLiteTensor tensors[tensors_size] = { in TestHardSwishFloat()
238 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestHardSwishFloat()
|
| D | neg_test.cc | 35 constexpr int tensors_size = inputs_size + outputs_size; in TestNegFloat() local
36 TfLiteTensor tensors[tensors_size] = { in TestNegFloat()
47 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestNegFloat()
|
| D | round_test.cc | 34 constexpr int tensors_size = inputs_size + outputs_size; in TestRound() local
35 TfLiteTensor tensors[tensors_size] = { in TestRound()
46 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestRound()
|
| D | floor_test.cc | 35 constexpr int tensors_size = inputs_size + outputs_size; in TestFloor() local
36 TfLiteTensor tensors[tensors_size] = { in TestFloor()
47 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestFloor()
|
| D | exp_test.cc | 35 constexpr int tensors_size = inputs_size + outputs_size; in TestExp() local
36 TfLiteTensor tensors[tensors_size] = { in TestExp()
47 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestExp()
|
