Home
last modified time | relevance | path

Searched refs:GetMutableOutputAddr (Results 1 – 25 of 26) sorted by relevance

12

/third_party/mindspore/mindspore/ccsrc/runtime/framework/actor/
Ddata_prepare_actor.cc84 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(node, i, false); in FetchContinuousMemoryInfo()
255 auto device_address = AnfAlgo::GetMutableOutputAddr(input_node, 0, false); in PrepareDataForHostTensorQueue()
318 auto device_tensor = AnfAlgo::GetMutableOutputAddr(input_node, 0, false); in PrepareDataForStepMode()
349 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(node, i, false); in PrepareDataForValueNodeTensor()
376 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(node, 0, false); in PrepareDataForValueNode()
412 auto device_tensor = AnfAlgo::GetMutableOutputAddr(backend_node, 0, false); in PrepareDataForWeightNode()
568 auto device_address = AnfAlgo::GetMutableOutputAddr(backend_node, 0, false); in PrepareHostTensorQueueForControlNode()
Doutput_actor.cc38 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(output_node, output_index, false); in CreateOutputTensor()
88 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(output_node, output_index, false); in UpdateOutputDeviceAddress()
Dactor_common.cc182 auto device_tensor = AnfAlgo::GetMutableOutputAddr(node, output_idx, false); in UpdateRefCount()
Dgather_actor.cc119 if (AnfAlgo::GetMutableOutputAddr(backend_node.first, backend_node.second, false).get() == in SendOutput()
Ddata_source_actor.cc225 auto device_address = AnfAlgo::GetMutableOutputAddr(data_node, 0, false); in FillDataBuffer()
Dswitch_actor.cc414 …AnfAlgo::GetMutableOutputAddr(backend_node.first, j, false).get() == input_device_tensors_[from_in… in SendOutput()
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_simple_mem_plan.cc79 …auto address = AnfAlgo::GetMutableOutputAddr(kernel_with_index.first, kernel_with_index.second, tr… in MemAssign()
89 auto address = AnfAlgo::GetMutableOutputAddr(kernel, i); in MemAssign()
Dcpu_memory_manager.cc142 auto address = AnfAlgo::GetMutableOutputAddr(node, index); in IncreaseSummaryRefCount()
158 auto address = AnfAlgo::GetMutableOutputAddr(node, index); in DecreaseSummaryRefCount()
Dcpu_kernel_runtime.cc197 auto address = AnfAlgo::GetMutableOutputAddr(node, index); in CreatTensorForOutput()
319 auto address = AnfAlgo::GetMutableOutputAddr(item, 0); in BindInputTensorAddressPtr()
445 auto device_address = AnfAlgo::GetMutableOutputAddr(kernel, i).get(); in Run()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_kernel_runtime.cc334 auto device_address = GetMutableOutputAddr(item[0], output_index, false); in AllocInplaceNodeMemory()
688 auto device_address = AnfAlgo::GetMutableOutputAddr(kernel, i, false); in ClearKernelOutputAddress()
945 auto device_address = GetMutableOutputAddr(need_swap_kernel, mem_swap_info.output_idx_, false); in AddMemorySwapTask()
1146 auto device_address = GetMutableOutputAddr(kernel, i, false); in AllocKernelOutputDynamicRes()
1246 auto device_address = GetMutableOutputAddr(kernel, i, false); in AllocCommunicationOpOutputDynamicRes()
1321 auto device_address = GetMutableOutputAddr(kernel, i, false); in FreeKernelDynamicRes()
1359 auto addr = kernel_info->GetMutableOutputAddr(prev_node_with_index.second); in GetPrevNodeMutableOutputAddr()
1364 DeviceAddressPtr GPUKernelRuntime::GetMutableOutputAddr(const AnfNodePtr &node, size_t i, bool visi… in GetMutableOutputAddr() function in mindspore::device::gpu::GPUKernelRuntime
1366 return AnfAlgo::GetMutableOutputAddr(node, i, visit_nop_node); in GetMutableOutputAddr()
1382 auto device_address = AnfAlgo::GetMutableOutputAddr(node, i, visit_nop_node); in GetMutableOutputAddr()
[all …]
Dgpu_kernel_runtime.h114 DeviceAddressPtr GetMutableOutputAddr(const AnfNodePtr &node, size_t i, bool visit_nop_node);
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dascend_inference_session.cc44 auto device_address = AnfAlgo::GetMutableOutputAddr(pk_node, 0); in LoadInputData()
70 auto device_address = AnfAlgo::GetMutableOutputAddr(pk_node, 0); in CompileGraphImpl()
Dgpu_inference_session.cc44 auto device_address = AnfAlgo::GetMutableOutputAddr(pk_node, 0); in LoadInputData()
70 auto device_address = AnfAlgo::GetMutableOutputAddr(pk_node, 0); in CompileGraphImpl()
Dgpu_session.cc342 auto device_address = AnfAlgo::GetMutableOutputAddr(pk_node, 0); in LoadInputData()
579 auto address = AnfAlgo::GetMutableOutputAddr(node, output_index); in UpdateOutputTensors()
Dascend_session.cc326 auto device_address = AnfAlgo::GetMutableOutputAddr(parameter, 0); in TensorNeedSync()
496 auto device_address = AnfAlgo::GetMutableOutputAddr(input_node, 0); in LoadInputData()
824 address = AnfAlgo::GetMutableOutputAddr(node, output_index, false); in BindAddressToTensor()
826 address = AnfAlgo::GetMutableOutputAddr(node, output_index); in BindAddressToTensor()
2035 const auto &address = AnfAlgo::GetMutableOutputAddr(node, output_index); in UpdateOutputTensors()
Dcpu_session.cc180 auto address = AnfAlgo::GetMutableOutputAddr(input_node, 0); in LoadInputData()
Danf_runtime_algorithm.h189 …static DeviceAddressPtr GetMutableOutputAddr(const AnfNodePtr &node, size_t output_idx, bool visit…
Dsession_basic.cc573 auto address = AnfAlgo::GetMutableOutputAddr(ref_real_node, ref_real_node_index); in InitInternalOutputParameter()
1628 address = AnfAlgo::GetMutableOutputAddr(node, output_index, false); in UpdateOutputs()
1630 address = AnfAlgo::GetMutableOutputAddr(node, output_index); in UpdateOutputs()
1736 const auto &address = AnfAlgo::GetMutableOutputAddr(node, output_index); in UpdateOutputTensors()
Danf_runtime_algorithm.cc1053 DeviceAddressPtr AnfRuntimeAlgorithm::GetMutableOutputAddr(const AnfNodePtr &node, size_t output_id… in GetMutableOutputAddr() function in mindspore::session::AnfRuntimeAlgorithm
1069 auto addr = kernel_info->GetMutableOutputAddr(output_idx); in GetMutableOutputAddr()
1112 …return AnfRuntimeAlgorithm::GetMutableOutputAddr(kernel_with_index.first, kernel_with_index.second… in GetPrevNodeMutableOutputAddr()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dkernel_info.h53 DeviceAddressPtr GetMutableOutputAddr(size_t index) const;
Dkernel_info.cc33 DeviceAddressPtr KernelInfo::GetMutableOutputAddr(size_t index) const { in GetMutableOutputAddr() function in mindspore::device::KernelInfo
Dkernel_runtime.cc128 address = AnfAlgo::GetMutableOutputAddr(input_node, input_node_with_index.second); in RunOpGetCommunicationInputInfo()
183 address = AnfAlgo::GetMutableOutputAddr(node, i); in RunOpGetCommunicationOutputInfo()
456 auto address = AnfAlgo::GetMutableOutputAddr(kernel, i, false); in RunOpAssignOutputMemory()
672 …auto origin_node_output_addr = AnfAlgo::GetMutableOutputAddr(origin_pair.first, origin_pair.second… in UpdateRefNodeOutputMem()
674 auto cur_node_output_addr = AnfAlgo::GetMutableOutputAddr(kernel, i); in UpdateRefNodeOutputMem()
1018 auto device_address = AnfAlgo::GetMutableOutputAddr(value_node, 0); in AssignStaticMemoryValueNode()
1351 auto device_address = AnfAlgo::GetMutableOutputAddr(kernel, j, true); in SyncNodeOutputTensors()
1399 auto device_address = AnfAlgo::GetMutableOutputAddr(input_node, 0); in InitGraphInputTensors()
/third_party/mindspore/mindspore/ccsrc/runtime/framework/
Dgraph_compiler.cc241 auto device_address = AnfAlgo::GetMutableOutputAddr(group_nodes[0], output_index, false); in UpdateDeviceAddressForInplaceNode()
272 auto device_address = AnfAlgo::GetMutableOutputAddr(node, index, false); in SetSummaryNodesRefCount()
284 …auto device_address = AnfAlgo::GetMutableOutputAddr(item_with_index.first, item_with_index.second,… in UpdateRefCountForGraphOutput()
Dgraph_scheduler.cc1139 auto from_device_tensor = AnfAlgo::GetMutableOutputAddr(from_kernel, from_output_index, false); in LinkDataArrowForCopyActor()
1873 …auto device_tensor = AnfAlgo::GetMutableOutputAddr(from_actor->kernel_, backend_node.second, false… in LinkDataArrowByControlNode()
1881 …auto device_tensor = AnfAlgo::GetMutableOutputAddr(from_actor->kernel_, input_with_index.second, f… in LinkDataArrowByControlNode()
1907 …auto device_tensor = AnfAlgo::GetMutableOutputAddr(from_actor->data_nodes_[iter->second], 0, false… in LinkDataArrowByControlNode()
2235 auto device_tensor = AnfAlgo::GetMutableOutputAddr(value_node, 0, false); in PersistDeviceTensor()
2261 auto device_tensor = AnfAlgo::GetMutableOutputAddr(input_node, 0, false); in PersistDeviceTensor()
2288 auto device_tensor = AnfAlgo::GetMutableOutputAddr(value_node.first, 0, false); in PersistDeviceTensor()
2523 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(data_kernel, i, false); in DumpDSActor()
2536 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(data_node, 0, false); in DumpDSActor()
2568 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(kernel, i, false); in DumpKernelActor()
/third_party/mindspore/mindspore/ccsrc/vm/
Dbackend.cc209 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(output_node, output_index, false); in CreateOutputTensor()
245 const auto &device_tensor = AnfAlgo::GetMutableOutputAddr(output_node, output_index, false); in UpdateOutputDeviceAddress()

12