Searched defs:KernelWithIndex (Results 1 – 11 of 11) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_compiler.cc | 453 std::map<KernelWithIndex, std::vector<std::vector<size_t>>> *output_indexes) { in GetParamAndOutputIndex()
460 const std::map<KernelWithIndex, TensorPtr> &op_output, in GetSingleOpInputTensors()
469 … const std::map<KernelWithIndex, TensorPtr> &op_output, in GetSingleOpInputTensorByIndex()
486 void GraphCompiler::CalculateRefCount(const KernelGraphPtr &graph, std::map<KernelWithIndex, size_t… in CalculateRefCount()
492 std::map<KernelWithIndex, size_t> *ref_count, in UpdateRefCount()
499 const std::map<KernelWithIndex, size_t> &ref_count, in RecoverGraphOutput()
|
| /third_party/mindspore/mindspore/ccsrc/vm/ |
| D | backend.h | 46 using KernelWithIndex = session::KernelWithIndex; variable
|
| D | backend.cc | 511 … const CNodePtr &backend_cnode, const std::map<KernelWithIndex, tensor::TensorPtr> &op_output_map, in GetControlOpInput()
601 … const CNodePtr &kernel, const std::map<KernelWithIndex, tensor::TensorPtr> &op_output_map, in RunControlOperator()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/ |
| D | split_n_optimizer.cc | 30 using KernelWithIndex = std::pair<AnfNodePtr, size_t>; typedef
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/pass/ |
| D | eliminate_redundant_op.cc | 29 using KernelWithIndex = std::pair<CNodePtr, size_t>; typedef
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | session_basic.cc | 379 … std::map<KernelWithIndex, std::vector<std::vector<size_t>>> *output_indexes) { in CreateNodeOutputPlaceholder()
411 … std::map<KernelWithIndex, std::vector<std::vector<size_t>>> *output_indexes) { in CreateNodeOutputPlaceholder()
1275 std::map<KernelWithIndex, std::vector<std::vector<size_t>>> *output_indexes) { in CreateOutputPlaceholder()
1288 void SessionBasic::GetRefCount(const KernelGraph *graph, std::map<KernelWithIndex, size_t> *ref_cou… in GetRefCount()
1303 std::map<KernelWithIndex, size_t> *ref_count, in HandleOpInputs()
1333 const std::map<KernelWithIndex, size_t> &ref_count, in HandleOpOutputs()
1424 … const std::map<KernelWithIndex, tensor::TensorPtr> &op_output) { in GetCNodeOutputTensor()
1433 const std::map<KernelWithIndex, tensor::TensorPtr> &op_output, in GetOpInputTensors()
1469 … const std::map<KernelWithIndex, tensor::TensorPtr> &op_output, in GetOpInputTensorByIndex()
|
| D | ascend_session.cc | 167 … const std::map<KernelWithIndex, OutputTensorInfo> &node_output_info, in GetCNodeOutputStubTensor()
179 const std::map<KernelWithIndex, size_t> &cnode_refcount, in GenOpOutputStubTensor()
1082 … const std::map<KernelWithIndex, OutputTensorInfo> &node_output_info, in GetOpInputStubTensors()
1119 const std::map<KernelWithIndex, size_t> &cnode_refcount) { in BuildOpsInGraph()
|
| D | session_basic.h | 244 const std::map<KernelWithIndex, size_t> &cnode_refcount) {} in BuildOpsInGraph()
|
| D | kernel_graph.h | 39 using KernelWithIndex = std::pair<AnfNodePtr, size_t>; variable
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | set_fracz_group_attr.cc | 29 using KernelWithIndex = std::pair<AnfNodePtr, size_t>; typedef
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | device_address.h | 58 using KernelWithIndex = std::pair<AnfNodePtr, size_t>; variable
|