Fix when use_nnpack and isDeconv == true.

74f8ede9 · hedaoyuan · 953f8ddf · 74f8ede9
隐藏空白更改
内联并排

Showing with 1 addition and 2 deletion

paddle/gserver/layers/ExpandConvLayer.cpp paddle/gserver/layers/ExpandConvLayer.cpp +1 -2

未找到文件。
--- a/paddle/gserver/layers/ExpandConvLayer.cpp
+++ b/paddle/gserver/layers/ExpandConvLayer.cpp
@@ -57,8 +57,7 @@ bool ExpandConvLayer::init(const LayerMap &layerMap,
      convGradFilterType = "GemmConvGradFilter";
    }

-    if (FLAGS_use_nnpack) {
-      CHECK_EQ(isDeconv_, false);
+    if (FLAGS_use_nnpack && !isDeconv_) {
      createFunction(forward_,
                     "NNPACKConv",
                     FuncConfig()