Searched refs:origin_pair (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | deal_ref_and_split_unsupported_transdata.cc | 78 … const session::KernelWithIndex &origin_pair) const { in AddRefPairToKernelGraph()
92 … << origin_pair.first.get() << ", info is " << origin_pair.first->DebugString() << " : index " in AddRefPairToKernelGraph()
93 << origin_pair.second << "}"; in AddRefPairToKernelGraph()
94 kernel_graph->AddRefCorrespondPairs(final_pair, origin_pair); in AddRefPairToKernelGraph()
107 session::KernelWithIndex origin_pair = FindRefOriginNode(input_node); in AddAdditionalToRefOutput() local
108 MS_EXCEPTION_IF_NULL(origin_pair.first); in AddAdditionalToRefOutput()
109 if (!origin_pair.first->isa<Parameter>()) { in AddAdditionalToRefOutput()
113 << origin_pair.first->DebugString() << ", index is " << origin_pair.second; in AddAdditionalToRefOutput()
114 auto origin_format = AnfAlgo::GetOutputFormat(origin_pair.first, origin_pair.second); in AddAdditionalToRefOutput()
115 auto origin_type = AnfAlgo::GetOutputDeviceDataType(origin_pair.first, origin_pair.second); in AddAdditionalToRefOutput()
[all …]
|
| D | deal_ref_and_split_unsupported_transdata.h | 49 const session::KernelWithIndex &origin_pair) const;
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/ |
| D | mem_reuse.cc | 67 auto origin_pair = graph_->GetRefCorrespondOutput(out_pair); in InitDynamicOutputKernelRef() local
68 MS_EXCEPTION_IF_NULL(origin_pair.first); in InitDynamicOutputKernelRef()
69 …MS_LOG(INFO) << "REF origin op is " << origin_pair.first->fullname_with_scope() << ", output index… in InitDynamicOutputKernelRef()
70 … << origin_pair.second << ", cur op is " << kernel_cnode->fullname_with_scope() in InitDynamicOutputKernelRef()
72 if (origin_pair.first->isa<CNode>()) { in InitDynamicOutputKernelRef()
73 auto cnode = origin_pair.first->cast<CNodePtr>(); in InitDynamicOutputKernelRef()
74 auto ref_ptr = GetRef(cnode, origin_pair.second); in InitDynamicOutputKernelRef()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/buffer_fusion/ |
| D | ub_pattern_fusion.cc | 380 auto origin_pair = kernel_graph->GetRefCorrespondOutput(out_pair); in SetFusionOpRefInfos() local
382 kernel_graph->AddRefCorrespondPairs(fusion_final_pair, origin_pair); in SetFusionOpRefInfos()
387 auto origin_pair = kernel_graph->GetRefCorrespondOutput(out_pair); in SetFusionOpRefInfos() local
389 kernel_graph->AddRefCorrespondPairs(fusion_final_pair, origin_pair); in SetFusionOpRefInfos()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 670 auto origin_pair = graph.GetRefCorrespondOutput(out_pair); in UpdateRefNodeOutputMem() local
671 MS_EXCEPTION_IF_NULL(origin_pair.first); in UpdateRefNodeOutputMem()
672 …auto origin_node_output_addr = AnfAlgo::GetMutableOutputAddr(origin_pair.first, origin_pair.second… in UpdateRefNodeOutputMem()
677 … MS_LOG(DEBUG) << "REF origin op is " << origin_pair.first->DebugString() << ", output index is " in UpdateRefNodeOutputMem()
678 … << origin_pair.second << ", cur op is " << kernel->DebugString() << ", out index is " << i; in UpdateRefNodeOutputMem()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/ |
| D | somas.cc | 808 auto origin_pair = graph->GetRefCorrespondOutput(out_pair); in RefNodeProcess() local
809 MS_EXCEPTION_IF_NULL(origin_pair.first); in RefNodeProcess()
817 if (AnfAlgo::IsRealCNodeKernel(origin_pair.first)) { in RefNodeProcess()
818 auto ori_node = origin_pair.first->cast<CNodePtr>(); in RefNodeProcess()
819 auto ori_index = origin_pair.second; in RefNodeProcess()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | kernel_graph.h | 172 void AddRefCorrespondPairs(const AnfWithOutIndex &final_pair, const AnfWithOutIndex &origin_pair);
|
| D | kernel_graph.cc | 880 …aph::AddRefCorrespondPairs(const AnfWithOutIndex &final_pair, const AnfWithOutIndex &origin_pair) { in AddRefCorrespondPairs() argument
885 (void)ref_out_in_map_.insert(std::make_pair(final_pair, origin_pair)); in AddRefCorrespondPairs()
|