Searched refs:ConvolutionOpModel (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | conv_test.cc | 112 class ConvolutionOpModel : public BaseConvolutionOpModel { class
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()
325 ConvolutionOpModel m( in TEST_P()
391 ConvolutionOpModel m( in TEST_P()
445 ConvolutionOpModel m( in TEST_P()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate_test.cc | 263 class ConvolutionOpModel : public SingleOpModelWithNNAPI { class
265 ConvolutionOpModel( in ConvolutionOpModel() function in tflite::__anon0c68fcb70111::ConvolutionOpModel
349 ConvolutionOpModel m({TensorType_UINT8, {2, 2, 4, 1}, -63.5, 64}, in TEST()
387 ConvolutionOpModel m({TensorType_FLOAT32, {2, 2, 4, 1}}, in TEST()
418 ConvolutionOpModel m({TensorType_FLOAT32, {2, 2, 4, 1}}, in TEST()
|