| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_memory_manager.cc | 178 auto address = AnfAlgo::GetPrevNodeMutableOutputAddr(kernel, i); in IncreaseAddressRefCount()
199 auto address = AnfAlgo::GetPrevNodeMutableOutputAddr(kernel, i); in DecreaseAddressRefCount()
|
| D | cpu_kernel_runtime.cc | 439 auto device_address = AnfAlgo::GetPrevNodeMutableOutputAddr(kernel, i).get(); in Run()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ |
| D | gpu_kernel_runtime.cc | 1106 device_address = GetPrevNodeMutableOutputAddr(kernel, i, false); in AllocKernelInputDynamicRes()
1109 device_address = GetPrevNodeMutableOutputAddr(kernel, i, true); in AllocKernelInputDynamicRes()
1119 device_address = GetPrevNodeMutableOutputAddr(skip_node, 0, false); in AllocKernelInputDynamicRes()
1217 device_address = GetPrevNodeMutableOutputAddr(kernel, i, false); in AllocCommunicationOpInputDynamicRes()
1220 device_address = GetPrevNodeMutableOutputAddr(kernel, i, true); in AllocCommunicationOpInputDynamicRes()
1303 device_address = GetPrevNodeMutableOutputAddr(kernel, i, false); in FreeKernelDynamicRes()
1306 device_address = GetPrevNodeMutableOutputAddr(kernel, i, true); in FreeKernelDynamicRes()
1339 DeviceAddressPtr GPUKernelRuntime::GetPrevNodeMutableOutputAddr(const AnfNodePtr &node, size_t i, b… in GetPrevNodeMutableOutputAddr() function in mindspore::device::gpu::GPUKernelRuntime
1341 return AnfAlgo::GetPrevNodeMutableOutputAddr(node, i, visit_nop_node); in GetPrevNodeMutableOutputAddr()
|
| D | gpu_kernel_runtime.h | 113 …DeviceAddressPtr GetPrevNodeMutableOutputAddr(const AnfNodePtr &node, size_t i, bool visit_nop_nod…
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/executor/ |
| D | dynamic_kernel.cc | 66 auto output_addr = AnfAlgo::GetPrevNodeMutableOutputAddr(cnode, depend, visit_nop_node); in RebuildDependTensor()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/ |
| D | dynamic_broadcast_gradient_args_kernel.cc | 124 auto address_x = AnfAlgo::GetPrevNodeMutableOutputAddr(cnode, index); in GetInputShape()
|
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | anf_runtime_algorithm.h | 197 static DeviceAddressPtr GetPrevNodeMutableOutputAddr(const AnfNodePtr &anf_node, size_t input_idx,
|
| D | anf_runtime_algorithm.cc | 1060 return AnfRuntimeAlgorithm::GetPrevNodeMutableOutputAddr(cnode, 0); in GetMutableOutputAddr()
1109 DeviceAddressPtr AnfRuntimeAlgorithm::GetPrevNodeMutableOutputAddr(const AnfNodePtr &anf_node, size… in GetPrevNodeMutableOutputAddr() function in mindspore::session::AnfRuntimeAlgorithm
2361 auto device_address = GetPrevNodeMutableOutputAddr(kernel, real_input); in CacheAddrForGraph()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/ |
| D | data_prepare_actor.cc | 75 const auto &device_tensor = AnfAlgo::GetPrevNodeMutableOutputAddr(node, i, false); in FetchContinuousMemoryInfo()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 467 auto device_address = AnfAlgo::GetPrevNodeMutableOutputAddr(kernel, i); in RunOpAssignOutputMemory()
1512 auto device_address = AnfAlgo::GetPrevNodeMutableOutputAddr(kernel, real_input); in LaunchKernelMod()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/framework/ |
| D | graph_scheduler.cc | 1343 auto device_tensor = AnfAlgo::GetPrevNodeMutableOutputAddr(from_allreduce_node, i, false); in LinkControlArrowBySendRecvNodes()
|