Searched defs:device_mem_pool_offset (Results 1 – 2 of 2) sorted by relevance
128 void AscendMemoryPool::set_device_mem_pool_offset(uint64_t device_mem_pool_offset) { in set_device_mem_pool_offset()136 uint64_t AscendMemoryPool::device_mem_pool_offset() const { return device_mem_pool_offset_; } in device_mem_pool_offset() function in mindspore::device::ascend::AscendMemoryPool
142 auto device_mem_pool_offset = AscendMemoryPool::GetInstance().device_mem_pool_offset(); in MallocStaticMem() local176 auto device_mem_pool_offset = AscendMemoryPool::GetInstance().device_mem_pool_offset(); in MallocDynamicMem() local