Browse Source

Fix ACL_GetInputDimsV2

tags/v1.2.0
zhangxiaokun 3 years ago
parent
commit
a51d0d7c9e
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/graph/load/new_model_manager/davinci_model.h

+ 1
- 1
ge/graph/load/new_model_manager/davinci_model.h View File

@@ -605,7 +605,7 @@ class DavinciModel {


void CreateInputDimsInfo(const OpDescPtr &op_desc, Format format, ShapeDescription &shape1, ShapeDescription &shape2); void CreateInputDimsInfo(const OpDescPtr &op_desc, Format format, ShapeDescription &shape1, ShapeDescription &shape2);


void SetInputDimsInfo(const vector<int64_t> &model_input_dims, Format &format, ShapeDescription &shape_info);
void SetInputDimsInfo(const vector<int64_t> &input_dims, Format &format, ShapeDescription &shape_info);


Status GetInputDescInfo(vector<InputOutputDescInfo> &input_desc, vector<uint32_t> &input_formats, bool by_dims) const; Status GetInputDescInfo(vector<InputOutputDescInfo> &input_desc, vector<uint32_t> &input_formats, bool by_dims) const;
Status GetOutputDescInfo(vector<InputOutputDescInfo> &output_desc, vector<uint32_t> &output_formats) const; Status GetOutputDescInfo(vector<InputOutputDescInfo> &output_desc, vector<uint32_t> &output_formats) const;


Loading…
Cancel
Save