Searched refs:MallocDynamicMem (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | memory_manager.cc | 50 auto base_ptr = MallocDynamicMem(total_allocated_size, false); in MallocSomasDynamicMem()
99 ptr = MallocDynamicMem(size, communication_mem); in MallocOutputMem()
109 ptr = MallocDynamicMem(size, false); in MallocOutputMem()
123 return MallocDynamicMem(size, false); in MallocWorkSpaceMem()
133 ptr = MallocDynamicMem(size, false); in MallocMem()
139 uint8_t *MemoryManager::MallocDynamicMem(size_t size, bool communication_mem) { in MallocDynamicMem() function in mindspore::device::MemoryManager
|
| D | memory_manager.h | 112 virtual uint8_t *MallocDynamicMem(size_t size, bool communication_mem);
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_memory_manager.h | 50 uint8_t *MallocDynamicMem(size_t size, bool communication_mem) override;
|
| D | ascend_memory_manager.cc | 168 uint8_t *AscendMemoryManager::MallocDynamicMem(size_t size, bool communication_mem) { in MallocDynamicMem() function in mindspore::device::ascend::AscendMemoryManager
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/ |
| D | cpu_memory_manager.h | 57 uint8_t *MallocDynamicMem(size_t size, bool communication_mem) override;
|
| D | cpu_memory_manager.cc | 43 uint8_t *CPUMemoryManager::MallocDynamicMem(size_t size, bool) { in MallocDynamicMem() function in mindspore::device::cpu::CPUMemoryManager
|