From 2e2008406df96bf89d028eb6922389552abce749 Mon Sep 17 00:00:00 2001
From: Travis CI ProtoMaker
to describe the Operator’s input, output, and additional comments:
class MulOpMaker : public framework::OpProtoAndCheckerMaker {
public:
- MulOpMaker(framework::OpProto *proto, framework::OpAttrChecker *op_checker)
+ MulOpMaker(OpProto *proto, OpAttrChecker *op_checker)
: OpProtoAndCheckerMaker(proto, op_checker) {
AddInput("X", "(Tensor), 2D tensor of size (M x K)");
AddInput("Y", "(Tensor), 2D tensor of size (K x N)");
@@ -279,7 +279,7 @@ Registering the Op | Ops are registered in template <typename AttrType>
class ScaleOpMaker : public framework::OpProtoAndCheckerMaker {
public:
- ScaleOpMaker(framework::OpProto *proto, framework::OpAttrChecker *op_checker)
+ ScaleOpMaker(OpProto *proto, OpAttrChecker *op_checker)
: OpProtoAndCheckerMaker(proto, op_checker) {
AddInput("X", "The input tensor of scale operator.").NotInGradient();
AddOutput("Out", "The output tensor of scale operator.").NotInGradient();
diff --git a/develop/doc_cn/_sources/howto/dev/new_op_cn.md.txt b/develop/doc_cn/_sources/howto/dev/new_op_cn.md.txt
index 757a5840bc..3109d72001 100644
--- a/develop/doc_cn/_sources/howto/dev/new_op_cn.md.txt
+++ b/develop/doc_cn/_sources/howto/dev/new_op_cn.md.txt
@@ -53,7 +53,7 @@ Kernel实现 | CPU、CUDA共享Kernel实现在`.h`文件中,否则,CPU
```cpp
class MulOpMaker : public framework::OpProtoAndCheckerMaker {
public:
- MulOpMaker(framework::OpProto *proto, framework::OpAttrChecker *op_checker)
+ MulOpMaker(OpProto *proto, OpAttrChecker *op_checker)
: OpProtoAndCheckerMaker(proto, op_checker) {
AddInput("X", "(Tensor), 2D tensor of size (M x K)");
AddInput("Y", "(Tensor), 2D tensor of size (K x N)");
@@ -82,7 +82,7 @@ The equation is: Out = X * Y
template
class ScaleOpMaker : public framework::OpProtoAndCheckerMaker {
public:
- ScaleOpMaker(framework::OpProto *proto, framework::OpAttrChecker *op_checker)
+ ScaleOpMaker(OpProto *proto, OpAttrChecker *op_checker)
: OpProtoAndCheckerMaker(proto, op_checker) {
AddInput("X", "The input tensor of scale operator.").NotInGradient();
AddOutput("Out", "The output tensor of scale operator.").NotInGradient();
diff --git a/develop/doc_cn/howto/dev/new_op_cn.html b/develop/doc_cn/howto/dev/new_op_cn.html
index db2e35ee2d..42a71c04c1 100644
--- a/develop/doc_cn/howto/dev/new_op_cn.html
+++ b/develop/doc_cn/howto/dev/new_op_cn.html
@@ -269,7 +269,7 @@ Kernel实现 | CPU、CUDA共享Kernel实现在ProtoMaker
来描述该Op的输入、输出,并添加注释:
class MulOpMaker : public framework::OpProtoAndCheckerMaker {
public:
- MulOpMaker(framework::OpProto *proto, framework::OpAttrChecker *op_checker)
+ MulOpMaker(OpProto *proto, OpAttrChecker *op_checker)
: OpProtoAndCheckerMaker(proto, op_checker) {
AddInput("X", "(Tensor), 2D tensor of size (M x K)");
AddInput("Y", "(Tensor), 2D tensor of size (K x N)");
@@ -293,7 +293,7 @@ Kernel实现 | CPU、CUDA共享Kernel实现在template <typename AttrType>
class ScaleOpMaker : public framework::OpProtoAndCheckerMaker {
public:
- ScaleOpMaker(framework::OpProto *proto, framework::OpAttrChecker *op_checker)
+ ScaleOpMaker(OpProto *proto, OpAttrChecker *op_checker)
: OpProtoAndCheckerMaker(proto, op_checker) {
AddInput("X", "The input tensor of scale operator.").NotInGradient();
AddOutput("Out", "The output tensor of scale operator.").NotInGradient();
--
GitLab