算子范围
KAIL_DNN目前支持的算子如表1所示。
算子名称 |
算子说明 |
---|---|
Eltwise |
逐元素操作算子。 |
Layer Normalization |
层归一化算子。 |
Inner Product |
矩阵内积算子。 |
Softmax |
Softmax归一化算子。 |
Sum |
求和算子。 |
Matmul |
矩阵乘法算子。 |
Convolution |
卷积算子。 |
Deconvolution |
计算反卷积操作。 |
Concat |
拼接算子。 |
Resampling |
数据重新采样放缩算子。 |
Shuffle |
数据混洗算子。 |
Reorder |
数据重新排布算子。 |
Pool |
池化操作算子。 |
Batch Normalization(bnormal) |
批次归一化算子。 |
Local Response Normalization(lrn) |
局部响应归一化算子。 |
Reduction |
推断算子。 |
PReLU |
使用训练alpha的激活算子(Leaky ReLU)。 |
Binary |
二进制基元计算张量算子。 |
RNN |
循环神经网络算子。 |
父主题: KAIL_DNN算子说明