colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix depth2space mode enum

This commit is contained in:
meseraph 2023-01-12 18:50:12 +08:00 committed by Sven
parent 9ab00a57a6
commit 20f759e58a
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
include/tim/vx/ops/depth2space.h
View File

@ -48,8 +48,8 @@ namespace ops {
class DepthToSpace : public BuiltinOp { class DepthToSpace : public BuiltinOp {
public: public:
enum depth2space_mode { enum depth2space_mode {
CRD_mode = 0, DCR_mode = 0,
DCR_mode = 1, CRD_mode = 1,
}; };
DepthToSpace(Graph* Graph, int block_size, DepthToSpace(Graph* Graph, int block_size,
DataLayout layout = DataLayout::WHCN); DataLayout layout = DataLayout::WHCN);