fix topi c++ conv2d_nchw lambda expr issue (apache#3570)

neo-ai · Sep 6, 2019 · 4e198a8 · 4e198a8
1 parent fe66e75
commit 4e198a8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/topi/include/topi/nn.h b/topi/include/topi/nn.h
@@ -268,7 +268,7 @@ inline tvm::Tensor conv2d_nchw(const tvm::Tensor& I,
                : pad(I, {tvm::Expr(0), tvm::Expr(0), pad_h, pad_w});
   auto l = [&](tvm::Var b, tvm::Var o, tvm::Var h, tvm::Var w) {
     return tvm::sum(
-        T(b, i, stride_h * h + kh, stride_w * w + kw) * W(i, o, kh, kw),
+        T(b, i, stride_h * h + kh, stride_w * w + kw) * W(o, i, kh, kw),
         {i, kh, kw});
   };
   return tvm::compute(output_shape, l, name, tag);