更新时间:2021-03-18 GMT+08:00
分享

内置的算子类型列表

您可以在注册算子类型的头文件中查看内置的算子类型,注册各算子类型时设置的算子输入、输出、属性等信息。

表1 算子列表

算子名称

所在头文件

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

分享:

    相关文档

    相关产品