| /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()
61 TfLiteTensor tensors[tensors_size] = { in TestDequantizeToFloat() local
67 ValidateDequantizeGoldens(tensors, tensors_size, expected_output_data, in TestDequantizeToFloat()
84 TfLiteTensor tensors[tensors_size] = { in TestDequantizeToInt32() local
90 tensors[1].params.scale = output_scale; in TestDequantizeToInt32()
91 tensors[1].params.zero_point = output_zero_point; in TestDequantizeToInt32()
93 ValidateDequantizeGoldens(tensors, tensors_size, expected_output_data, in TestDequantizeToInt32()
|
| D | add_n_test.cc | 31 void ExecuteAddN(TfLiteTensor* tensors, int tensors_count) { in ExecuteAddN() argument
42 micro::KernelRunner runner(registration, tensors, tensors_count, inputs_array, in ExecuteAddN()
59 TfLiteTensor tensors[kMaxInputTensors + kMaxOutputTensors] = {}; in TestAddN() local
61 tensors[i] = CreateTensor(input_data[i], input_dims); in TestAddN()
63 tensors[input_data_count] = CreateTensor(output_data, output_dims); in TestAddN()
65 ExecuteAddN(tensors, input_data_count + 1); in TestAddN()
104 TfLiteTensor tensors[kMaxInputTensors + kMaxOutputTensors] = {}; in TestAddNQuantized() local
106 tensors[i] = CreateQuantizedTensor(input_data[i], params->input_data[i], in TestAddNQuantized()
109 tensors[kNumInputs] = CreateQuantizedTensor(params->output_data, output_dims, in TestAddNQuantized()
112 ExecuteAddN(tensors, kNumInputs + 1); in TestAddNQuantized()
|
| D | elu_test.cc | 56 void ExecuteEluTest(TfLiteTensor* tensors, int tensors_count) { in ExecuteEluTest() argument
63 micro::KernelRunner runner(registration, tensors, tensors_count, inputs_array, in ExecuteEluTest()
77 TfLiteTensor tensors[] = { in TestElu() local
81 constexpr int tensors_count = std::extent<decltype(tensors)>::value; in TestElu()
82 ExecuteEluTest(tensors, tensors_count); in TestElu()
102 TfLiteTensor tensors[] = { in TestEluQuantized() local
107 constexpr int kTensorsCount = std::extent<decltype(tensors)>::value; in TestEluQuantized()
109 ExecuteEluTest(tensors, kTensorsCount); in TestEluQuantized()
|
| 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()
104 TfLiteTensor tensors[tensors_size] = {CreateTensor(input_data, input_dims), in TestPadFloat() local
109 tensors[1].allocation_type = kTfLiteMmapRo; in TestPadFloat()
112 ValidatePadGoldens(tensors, tensors_size, golden, in TestPadFloat()
131 TfLiteTensor tensors[tensors_size] = { in TestPadV2Float() local
137 tensors[1].allocation_type = kTfLiteMmapRo; 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 | detection_postprocess_test.cc | 103 TfLiteTensor tensors[tensors_size]; in TestDetectionPostprocess() local
116 tensors[0] = in TestDetectionPostprocess()
119 tensors[1] = in TestDetectionPostprocess()
122 tensors[2] = in TestDetectionPostprocess()
126 tensors[0] = CreateTensor(input_data1, input_dims1); in TestDetectionPostprocess()
127 tensors[1] = CreateTensor(input_data2, input_dims2); in TestDetectionPostprocess()
128 tensors[2] = CreateTensor(input_data3, input_dims3); in TestDetectionPostprocess()
130 tensors[3] = CreateTensor(output_data1, output_dims1); in TestDetectionPostprocess()
131 tensors[4] = CreateTensor(output_data2, output_dims2); in TestDetectionPostprocess()
132 tensors[5] = CreateTensor(output_data3, output_dims3); in TestDetectionPostprocess()
[all …]
|
| D | floor_div_test.cc | 28 void ExecuteFloorDivTest(TfLiteTensor* tensors, int tensors_count) { in ExecuteFloorDivTest() argument
35 micro::KernelRunner runner(registration, tensors, tensors_count, inputs_array, in ExecuteFloorDivTest()
51 TfLiteTensor tensors[] = { in TestFloorDiv() local
56 constexpr int tensors_count = std::extent<decltype(tensors)>::value; in TestFloorDiv()
58 ExecuteFloorDivTest(tensors, tensors_count); in TestFloorDiv()
|
| D | floor_mod_test.cc | 28 void ExecuteFloorModTest(TfLiteTensor* tensors, int tensors_count) { in ExecuteFloorModTest() argument
35 micro::KernelRunner runner(registration, tensors, tensors_count, inputs_array, in ExecuteFloorModTest()
51 TfLiteTensor tensors[] = { in TestFloorMod() local
56 constexpr int tensors_count = std::extent<decltype(tensors)>::value; in TestFloorMod()
58 ExecuteFloorModTest(tensors, tensors_count); in TestFloorMod()
|
| D | fill_test.cc | 32 TfLiteTensor tensors[] = { in TestFill() local
43 tensors, in TestFill()
44 sizeof(tensors) / sizeof(TfLiteTensor), in TestFill()
64 TF_LITE_MICRO_EXPECT_EQ(tensors[output_index].type, in TestFill()
65 tensors[value_index].type); in TestFill()
|
| D | leaky_relu_test.cc | 44 TfLiteTensor* tensors) { in ExecuteLeakyReluTest() argument
54 micro::KernelRunner runner(registration, tensors, tensors_count, inputs_array, in ExecuteLeakyReluTest()
69 TfLiteTensor tensors[] = { in TestLeakyRelu() local
73 constexpr int tensors_count = std::extent<decltype(tensors)>::value; in TestLeakyRelu()
74 ExecuteLeakyReluTest(params.alpha, tensors_count, tensors); in TestLeakyRelu()
90 TfLiteTensor tensors[] = { in TestLeakyReluQuantized() local
96 constexpr int kTensorsCount = std::extent<decltype(tensors)>::value; in TestLeakyReluQuantized()
97 ExecuteLeakyReluTest(params.alpha, kTensorsCount, tensors); in TestLeakyReluQuantized()
|
| D | log_softmax_test.cc | 29 void ExecuteLogSoftmaxTest(int tensors_count, TfLiteTensor* tensors) { in ExecuteLogSoftmaxTest() argument
36 micro::KernelRunner runner(registration, tensors, tensors_count, inputs_array, in ExecuteLogSoftmaxTest()
51 TfLiteTensor tensors[] = { in TestLogSoftmax() local
55 constexpr int kTensorsCount = std::extent<decltype(tensors)>::value; in TestLogSoftmax()
56 ExecuteLogSoftmaxTest(kTensorsCount, tensors); in TestLogSoftmax()
90 TfLiteTensor tensors[] = { in TestLogSoftmaxQuantized() local
96 constexpr int kTensorsCount = std::extent<decltype(tensors)>::value; in TestLogSoftmaxQuantized()
98 ExecuteLogSoftmaxTest(kTensorsCount, tensors); in TestLogSoftmaxQuantized()
|
| 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()
61 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestPackTwoInputsFloat() local
74 ValidatePackGoldens(tensors, tensors_size, builtin_data, inputs_array, in TestPackTwoInputsFloat()
94 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestPackThreeInputsFloat() local
108 ValidatePackGoldens(tensors, tensors_size, builtin_data, inputs_array, in TestPackThreeInputsFloat()
125 TfLiteTensor tensors[tensors_size] = { in TestPackTwoInputsQuantized() local
142 ValidatePackGoldens(tensors, tensors_size, builtin_data, inputs_array, in TestPackTwoInputsQuantized()
159 TfLiteTensor tensors[tensors_size] = {CreateTensor(input1_data, input1_dims), in TestPackTwoInputsQuantized32() local
172 ValidatePackGoldens(tensors, tensors_size, builtin_data, inputs_array, in TestPackTwoInputsQuantized32()
|
| D | reshape_test.cc | 34 void ValidateReshapeGoldens(TfLiteTensor* tensors, int tensors_size, in ValidateReshapeGoldens() argument
43 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in ValidateReshapeGoldens()
55 TfLiteTensor* output_tensor = &tensors[outputs_array->data[0]]; in ValidateReshapeGoldens()
75 TfLiteTensor tensors[tensors_size]; in TestReshapeWithShape() local
76 tensors[0] = *input_tensor; in TestReshapeWithShape()
77 tensors[1] = *shape_tensor; in TestReshapeWithShape()
78 tensors[2] = *output_tensor; in TestReshapeWithShape()
85 ValidateReshapeGoldens(tensors, tensors_size, inputs_array, outputs_array, in TestReshapeWithShape()
101 TfLiteTensor tensors[tensors_size]; in TestReshapeWithoutShape() local
102 tensors[0] = *input_tensor; in TestReshapeWithoutShape()
[all …]
|
| D | activations_test.cc | 37 TfLiteTensor tensors[tensors_size] = { in TestReluFloat() local
48 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestReluFloat()
70 TfLiteTensor tensors[tensors_size] = { in TestRelu6Float() local
81 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestRelu6Float()
105 TfLiteTensor tensors[tensors_size] = { in TestReluInt8() local
118 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestReluInt8()
145 TfLiteTensor tensors[tensors_size] = { in TestRelu6Int8() local
158 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestRelu6Int8()
|
| 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()
59 TfLiteTensor tensors[tensors_size] = { in TestPreluFloat() local
65 ValidatePreluGoldens(tensors, tensors_size, expected_output_data, in TestPreluFloat()
87 TfLiteTensor tensors[tensors_size] = { in TestPreluQuantized() local
99 ValidatePreluGoldens(tensors, tensors_size, golden_quantized, in TestPreluQuantized()
|
| D | logistic_test.cc | 50 void ValidateLogisticGoldens(TfLiteTensor* tensors, const int tensor_count, in ValidateLogisticGoldens() argument
59 micro::KernelRunner runner(registration, tensors, tensor_count, inputs_array, in ValidateLogisticGoldens()
80 TfLiteTensor tensors[tensors_size] = { in TestLogisticFloat() local
85 ValidateLogisticGoldens(tensors, tensors_size, output_data, golden, in TestLogisticFloat()
103 TfLiteTensor tensors[tensors_size] = { in TestLogisticQuantized() local
112 ValidateLogisticGoldens(tensors, tensors_size, output_data, golden_quantized, in TestLogisticQuantized()
|
| D | space_to_batch_nd_test.cc | 41 TfLiteStatus ValidateSpaceToBatchNdGoldens(TfLiteTensor* tensors, in ValidateSpaceToBatchNdGoldens() argument
50 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in ValidateSpaceToBatchNdGoldens()
79 TfLiteTensor tensors[tensors_size] = { in TestSpaceToBatchNdFloat() local
86 return ValidateSpaceToBatchNdGoldens(tensors, tensors_size, golden, in TestSpaceToBatchNdFloat()
106 TfLiteTensor tensors[tensors_size] = { in TestSpaceToBatchNdQuantized() local
118 return ValidateSpaceToBatchNdGoldens(tensors, tensors_size, golden_quantized, in TestSpaceToBatchNdQuantized()
|
| D | batch_to_space_nd_test.cc | 41 TfLiteStatus ValidateBatchToSpaceNdGoldens(TfLiteTensor* tensors, in ValidateBatchToSpaceNdGoldens() argument
50 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in ValidateBatchToSpaceNdGoldens()
79 TfLiteTensor tensors[tensors_size] = { in TestBatchToSpaceNdFloat() local
86 return ValidateBatchToSpaceNdGoldens(tensors, tensors_size, golden, in TestBatchToSpaceNdFloat()
106 TfLiteTensor tensors[tensors_size] = { in TestBatchToSpaceNdQuantized() local
118 return ValidateBatchToSpaceNdGoldens(tensors, tensors_size, golden_quantized, in TestBatchToSpaceNdQuantized()
|
| 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()
81 TfLiteStatus ValidateTransposeConvGoldens(TfLiteTensor* tensors, in ValidateTransposeConvGoldens() argument
88 tensors, tensors_size, output_length, conv_params, output_data); in ValidateTransposeConvGoldens()
117 TfLiteTensor tensors[tensors_size] = { in TestTransposeConvFloat() local
125 return ValidateTransposeConvGoldens(tensors, tensors_size, in TestTransposeConvFloat()
161 TfLiteTensor tensors[tensors_size] = { in TestTransposeConvQuantized() local
171 tensors, tensors_size, expected_output_quantized, output_dims_count, in TestTransposeConvQuantized()
207 TfLiteTensor tensors[tensors_size] = { in TestTransposeConvQuantized() local
219 tensors, tensors_size, expected_output_quantized, output_dims_count, in TestTransposeConvQuantized()
[all …]
|
| D | squeeze_test.cc | 46 TfLiteTensor tensors[tensors_size]; in TestSqueezeOp() local
47 tensors[0] = CreateTensor(input_data, input_dims1); in TestSqueezeOp()
48 tensors[1] = CreateTensor(output_data, output_dims1); in TestSqueezeOp()
56 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestSqueezeOp()
|
| D | cast_test.cc | 36 TfLiteTensor tensors[tensors_size] = { in TestCastFloatToInt8() local
47 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestCastFloatToInt8()
68 TfLiteTensor tensors[tensors_size] = { in TestCastInt8ToFloat() local
79 micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array, in TestCastInt8ToFloat()
|
| D | depth_to_space_test.cc | 36 TfLiteTensor* tensors, int tensors_count) { in ExecuteDepthToSpaceTest() argument
46 micro::KernelRunner runner(registration, tensors, tensors_count, inputs_array, in ExecuteDepthToSpaceTest()
62 TfLiteTensor tensors[] = { in TestDepthToSpace() local
66 constexpr int tensors_count = std::extent<decltype(tensors)>::value; in TestDepthToSpace()
67 ExecuteDepthToSpaceTest(params, tensors, tensors_count); in TestDepthToSpace()
76 TF_LITE_MICRO_EXPECT_EQ(expected_dims->data[i], tensors[1].dims->data[i]); in TestDepthToSpace()
113 TfLiteTensor tensors[] = { in TestDepthToSpaceQuantized() local
119 constexpr int kTensorsCount = std::extent<decltype(tensors)>::value; in TestDepthToSpaceQuantized()
121 ExecuteDepthToSpaceTest(params, tensors, kTensorsCount); in TestDepthToSpaceQuantized()
133 TF_LITE_MICRO_EXPECT_EQ(expected_dims->data[i], tensors[1].dims->data[i]); in TestDepthToSpaceQuantized()
|
| D | shape_test.cc | 27 void ValidateShape(TfLiteTensor* tensors, const int tensor_count, in ValidateShape() argument
36 micro::KernelRunner runner(registration, tensors, tensor_count, inputs_array, in ValidateShape()
57 TfLiteTensor tensors[tensors_size] = { in TestShape() local
62 ValidateShape(tensors, tensors_size, output_data, expected_output_data, in TestShape()
|
| /tflite-micro-3.4.0-2.7.6/tensorflow/lite/micro/ |
| D | micro_allocator_test.cc | 95 &subgraph_allocations[subgraph_idx].tensors[tensor_idx]); in VerifyMockTensor()
107 &subgraph_allocations[subgraph_idx].tensors[tensor_idx]); in VerifyMockWeightTensor()
309 TfLiteEvalTensor* eval_tensors = subgraph_allocations[0].tensors; in TF_LITE_MICRO_TEST()
378 uint8_t* start = subgraph_allocations[0].tensors[0].data.uint8; in TF_LITE_MICRO_TEST()
382 0, subgraph_allocations[0].tensors[0].data.uint8 - start); in TF_LITE_MICRO_TEST()
387 &subgraph_allocations[0].tensors[0], &buffer_size)); in TF_LITE_MICRO_TEST()
391 96, subgraph_allocations[0].tensors[1].data.uint8 - start); in TF_LITE_MICRO_TEST()
395 48, subgraph_allocations[0].tensors[2].data.uint8 - start); in TF_LITE_MICRO_TEST()
398 0, subgraph_allocations[0].tensors[3].data.uint8 - start); in TF_LITE_MICRO_TEST()
445 model, subgraph_allocations[0].tensors, 1); in TF_LITE_MICRO_TEST()
[all …]
|
