Searched refs:op_output_map (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_compiler.h | 143 std::map<KernelWithIndex, tensor::TensorPtr> *op_output_map) const;
148 std::map<KernelWithIndex, TensorPtr> *op_output_map,
|
| D | graph_compiler.cc | 493 … std::map<KernelWithIndex, tensor::TensorPtr> *op_output_map) const { in UpdateRefCount()
495 session_->HandleOpInputs(input_kernels_with_index, ref_count, op_output_map); in UpdateRefCount()
500 std::map<KernelWithIndex, TensorPtr> *op_output_map, in RecoverGraphOutput() argument
503 session_->HandleOpOutputs(kernel, op_outputs, ref_count, op_output_map, graph_output_info); in RecoverGraphOutput()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | backend.cc | 511 … const CNodePtr &backend_cnode, const std::map<KernelWithIndex, tensor::TensorPtr> &op_output_map, in GetControlOpInput() argument
529 …TensorPtr tensor = graph_compiler->GetSingleOpInputTensorByIndex(backend_cnode, op_output_map, par… in GetControlOpInput()
601 … const CNodePtr &kernel, const std::map<KernelWithIndex, tensor::TensorPtr> &op_output_map, in RunControlOperator() argument
630 …GetControlOpInput(graph_compiler, cnode, kernel, op_output_map, parameter_index, graph_inputs, inp… in RunControlOperator()
719 std::map<KernelWithIndex, tensor::TensorPtr> op_output_map; in RunGraphBySingleOp() local
747 …graph_compiler_->GetSingleOpInputTensors(kernel, op_output_map, parameter_index, inputs[graph_inde… in RunGraphBySingleOp()
757 …RunControlOperator(graph_compiler_, graph, kernel, op_output_map, parameter_index, inputs[graph_in… in RunGraphBySingleOp()
761 … graph_compiler_->UpdateRefCount(input_tensor_info.input_kernel, &cnode_ref_count, &op_output_map); in RunGraphBySingleOp()
764 …graph_compiler_->RecoverGraphOutput(kernel, op_outputs, cnode_ref_count, &op_output_map, &graph_ou… in RunGraphBySingleOp()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | session_basic.cc | 1304 std::map<KernelWithIndex, tensor::TensorPtr> *op_output_map) { in HandleOpInputs() argument
1306 MS_EXCEPTION_IF_NULL(op_output_map); in HandleOpInputs()
1323 auto output_iter = op_output_map->find(kernel_with_index); in HandleOpInputs()
1324 if (output_iter == op_output_map->end()) { in HandleOpInputs()
1328 op_output_map->erase(output_iter); in HandleOpInputs()
1334 std::map<KernelWithIndex, tensor::TensorPtr> *op_output_map, in HandleOpOutputs() argument
1337 MS_EXCEPTION_IF_NULL(op_output_map); in HandleOpOutputs()
1348 (*op_output_map)[kernel_with_index] = output_tensor; in HandleOpOutputs()
2290 std::map<KernelWithIndex, tensor::TensorPtr> op_output_map; in RunOpsInGraphImpl() local
2294 GetOpInputTensors(kernel, op_output_map, parameter_index, inputs, &input_tensor_info); in RunOpsInGraphImpl()
[all …]
|
| D | session_basic.h | 193 std::map<KernelWithIndex, tensor::TensorPtr> *op_output_map);
197 std::map<KernelWithIndex, tensor::TensorPtr> *op_output_map,
|