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

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);
}