| 748 | } |
| 749 | |
| 750 | TfLiteRegistration* Register_CONV_2D() { |
| 751 | #if defined TFLITE_USE_APPLE_ACCELERATE_FOR_CONV |
| 752 | return Register_CONVOLUTION_CBLAS_OPT(); |
| 753 | #elif defined TFLITE_WITH_RUY |
| 754 | // tflite_with_ruy optimizes the generic kernel type. |
| 755 | return Register_CONVOLUTION_GENERIC_OPT(); |
| 756 | #else |
| 757 | return Register_CONVOLUTION_MULTITHREADED_OPT(); |
| 758 | #endif |
| 759 | } |
| 760 | |
| 761 | } // namespace builtin |
| 762 | } // namespace ops |
no test coverage detected