提交 656f775c 编写于 作者: H hedaoyuan

Fix the doc.

上级 b4ba35ca
...@@ -69,15 +69,17 @@ class Conv2DOpMaker : public framework::OpProtoAndCheckerMaker { ...@@ -69,15 +69,17 @@ class Conv2DOpMaker : public framework::OpProtoAndCheckerMaker {
"Filter", "Filter",
"The filter tensor of convolution operator." "The filter tensor of convolution operator."
"The format of the filter tensor is MCHW, where M is the number of " "The format of the filter tensor is MCHW, where M is the number of "
"output " "output image channels, C is the number of input image channels, "
"image channels, C is the number of input image channels, H and W is " "H and W is height and width of filter. "
"height and width of filter."); "If the groups attribute is greater than 1, C equal the number of "
"input image channels divided by the groups.");
AddOutput("Output", AddOutput("Output",
"The output tensor of convolution operator." "The output tensor of convolution operator."
"The format of output tensor is also NCHW."); "The format of output tensor is also NCHW.");
AddComment(R"DOC( AddComment(R"DOC(
The convolution operation calculates the output based on The convolution operation calculates the output based on the input, filter
the input, filter and strides, paddings parameters. and strides, paddings, groups parameters. The size of each dimension of the
parameters is checked in the infer-shape.
)DOC"); )DOC");
AddAttr<std::vector<int>>("strides", "strides of convolution operator."); AddAttr<std::vector<int>>("strides", "strides of convolution operator.");
AddAttr<std::vector<int>>("paddings", "paddings of convolution operator."); AddAttr<std::vector<int>>("paddings", "paddings of convolution operator.");
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册