| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | zeros_like_test.cc | 27 void TestZerosLikeFloat(const int* input_dims_data, const float* input_data, in TestZerosLikeFloat() argument
29 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestZerosLikeFloat()
30 TfLiteIntArray* output_dims = IntArrayFromInts(input_dims_data); in TestZerosLikeFloat()
58 void TestZerosLikeInt32(const int* input_dims_data, const int32_t* input_data, in TestZerosLikeInt32() argument
61 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestZerosLikeInt32()
62 TfLiteIntArray* output_dims = IntArrayFromInts(input_dims_data); in TestZerosLikeInt32()
90 void TestZerosLikeInt64(const int* input_dims_data, const int64_t* input_data, in TestZerosLikeInt64() argument
93 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestZerosLikeInt64()
94 TfLiteIntArray* output_dims = IntArrayFromInts(input_dims_data); in TestZerosLikeInt64()
|
| D | cast_test.cc | 27 void TestCastFloatToInt8(const int* input_dims_data, const float* input_data, in TestCastFloatToInt8() argument
30 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestCastFloatToInt8()
31 TfLiteIntArray* output_dims = IntArrayFromInts(input_dims_data); in TestCastFloatToInt8()
59 void TestCastInt8ToFloat(const int* input_dims_data, const int8_t* input_data, in TestCastInt8ToFloat() argument
62 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestCastInt8ToFloat()
63 TfLiteIntArray* output_dims = IntArrayFromInts(input_dims_data); in TestCastInt8ToFloat()
|
| D | activations_test.cc | 27 void TestReluFloat(const int* input_dims_data, const float* input_data, in TestReluFloat() argument
30 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestReluFloat()
60 void TestRelu6Float(const int* input_dims_data, const float* input_data, in TestRelu6Float() argument
63 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestRelu6Float()
93 void TestReluUint8(const int* input_dims_data, const float* input_data, in TestReluUint8() argument
99 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestReluUint8()
134 void TestRelu6Uint8(const int* input_dims_data, const float* input_data, in TestRelu6Uint8() argument
140 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestRelu6Uint8()
175 void TestReluInt8(const int* input_dims_data, const float* input_data, in TestReluInt8() argument
181 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestReluInt8()
[all …]
|
| D | round_test.cc | 27 void TestRound(const int* input_dims_data, const float* input_data, in TestRound() argument
29 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestRound()
30 TfLiteIntArray* output_dims = IntArrayFromInts(input_dims_data); in TestRound()
|
| D | ceil_test.cc | 27 void TestCeil(const int* input_dims_data, const float* input_data, in TestCeil() argument
29 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestCeil()
30 TfLiteIntArray* output_dims = IntArrayFromInts(input_dims_data); in TestCeil()
|
| D | exp_test.cc | 28 void TestExp(const int* input_dims_data, const float* input_data, in TestExp() argument
30 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestExp()
31 TfLiteIntArray* output_dims = IntArrayFromInts(input_dims_data); in TestExp()
|
| D | dequantize_test.cc | 50 void TestDequantizeToFloat(const int* input_dims_data, const float* input_data, in TestDequantizeToFloat() argument
55 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestDequantizeToFloat()
72 void TestDequantizeToInt32(const int* input_dims_data, const float* input_data, in TestDequantizeToInt32() argument
78 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestDequantizeToInt32()
|
| D | unpack_test.cc | 27 const int* input_dims_data, const float* input_data, int axis, in TestUnpackThreeOutputsFloat() argument
32 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestUnpackThreeOutputsFloat()
93 void TestUnpackOneOutputFloat(const int* input_dims_data, in TestUnpackOneOutputFloat() argument
98 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestUnpackOneOutputFloat()
137 const int* input_dims_data, const uint8_t* input_data, int axis, in TestUnpackThreeOutputsQuantized() argument
142 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestUnpackThreeOutputsQuantized()
207 const int* input_dims_data, const int32_t* input_data, int axis, in TestUnpackThreeOutputsQuantized32() argument
212 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestUnpackThreeOutputsQuantized32()
|
| D | reshape_test.cc | 115 void TestReshape(const int* input_dims_data, const float* input_data, in TestReshape() argument
121 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestReshape()
134 void TestReshapeQuantized(const int* input_dims_data, const T* input_data, in TestReshapeQuantized() argument
142 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestReshapeQuantized()
212 int input_dims_data[] = {3, 1, 2, 2}; in TF_LITE_MICRO_TEST() local
214 tflite::testing::IntArrayFromInts(input_dims_data); in TF_LITE_MICRO_TEST()
332 int input_dims_data[] = {1, 1}; in TF_LITE_MICRO_TEST() local
333 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TF_LITE_MICRO_TEST()
|
| D | pad_test.cc | 92 void TestPadFloat(const int* input_dims_data, const float* input_data, in TestPadFloat() argument
97 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestPadFloat()
117 void TestPadV2Float(const int* input_dims_data, const float* input_data, in TestPadV2Float() argument
122 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestPadV2Float()
145 void TestPadQuantized(const int* input_dims_data, const float* input_data, in TestPadQuantized() argument
152 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestPadQuantized()
178 void TestPadV2Quantized(const int* input_dims_data, const float* input_data, in TestPadV2Quantized() argument
188 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestPadV2Quantized()
|
| D | conv_test_common.cc | 114 TfLiteStatus TestConvFloat(const int* input_dims_data, const float* input_data, in TestConvFloat() argument
122 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestConvFloat()
143 const int* input_dims_data, const float* input_data, in TestConvQuantizedPerLayer() argument
151 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestConvQuantizedPerLayer()
185 const int* input_dims_data, const float* input_data, in TestConvQuantizedPerChannel() argument
194 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestConvQuantizedPerChannel()
|
| D | conv_test.h | 62 TfLiteStatus TestConvFloat(const int* input_dims_data, const float* input_data,
71 const int* input_dims_data, const float* input_data,
81 const int* input_dims_data, const float* input_data,
|
| D | tanh_test.cc | 68 void TestTanhFloat(const int input_dims_data[], const float* input_data, in TestTanhFloat() argument
72 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestTanhFloat()
103 void TestTanhQuantized(const int input_dims_data[], const float* input_data, in TestTanhQuantized() argument
111 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestTanhQuantized()
|
| D | prelu_test.cc | 48 void TestPreluFloat(const int* input_dims_data, const float* input_data, in TestPreluFloat() argument
52 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestPreluFloat()
72 void TestPreluQuantized(const int* input_dims_data, const float* input_data, in TestPreluQuantized() argument
80 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestPreluQuantized()
|
| D | logistic_test.cc | 71 void TestLogisticFloat(const int* input_dims_data, const float* input_data, in TestLogisticFloat() argument
74 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestLogisticFloat()
91 void TestLogisticQuantized(const int* input_dims_data, const float* input_data, in TestLogisticQuantized() argument
97 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestLogisticQuantized()
|
| D | batch_to_space_nd_test.cc | 67 const int* input_dims_data, const float* input_data, in TestBatchToSpaceNdFloat() argument
71 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestBatchToSpaceNdFloat()
92 const int* input_dims_data, const float* input_data, T* input_quantized, in TestBatchToSpaceNdQuantized() argument
98 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestBatchToSpaceNdQuantized()
|
| D | space_to_batch_nd_test.cc | 67 const int* input_dims_data, const float* input_data, in TestSpaceToBatchNdFloat() argument
71 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestSpaceToBatchNdFloat()
92 const int* input_dims_data, const float* input_data, T* input_quantized, in TestSpaceToBatchNdQuantized() argument
98 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestSpaceToBatchNdQuantized()
|
| D | elementwise_test.cc | 27 const int* input_dims_data, const float* input_data, in TestElementwiseFloat() argument
31 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestElementwiseFloat()
64 const int* input_dims_data, const bool* input_data, in TestElementwiseBool() argument
67 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestElementwiseBool()
|
| D | split_test.cc | 28 const int* input_dims_data, const float* input_data, in TestSplitTwoOutputsFloat() argument
33 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestSplitTwoOutputsFloat()
82 const int* input_dims_data, const float* input_data, in TestSplitFourOutputsFloat() argument
90 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestSplitFourOutputsFloat()
156 const int* input_dims_data, const uint8_t* input_data, in TestSplitTwoOutputsQuantized() argument
161 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestSplitTwoOutputsQuantized()
212 const int* input_dims_data, const int32_t* input_data, in TestSplitTwoOutputsQuantized32() argument
217 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestSplitTwoOutputsQuantized32()
|
| D | hard_swish_test.cc | 64 const int input_dims_data[] = {2, 1, size}; in TestHardSwishQuantized() local
78 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestHardSwishQuantized()
163 const int input_dims_data[] = {2, 1, size}; in TestHardSwishQuantizedBias() local
166 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestHardSwishQuantizedBias()
217 const int input_dims_data[] = {1, size}; in TestHardSwishFloat() local
220 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestHardSwishFloat()
|
| D | reduce_test.cc | 92 void TestMeanFloatInput4D(const int* input_dims_data, const float* input_data, in TestMeanFloatInput4D() argument
97 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestMeanFloatInput4D()
120 void TestReduceOpFloat(const int* input_dims_data, const float* input_data, in TestReduceOpFloat() argument
126 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestReduceOpFloat()
149 const int* input_dims_data, const float* input_data, T* input_data_quant, in TestReduceOpQuantized() argument
156 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestReduceOpQuantized()
185 void TestMeanOpQuantized(const int* input_dims_data, const float* input_data, in TestMeanOpQuantized() argument
194 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestMeanOpQuantized()
|
| D | floor_test.cc | 27 void TestFloor(const int* input_dims_data, const float* input_data, in TestFloor() argument
30 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestFloor()
|
| D | neg_test.cc | 27 void TestNegFloat(const int* input_dims_data, const float* input_data, in TestNegFloat() argument
30 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestNegFloat()
|
| D | arg_min_max_test.cc | 50 void TestArgMinMaxFloat(const int* input_dims_data, const float* input_values, in TestArgMinMaxFloat() argument
54 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestArgMinMaxFloat()
73 void TestArgMinMaxQuantized(const int* input_dims_data, in TestArgMinMaxQuantized() argument
80 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestArgMinMaxQuantized()
|
| D | pooling_test.cc | 61 void TestAveragePoolFloat(const int* input_dims_data, const float* input_data, in TestAveragePoolFloat() argument
68 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestAveragePoolFloat()
91 const int* input_dims_data, const T* input_data, const float input_scale, in TestAveragePoolQuantized() argument
99 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestAveragePoolQuantized()
121 void TestMaxPoolFloat(const int* input_dims_data, const float* input_data, in TestMaxPoolFloat() argument
126 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestMaxPoolFloat()
147 void TestMaxPoolQuantized(const int* input_dims_data, const T* input_data, in TestMaxPoolQuantized() argument
155 TfLiteIntArray* input_dims = IntArrayFromInts(input_dims_data); in TestMaxPoolQuantized()
|