Searched refs:total_mem_size (Results 1 – 11 of 11) sorted by relevance
24 size_t total_mem_size = 32; in MemPlan() local38 total_mem_size += address->size_; in MemPlan()47 total_mem_size += address->size_; in MemPlan()57 total_mem_size += address->size_; in MemPlan()62 return total_mem_size; in MemPlan()
30 size_t total_size = total_mem_size(); in Init()102 size_t GPUMemoryAllocator::total_mem_size() { return CudaDriver::total_mem_size(); } in total_mem_size() function in mindspore::device::gpu::GPUMemoryAllocator
38 size_t total_mem_size() override;
49 static size_t total_mem_size();
124 size_t CudaDriver::total_mem_size() { in total_mem_size() function in mindspore::device::gpu::CudaDriver
39 size_t total_mem_size() override;
94 size_t CPUMemoryPool::total_mem_size() { return GetSystemMemorySize("MemTotal"); } in total_mem_size() function in mindspore::device::cpu::CPUMemoryPool
43 size_t total_mem_size() override;
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() argument
117 virtual size_t total_mem_size() = 0;