diff --git a/mindspore/ccsrc/backend/session/gpu_session.cc b/mindspore/ccsrc/backend/session/gpu_session.cc index d4475da1c36c8aaf5e63ef18ac17d1fe9a1fbb2f..037780a8c807144dadda54610898af18396a2e71 100644 --- a/mindspore/ccsrc/backend/session/gpu_session.cc +++ b/mindspore/ccsrc/backend/session/gpu_session.cc @@ -133,7 +133,11 @@ void GPUSession::LoadInputData(const std::shared_ptr<KernelGraph> &kernel_graph, const std::vector<tensor::TensorPtr> &inputs_const) const { std::vector<tensor::TensorPtr> inputs(inputs_const); MS_EXCEPTION_IF_NULL(kernel_graph); - auto input_nodes = kernel_graph->inputs(); + std::vector<AnfNodePtr> input_nodes; + for (const auto &input_node : kernel_graph->inputs()) { + auto params = AnfAlgo::GetAllOutput(input_node); + std::copy(params.begin(), params.end(), std::back_inserter(input_nodes)); + } auto ms_context = MsContext::GetInstance(); MS_EXCEPTION_IF_NULL(ms_context); if (inputs.size() != input_nodes.size()) {