add_depthwiseConv_op_gpu
Yancey1989 7 years ago
parent df9c13a816
commit 9d52ad4c3a

@ -24,7 +24,7 @@
- `framework::OperatorWithKernel`继承自OperatorBaseOp有计算函数称作有Kernel。
- `class OpProtoAndCheckerMaker`描述该Op的输入、输出、属性、注释,主要用于Python API接口生成
依据是否包含kernel可以将Op分为两种包含Kernel的Op和不包含kernel的Op前者Op的定义继承自`OperatorBase`,后者继承自`OperatorWithKernel`。本教程主要介绍带Kernel的Op如何写简单总结Op需要包含的内容如下
依据是否包含kernel可以将Op分为两种包含Kernel的Op和不包含kernel的Op前者Op的定义继承自`OperatorWithKernel`,后者继承自`OperatorBase`。本教程主要介绍带Kernel的Op如何写简单总结Op需要包含的内容如下
内容 | 定义位置

Loading…
Cancel
Save