Browse Source

fix infer registry bug

tags/v1.2.0-rc1
liuyu 5 years ago
parent
commit
9914f37eae
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/lite/nnacl/infer/infer_register.h

+ 1
- 1
mindspore/lite/nnacl/infer/infer_register.h View File

@@ -214,7 +214,7 @@ enum PrimType {
PrimType_ResizeGrad = 187,
PrimType_Splice = 188,
PrimType_MIN = PrimType_NONE,
PrimType_MAX = PrimType_Splice
PrimType_MAX = PrimType_Splice + 1
};

void RegInfer(int prim_type, InferShape func);


Loading…
Cancel
Save