Searched refs:LoadMemToHost (Results 1 – 8 of 8) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/debug/debugger/ |
D | debugger_utils.cc | 73 …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()
|
D | debugger.cc | 1350 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/ |
D | gpu_device_address.h | 53 …bool LoadMemToHost(const std::string &tensor_name, int execution_order, const std::string &host_fm…
|
D | gpu_device_address.cc | 131 bool GPUDeviceAddress::LoadMemToHost(const std::string &tensor_name, int execution_order, const std… in LoadMemToHost() function in mindspore::device::gpu::GPUDeviceAddress
|
D | gpu_kernel_runtime.cc | 177 …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/ |
D | ascend_device_address.h | 57 …bool LoadMemToHost(const std::string &tensor_name, int execution_order, const std::string &host_fm…
|
D | ascend_device_address.cc | 540 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/ |
D | device_address.h | 92 …virtual bool LoadMemToHost(const std::string &tensor_name, int execution_order, const std::string … in LoadMemToHost() function
|