From a51d0d7c9e47174ceb30b8ecf6440c5e47b4337d Mon Sep 17 00:00:00 2001 From: zhangxiaokun Date: Wed, 6 Jan 2021 14:39:46 +0800 Subject: [PATCH] Fix ACL_GetInputDimsV2 --- ge/graph/load/new_model_manager/davinci_model.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ge/graph/load/new_model_manager/davinci_model.h b/ge/graph/load/new_model_manager/davinci_model.h index 4ec0ec62..970d0065 100755 --- a/ge/graph/load/new_model_manager/davinci_model.h +++ b/ge/graph/load/new_model_manager/davinci_model.h @@ -605,7 +605,7 @@ class DavinciModel { void CreateInputDimsInfo(const OpDescPtr &op_desc, Format format, ShapeDescription &shape1, ShapeDescription &shape2); - void SetInputDimsInfo(const vector &model_input_dims, Format &format, ShapeDescription &shape_info); + void SetInputDimsInfo(const vector &input_dims, Format &format, ShapeDescription &shape_info); Status GetInputDescInfo(vector &input_desc, vector &input_formats, bool by_dims) const; Status GetOutputDescInfo(vector &output_desc, vector &output_formats) const;