p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix 910 memory size of GetInputInfo 

Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
This commit is contained in:
zhoufeng 2020-12-10 09:15:51 +08:00
parent a2aea8cba5
commit d6fad5a0b3
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/ccsrc/cxx_api/graph/ms/ms_graph_impl.cc
View File

@ -220,7 +220,7 @@ Status MsGraphImpl::GetInputsInfo(std::vector<std::string> *names, std::vector<s
PushbackIfNotNull(names, input_names_[i]);
PushbackIfNotNull(shapes, tensor->shape());
PushbackIfNotNull(data_types, TransTypeId2InferDataType(tensor->data_type()));
PushbackIfNotNull(mem_sizes, tensor->DataSize());
PushbackIfNotNull(mem_sizes, tensor->Size());
}
return SUCCESS;
}
@ -244,7 +244,7 @@ Status MsGraphImpl::GetOutputsInfo(std::vector<std::string> *names, std::vector<
PushbackIfNotNull(names, output_names_[i]);
PushbackIfNotNull(shapes, tensor->shape());
PushbackIfNotNull(data_types, TransTypeId2InferDataType(tensor->data_type()));
PushbackIfNotNull(mem_sizes, tensor->DataSize());
PushbackIfNotNull(mem_sizes, tensor->Size());
}
return SUCCESS;