fix: set Deconv2d filter's opencl buffer type to CONV2D_FILTER

N/A Signed-off-by: N Luxuhui <luxuhui@xiaomi.com>

fix: set Deconv2d filter's opencl buffer type to CONV2D_FILTER
N/A Signed-off-by: N Luxuhui <luxuhui@xiaomi.com>
8c700b6a · luxuhui · 8b657bb6 · 8c700b6a
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

mace/ops/deconv_2d.cc mace/ops/deconv_2d.cc +1 -1

未找到文件。
--- a/mace/ops/deconv_2d.cc
+++ b/mace/ops/deconv_2d.cc
@@ -248,7 +248,7 @@ void RegisterDeconv2D(OpRegistry *op_registry) {
      op_registry,
      OpConditionBuilder("Deconv2D").SetInputMemoryTypeSetter(
          [](OpConditionContext *context) -> void {
-            SetFilterMemoryType(context, OpenCLBufferType::DW_CONV2D_FILTER);
+            SetFilterMemoryType(context, OpenCLBufferType::CONV2D_FILTER);
            if (context->device()->device_type() == DeviceType::GPU) {
              FrameworkType framework_type =
                  static_cast<FrameworkType>(