p49270318
/
InfiniTensor
forked from jiuyuan/InfiniTensor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add bang conv nhwc kernel

This commit is contained in:
whjthu 2023-06-25 08:24:02 +08:00
parent fa62e35e41
commit db9d5f4ffe
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/kernels/bang/conv.cc
View File

@ -177,4 +177,6 @@ class ConvCnnl : public BangKernelWithoutConfig {
REGISTER_KERNEL(Device::BANG, OpType::Conv, DataType::Float32, ConvCnnl,
"Conv_cnnl_BANG_Float32");
REGISTER_KERNEL(Device::BANG, OpType::ConvNHWC, DataType::Float32, ConvCnnl,
"ConvNHWC_cnnl_BANG_Float32");
}; // namespace infini