Searched defs:MallocDeviceMemory (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ | ||
| D | cpu_memory_manager.h | 37 void MallocDeviceMemory() override {} in MallocDeviceMemory() function |
| /third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/ | ||
| D | gpu_memory_manager.cc | 73 void GPUMemoryManager::MallocDeviceMemory() { in MallocDeviceMemory() function in mindspore::device::gpu::GPUMemoryManager |
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ | ||
| D | ascend_memory_manager.cc | 55 void AscendMemoryManager::MallocDeviceMemory() { in MallocDeviceMemory() function in mindspore::device::ascend::AscendMemoryManager |