diff options
author | Yu-Cheng Ling <ycling@google.com> | 2018-02-07 13:39:01 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-02-07 13:42:36 -0800 |
commit | d1eeceb562d6defc4b517ad83cea56a894ff4c98 (patch) | |
tree | bfa0bbf37245c3e2bd9e9e5ba818ce633ab7ab75 /tensorflow/contrib/lite/kernels/conv.cc | |
parent | 528d0c5e4d148655b797368fd55fe6304730fece (diff) |
Makefile flag to use Apple Accelerate for Conv on iOS.
PiperOrigin-RevId: 184888096
Diffstat (limited to 'tensorflow/contrib/lite/kernels/conv.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/conv.cc | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/tensorflow/contrib/lite/kernels/conv.cc b/tensorflow/contrib/lite/kernels/conv.cc index 1fba3cbbce..66d2c04bba 100644 --- a/tensorflow/contrib/lite/kernels/conv.cc +++ b/tensorflow/contrib/lite/kernels/conv.cc @@ -463,9 +463,7 @@ TfLiteRegistration* Register_CONVOLUTION_CBLAS_OPT() { } TfLiteRegistration* Register_CONV_2D() { -// TODO(ycling): Define a compilation flag and use CBLAS kernel when a -// fast CBLAS implementatino is available. -#ifdef TFLITE_USE_CBLAS_CONVOLUTION_KERNEL +#ifdef TFLITE_USE_APPLE_ACCELERATE_FOR_CONV return Register_CONVOLUTION_CBLAS_OPT(); #else return Register_CONVOLUTION_MULTITHREADED_OPT(); |