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;