Searched refs:GetOrMallocAddress (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_runtime.cc | 1202 GetOrMallocAddress(mem_scheduler, device_address, input); in GenAddrCleanLaunchArgs()
1224 GetOrMallocAddress(mem_scheduler, device_address, workspace); in GenAddrCleanLaunchArgs()
1286 void KernelRuntime::GetOrMallocAddress(const std::shared_ptr<MemScheduler> &mem_scheduler, in GetOrMallocAddress() function in mindspore::device::KernelRuntime
1321 GetOrMallocAddress(mem_scheduler, device_address, input); in AssignKernelAddress()
1329 GetOrMallocAddress(mem_scheduler, device_address, output); in AssignKernelAddress()
1337 GetOrMallocAddress(mem_scheduler, device_address, workspace); in AssignKernelAddress()
|
| D | kernel_runtime.h | 152 static void GetOrMallocAddress(const std::shared_ptr<MemScheduler> &mem_scheduler,
|