内置的算子类型列表
您可以在注册算子类型的头文件中查看内置的算子类型,注册各算子类型时设置的算子输入、输出、属性等信息。
算子名称 |
所在头文件 |
---|---|
Data |
array_defs.h |
Concat |
|
Flatten |
|
Reshape |
|
Split |
|
Const |
const_defs.h |
Permute |
detection_defs.h |
Add |
math_defs.h |
Mul |
|
Activation |
nn_defs.h |
BatchNorm |
|
Convolution |
|
Eltwise |
|
LRN |
|
ConvolutionDepthwise |
|
FullConnection |
|
Pooling |
|
Scale |
|
ShuffleChannel |
|
Softmax |