提交 1dd639eb 编写于 作者: H hedaoyuan

Bug fix.

上级 2340ceda
...@@ -33,7 +33,7 @@ class Conv2DOp : public framework::OperatorWithKernel { ...@@ -33,7 +33,7 @@ class Conv2DOp : public framework::OperatorWithKernel {
auto out = ctx.Output<Tensor>("Output"); auto out = ctx.Output<Tensor>("Output");
std::vector<int> strides = Attr<std::vector<int>>("strides"); std::vector<int> strides = Attr<std::vector<int>>("strides");
std::vector<int> paddings = Attr<std::vector<int>>("paddings"); std::vector<int> paddings = Attr<std::vector<int>>("paddings");
int groups = context.Attr<int>("groups"); int groups = Attr<int>("groups");
int input_channels = in->dims()[1]; int input_channels = in->dims()[1];
int output_channels = filter->dims()[0]; int output_channels = filter->dims()[0];
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册