Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_launch_kernel.cc29 size_t AlignMemorySize(size_t size) { in AlignMemorySize() function
Dgpu_bucket.cc35 size_t AlignMemorySize(size_t size) { in AlignMemorySize() function
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_memory_pool.cc114 size_t AscendMemoryPool::AlignMemorySize(size_t size) const { in AlignMemorySize() function in mindspore::device::ascend::AscendMemoryPool
/third_party/mindspore/mindspore/lite/micro/coder/allocator/
Dmemory_manager.cc25 static size_t AlignMemorySize(size_t size) { in AlignMemorySize() function
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/mem_reuse/
Dmem_dynamic_allocator.cc77 size_t DynamicMemPoolBestFit::AlignMemorySize(size_t size) const { in AlignMemorySize() function in mindspore::device::DynamicMemPoolBestFit
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/
Dnccl_collective_gpu_kernel.h247 size_t AlignMemorySize(size_t size) const { in AlignMemorySize() function