Home
last modified time | relevance | path

Searched defs:from_persistent_mem (Results 1 – 6 of 6) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/backend/common/mem_reuse/
Dmem_dynamic_allocator.cc91 DeviceMemPtr DynamicMemPoolBestFit::AllocTensorMem(size_t size, bool from_persistent_mem, bool need… in AllocTensorMem()
222 DeviceMemPtr DynamicMemPoolBestFit::FindAvailableMemBuf(size_t size, bool from_persistent_mem, uint… in FindAvailableMemBuf()
237 DeviceMemPtr DynamicMemPoolBestFit::FindMemBufByStatus(size_t size, bool from_persistent_mem, in FindMemBufByStatus()
249 DeviceMemPtr DynamicMemPoolBestFit::FindMemBufInSpecifiedMng(size_t size, bool from_persistent_mem, in FindMemBufInSpecifiedMng()
347 DeviceMemPtr DynamicMemPoolBestFit::AddMemBlockAndMemBuf(size_t size, bool from_persistent_mem, boo… in AddMemBlockAndMemBuf()
388 …emPtr DynamicMemPoolBestFit::AddMemBlockAndMemBufByEagerFree(size_t size, bool from_persistent_mem, in AddMemBlockAndMemBufByEagerFree()
421 DeviceMemPtr DynamicMemPoolBestFit::CreateMemBlockAndMemBuf(size_t size, bool from_persistent_mem, in CreateMemBlockAndMemBuf()
465 size_t DynamicMemPoolBestFit::CalMemBlockAllocSize(size_t size, bool from_persistent_mem, bool) { in CalMemBlockAllocSize() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/hal/device/
Dascend_memory_pool.cc81 size_t AscendMemoryPool::CalMemBlockAllocSize(size_t size, bool from_persistent_mem, bool need_recy… in CalMemBlockAllocSize()
143 DeviceMemPtr AscendMemoryPool::AllocOverflowTensorMem(size_t size, bool from_persistent_mem) { in AllocOverflowTensorMem()
Dascend_memory_manager.cc51 void *AscendMemoryManager::MallocMemFromMemPool(size_t size, bool from_persistent_mem, bool need_re… in MallocMemFromMemPool()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/hal/device/
Dgpu_memory_manager.cc26 void *GPUMemoryManager::MallocMemFromMemPool(size_t size, bool from_persistent_mem, bool, uint32_t … in MallocMemFromMemPool() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/device/
Dmemory_manager.cc134 void *MemoryManager::MallocMemFromMemPool(size_t size, bool from_persistent_mem, bool, uint32_t str… in MallocMemFromMemPool() argument
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/include/backend/
Ddevice_address.h245 bool from_persistent_mem() const { return from_persistent_mem_; } in from_persistent_mem() function
246 …void set_from_persistent_mem(bool from_persistent_mem) { from_persistent_mem_ = from_persistent_me… in set_from_persistent_mem()