Browse Source

!15487 load input for graph executable

From: @kisnwang
Reviewed-by: @zhoufeng54,@chujinjin
Signed-off-by: @chujinjin
pull/15487/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
1ffcd93fa2
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