Searched refs:input_node_with_index (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/reporter/ |
| D | graph_desc_reporter.cc | 36 auto input_node_with_index = AnfAlgo::GetPrevNodeOutput(node, i); in ReportData() local
37 auto input_node = input_node_with_index.first; in ReportData()
38 auto input_index = input_node_with_index.second; in ReportData()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/executor/ |
| D | dynamic_kernel.cc | 100 auto input_node_with_index = AnfAlgo::GetPrevNodeOutput(cnode, i); in InferShapeRecursive() local
101 auto input_node = input_node_with_index.first; in InferShapeRecursive()
|
| /third_party/mindspore/mindspore/ccsrc/profiler/device/gpu/ |
| D | gpu_profiling_utils.cc | 134 auto input_node_with_index = AnfAlgo::GetPrevNodeOutput(*iter, i); in SetTraceBpEnd() local
135 auto input_node = input_node_with_index.first; in SetTraceBpEnd()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/ |
| D | deal_ref_and_split_unsupported_transdata.cc | 233 auto input_node_with_index = AnfAlgo::GetPrevNodeOutput(cnode, i, true); in DealBroadCastAsRef() local
234 auto input_node = input_node_with_index.first; in DealBroadCastAsRef()
237 AddRefPairToKernelGraph(func_graph, cnode, nullptr, cnode, i, input_node_with_index); in DealBroadCastAsRef()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 123 auto input_node_with_index = AnfAlgo::GetPrevNodeOutput(node, i, true); in RunOpGetCommunicationInputInfo() local
124 auto input_node = input_node_with_index.first; in RunOpGetCommunicationInputInfo()
127 if (AnfAlgo::OutputAddrExist(input_node, input_node_with_index.second)) { in RunOpGetCommunicationInputInfo()
128 address = AnfAlgo::GetMutableOutputAddr(input_node, input_node_with_index.second); in RunOpGetCommunicationInputInfo()
131 address = PreAssignCNodeMemory(input_node, input_node_with_index.second); in RunOpGetCommunicationInputInfo()
820 auto input_node_with_index = AnfAlgo::GetPrevNodeOutput(anf_node, index); in PreAssignCNodeMemory() local
821 return PreAssignCNodeMemory(input_node_with_index.first, input_node_with_index.second); in PreAssignCNodeMemory()
845 auto input_node_with_index = AnfAlgo::GetPrevNodeOutput(node, i, true); in AssignCommunicationNodeInputMem() local
846 auto input_node = input_node_with_index.first; in AssignCommunicationNodeInputMem()
848 if (AnfAlgo::OutputAddrExist(input_node, input_node_with_index.second)) { in AssignCommunicationNodeInputMem()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/dump/ |
| D | data_dumper.cc | 455 auto input_node_with_index = AnfAlgo::GetPrevNodeOutput(kernel, i); in DumpKernelInput() local
456 auto input_node = input_node_with_index.first; in DumpKernelInput()
457 auto input_index = input_node_with_index.second; in DumpKernelInput()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/profiling/ |
| D | profiling_utils.cc | 175 auto input_node_with_index = AnfAlgo::GetPrevNodeOutput(*iter, i); in GetTraceBpEnd() local
176 auto input_node = input_node_with_index.first; in GetTraceBpEnd()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.cc | 2025 auto input_node_with_index = AnfAlgo::GetPrevNodeOutput(anf_node, index); in GetInputMaxShape() local
2026 return GetOutputMaxShape(input_node_with_index.first, input_node_with_index.second); in GetInputMaxShape()
2030 auto input_node_with_index = AnfAlgo::GetPrevNodeOutput(anf_node, index); in GetInputMinShape() local
2031 return GetOutputMinShape(input_node_with_index.first, input_node_with_index.second); in GetInputMinShape()
|