diff options
author | 2018-02-02 13:49:35 -0800 | |
---|---|---|
committer | 2018-02-02 13:53:43 -0800 | |
commit | 1a92f45677ee66af24f2219c6b1cbaeee87056b7 (patch) | |
tree | 1d983d8f089bda2676d3dd80e78725e860c1212f /tensorflow/contrib/lite/kernels/conv_test.cc | |
parent | cf330efa207d635efecfb1b703ee65ccfbc3f98c (diff) |
TFLite: Conv CBLAS kernel
PiperOrigin-RevId: 184328848
Diffstat (limited to 'tensorflow/contrib/lite/kernels/conv_test.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/conv_test.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/conv_test.cc b/tensorflow/contrib/lite/kernels/conv_test.cc index 7550f7cc0d..d2393c3c97 100644 --- a/tensorflow/contrib/lite/kernels/conv_test.cc +++ b/tensorflow/contrib/lite/kernels/conv_test.cc @@ -29,6 +29,7 @@ namespace builtin { TfLiteRegistration* Register_CONVOLUTION_REF(); TfLiteRegistration* Register_CONVOLUTION_GENERIC_OPT(); TfLiteRegistration* Register_CONVOLUTION_MULTITHREADED_OPT(); +TfLiteRegistration* Register_CONVOLUTION_CBLAS_OPT(); } // namespace builtin } // namespace ops @@ -105,6 +106,7 @@ const auto kKernelMap = new std::map<string, TfLiteRegistration*>({ {"GenericOptimized", ops::builtin::Register_CONVOLUTION_GENERIC_OPT()}, {"MultithreadedOptimized", ops::builtin::Register_CONVOLUTION_MULTITHREADED_OPT()}, + {"CblasOptimized", ops::builtin::Register_CONVOLUTION_CBLAS_OPT()}, }); class ConvolutionOpTest : public SingleOpTest { |