|
|
|
@@ -268,10 +268,10 @@ void GetOpInputTensors(const CNodePtr &cnode, const std::map<KernelWithIndex, te |
|
|
|
MS_EXCEPTION_IF_NULL(real_input); |
|
|
|
tensor::TensorPtr tensor = nullptr; |
|
|
|
if (real_input->isa<ValueNode>()) { |
|
|
|
auto value_node = input->cast<ValueNodePtr>(); |
|
|
|
auto value_node = real_input->cast<ValueNodePtr>(); |
|
|
|
MS_EXCEPTION_IF_NULL(value_node); |
|
|
|
auto value = GetValueNode(value_node); |
|
|
|
MS_EXCEPTION_IF_NULL(value_node); |
|
|
|
MS_EXCEPTION_IF_NULL(value); |
|
|
|
if (value->isa<ValueTuple>()) { |
|
|
|
auto value_tuple = value->cast<ValueTuplePtr>(); |
|
|
|
MS_EXCEPTION_IF_NULL(value_tuple); |
|
|
|
@@ -881,7 +881,7 @@ void AscendSession::RunOpImpl(const GraphInfo &graph_info, OpRunInfo *op_run_inf |
|
|
|
|
|
|
|
void AscendSession::RunOpsInGraphImpl(const GraphId &graph_id, const std::vector<tensor::TensorPtr> &inputs, |
|
|
|
VectorRef *outputs) { |
|
|
|
MS_LOG(INFO) << "Start"; |
|
|
|
MS_LOG(INFO) << "Start!"; |
|
|
|
auto kernel_graph = GetGraph(graph_id); |
|
|
|
std::map<AnfNodePtr, size_t> parameter_index; |
|
|
|
GetParameterIndex(kernel_graph.get(), inputs, ¶meter_index); |
|
|
|
@@ -910,6 +910,7 @@ void AscendSession::RunOpsInGraphImpl(const GraphId &graph_id, const std::vector |
|
|
|
HandleOpInputs(input_tensor_info.input_kernel, &cnode_ref, &op_output_map); |
|
|
|
HandleOpOutputs(kernel, op_outputs, output_indexes, cnode_ref, &op_output_map, outputs); |
|
|
|
} |
|
|
|
MS_LOG(INFO) << "Finish!"; |
|
|
|
} |
|
|
|
|
|
|
|
// compile graph steps |
|
|
|
|