fix depth2space mode enum
This commit is contained in:
parent
9ab00a57a6
commit
20f759e58a
|
|
@ -48,8 +48,8 @@ namespace ops {
|
|||
class DepthToSpace : public BuiltinOp {
|
||||
public:
|
||||
enum depth2space_mode {
|
||||
CRD_mode = 0,
|
||||
DCR_mode = 1,
|
||||
DCR_mode = 0,
|
||||
CRD_mode = 1,
|
||||
};
|
||||
DepthToSpace(Graph* Graph, int block_size,
|
||||
DataLayout layout = DataLayout::WHCN);
|
||||
|
|
|
|||
Loading…
Reference in New Issue