Browse Source

load input for graph executable

pull/15487/head
kswang 5 years ago
parent
commit
1dce6b9448
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/ccsrc/backend/session/session_basic.h

+ 3
- 0
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) {
auto kernel_graph = GetGraph(graph_id);
MS_EXCEPTION_IF_NULL(kernel_graph);
if (!kernel_graph->executable()) {
return;
}
MS_LOG(INFO) << "Load inputs";
LoadInputData(kernel_graph, inputs_const);
}


Loading…
Cancel
Save