From d6fad5a0b3601d57a5e6b402e2cce33958063108 Mon Sep 17 00:00:00 2001
From: zhoufeng <zhoufeng54@huawei.com>
Date: Thu, 10 Dec 2020 09:15:51 +0800
Subject: [PATCH] fix 910 memory size of GetInputInfo

Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
---
 mindspore/ccsrc/cxx_api/graph/ms/ms_graph_impl.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/mindspore/ccsrc/cxx_api/graph/ms/ms_graph_impl.cc b/mindspore/ccsrc/cxx_api/graph/ms/ms_graph_impl.cc
index f97cf94b6f..bd391a23f5 100644
--- a/mindspore/ccsrc/cxx_api/graph/ms/ms_graph_impl.cc
+++ b/mindspore/ccsrc/cxx_api/graph/ms/ms_graph_impl.cc
@@ -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;