Home
last modified time | relevance | path

Searched refs:LoadMemToHost (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/debug/debugger/
Ddebugger_utils.cc73 …auto ret = gpu_addr->LoadMemToHost(input_tensor_name, exec_order_, format, int_shapes, type, 0, tr… in LoadInputs()
102 … auto ret = gpu_addr->LoadMemToHost(tensor_name, exec_order_, format, int_shapes, type, j, false); in LoadOutputs()
Ddebugger.cc1350 bool ret = addr->LoadMemToHost(tensor_name, exec_order, format, int_shapes, type, 0, keep_prev); in LoadSingleAnfnode()
1423 auto ret = addr->LoadMemToHost(tensor_name, exec_order, format, int_shapes, type, j, false); in LoadGraphOutputs()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_device_address.h53 …bool LoadMemToHost(const std::string &tensor_name, int execution_order, const std::string &host_fm…
Dgpu_device_address.cc131 bool GPUDeviceAddress::LoadMemToHost(const std::string &tensor_name, int execution_order, const std… in LoadMemToHost() function in mindspore::device::gpu::GPUDeviceAddress
Dgpu_kernel_runtime.cc177 …auto ret = gpu_addr->LoadMemToHost(input_tensor_name, exec_order, format, int_shapes, type, 0, tru… in LoadKernelData()
204 … auto ret = gpu_addr->LoadMemToHost(tensor_name, exec_order, format, int_shapes, type, j, false); in LoadKernelData()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_device_address.h57 …bool LoadMemToHost(const std::string &tensor_name, int execution_order, const std::string &host_fm…
Dascend_device_address.cc540 bool AscendDeviceAddress::LoadMemToHost(const std::string &tensor_name, int execution_order, const … in LoadMemToHost() function in mindspore::device::ascend::AscendDeviceAddress
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Ddevice_address.h92 …virtual bool LoadMemToHost(const std::string &tensor_name, int execution_order, const std::string … in LoadMemToHost() function