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

load input for graph executable

This commit is contained in:
kswang 2021-04-21 20:13:31 +08:00
parent 2e43434221
commit 1dce6b9448
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/backend/session/session_basic.h
View File

@ -197,6 +197,9 @@ class SessionBasic : public std::enable_shared_from_this<SessionBasic> {
void LoadInputs(const GraphId &graph_id, const std::vector<tensor::TensorPtr> &inputs_const) { void LoadInputs(const GraphId &graph_id, const std::vector<tensor::TensorPtr> &inputs_const) {
auto kernel_graph = GetGraph(graph_id); auto kernel_graph = GetGraph(graph_id);
MS_EXCEPTION_IF_NULL(kernel_graph); MS_EXCEPTION_IF_NULL(kernel_graph);
if (!kernel_graph->executable()) {
return;
}
MS_LOG(INFO) << "Load inputs"; MS_LOG(INFO) << "Load inputs";
LoadInputData(kernel_graph, inputs_const); LoadInputData(kernel_graph, inputs_const);
} }