From 9914f37eae47b7576fdf2514afd74e556b4f9e72 Mon Sep 17 00:00:00 2001 From: liuyu Date: Thu, 18 Mar 2021 16:11:56 +0800 Subject: [PATCH] fix infer registry bug --- mindspore/lite/nnacl/infer/infer_register.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/lite/nnacl/infer/infer_register.h b/mindspore/lite/nnacl/infer/infer_register.h index ad5b3ebaf4..4e18c2a6b2 100644 --- a/mindspore/lite/nnacl/infer/infer_register.h +++ b/mindspore/lite/nnacl/infer/infer_register.h @@ -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);