opencl convolution support NC4HW4

5 years ago · 4140b13ce1
parent 0328245909
commit 4140b13ce1
3 changed files with 485 additions and 177 deletions
--- a/mindspore/lite/src/runtime/kernel/opencl/kernel/convolution.cc
+++ b/mindspore/lite/src/runtime/kernel/opencl/kernel/convolution.cc
--- a/mindspore/lite/src/runtime/kernel/opencl/kernel/convolution.h
+++ b/mindspore/lite/src/runtime/kernel/opencl/kernel/convolution.h
@ -64,7 +64,9 @@ class ConvolutionOpenCLKernel : public OpenCLKernel {
  cl::Kernel kernel_conv;
  cl::Kernel kernel_36to4x4;

-  std::string CodeGenConvolution();
+  std::string CodeGenConvolutionNHWC4();
+  std::string CodeGenConvolutionNC4HW4();
+
  std::string CodeGenWinograd4x4To36();
  std::string CodeGenWinogradConvolution();
  std::string CodeGenWinograd36To4x4();
--- a/mindspore/lite/test/ut/src/runtime/kernel/opencl/convolution_tests.cc
+++ b/mindspore/lite/test/ut/src/runtime/kernel/opencl/convolution_tests.cc