| /external/tensorflow/tensorflow/lite/kernels/ |
| D | conv_test.cc | 142 ConvolutionOpModel m(GetRegistration(), {TensorType_FLOAT32, {2, 2, 4, 1}}, in TEST_P()
177 ConvolutionOpModel m(GetRegistration(), {TensorType_FLOAT32, {2, 2, 4, 2}}, in TEST_P()
207 ConvolutionOpModel m(GetRegistration(), {TensorType_FLOAT32, {2, 2, 4, 1}}, in TEST_P()
231 ConvolutionOpModel m(GetRegistration(), {TensorType_FLOAT32, {2, 2, 4, 2}}, in TEST_P()
264 ConvolutionOpModel m(GetRegistration(), {TensorType_FLOAT32, {2, 2, 4, 2}}, in TEST_P()
294 ConvolutionOpModel m(GetRegistration(), {TensorType_FLOAT32, {1, 3, 6, 1}}, in TEST_P()
326 GetRegistration(), in TEST_P()
392 GetRegistration(), in TEST_P()
446 GetRegistration(), in TEST_P()
502 GetRegistration(), in TEST_P()
[all …]
|
| D | depthwise_conv_test.cc | 147 DepthwiseConvolutionOpModel m(GetRegistration(), in TEST_P()
182 GetRegistration(), in TEST_P()
235 GetRegistration(), in TEST_P()
296 GetRegistration(), {TensorType_UINT8, {1, 3, 2, 2}, -63.5, 64}, in TEST_P()
331 GetRegistration(), {TensorType_UINT8, {1, 3, 2, 2}, -63.5, 64}, in TEST_P()
334 DepthwiseConvolutionOpModel float_op(GetRegistration(), in TEST_P()
375 GetRegistration(), in TEST_P()
434 GetRegistration(), in TEST_P()
492 GetRegistration(), {TensorType_INT8, {1, 2, 3, 2}, -63.5, 64, 0.5, -1}, in TEST_P()
|
| D | transpose_conv_test.cc | 100 TransposeConvOpModel m(GetRegistration(), {TensorType_FLOAT32, {1, 3, 3, 1}}, in TEST_P()
128 TransposeConvOpModel m(GetRegistration(), {TensorType_FLOAT32, {1, 3, 3, 2}}, in TEST_P()
156 TransposeConvOpModel m(GetRegistration(), {TensorType_FLOAT32, {1, 3, 3, 2}}, in TEST_P()
185 TransposeConvOpModel m(GetRegistration(), {TensorType_FLOAT32, {1, 3, 3, 1}}, in TEST_P()
211 TransposeConvOpModel m(GetRegistration(), {TensorType_FLOAT32, {2, 3, 3, 1}}, in TEST_P()
241 TransposeConvOpModel m(GetRegistration(), {TensorType_FLOAT32, {1, 3, 3, 1}}, in TEST_P()
|
| D | fully_connected_test.cc | 373 FloatFullyConnectedOpModel m(GetRegistration(), /*units=*/3, /*batches=*/2, in TEST_P()
393 FloatFullyConnectedOpModel m(GetRegistration(), /*units=*/1, /*batches=*/2, in TEST_P()
433 GetRegistration(), /*units=*/3, /*batches*/ 2, in TEST_P()
491 GetRegistration(), /*units=*/3, /*batches*/ 2, in TEST_P()
641 GetRegistration(), input_depth, output_depth, batch, in TEST_P()
665 GetRegistration(), input_depth, output_depth, batch, in TEST_P()
732 FloatFullyConnectedOpModel m(GetRegistration(), in TEST_P()
757 GetRegistration(), /*units=*/3, /*batches=*/2, in TEST_P()
789 GetRegistration(), /*units=*/3, /*batches=*/2, in TEST_P()
827 FloatFullyConnectedOpModel m(GetRegistration(), /*units=*/16, /*batches=*/2, in TEST_P()
|
| D | test_util.h | 507 TfLiteRegistration* GetRegistration() { in GetRegistration() function
|