Searched defs:total_mem_size (Results 1 – 6 of 6) sorted by relevance
94 size_t CPUMemoryPool::total_mem_size() { return GetSystemMemorySize("MemTotal"); } in total_mem_size() function in mindspore::device::cpu::CPUMemoryPool
102 size_t GPUMemoryAllocator::total_mem_size() { return CudaDriver::total_mem_size(); } in total_mem_size() function in mindspore::device::gpu::GPUMemoryAllocator
124 size_t CudaDriver::total_mem_size() { in total_mem_size() function in mindspore::device::gpu::CudaDriver
24 size_t total_mem_size = 32; in MemPlan() local
146 size_t AscendMemoryPool::total_mem_size() { return device_mem_size_ - graph_dynamic_mem_offset_; } in total_mem_size() function in mindspore::device::ascend::AscendMemoryPool
194 void SetMemTotal(uint64_t total_mem_size) { this->mem_total_ = total_mem_size; } in SetMemTotal()