|
|
|
@ -206,7 +206,7 @@ MulOp(const std::string &type, const framework::VariableNameMap &inputs,
|
|
|
|
|
|
|
|
|
|
|
|
- `REGISTER_OP` ： 注册`ops::MulOp`类，类型名为`mul`，该类的`ProtoMaker`为`ops::MulOpMaker`，注册`ops::MulOpGrad`，类型名为`mul_grad`。
|
|
|
|
- `REGISTER_OP` ： 注册`ops::MulOp`类，类型名为`mul`，该类的`ProtoMaker`为`ops::MulOpMaker`，注册`ops::MulOpGrad`，类型名为`mul_grad`。
|
|
|
|
- `REGISTER_OP_WITHOUT_GRADIENT` ： 用于注册没有反向的Op。
|
|
|
|
- `REGISTER_OP_WITHOUT_GRADIENT` ： 用于注册没有反向的Op。
|
|
|
|
- `REGISTER_OP_CPU_KERNEL` ：注册`ops::MulKernel`类，并特化模板参数为`paddle::platform::CPUPlace`和`float`类型，同理，注册`ops::MulKernel`类。
|
|
|
|
- `REGISTER_OP_CPU_KERNEL` ：注册`ops::MulKernel`类，并特化模板参数为`paddle::platform::CPUPlace`和`float`类型，同理，注册`ops::MulGradKernel`类。
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
- 在 `.cu`文件中注册GPU Kernel。
|
|
|
|
- 在 `.cu`文件中注册GPU Kernel。
|
|
|
|
|
kexinzhao
8 years ago
committed by
Kexin Zhao