Home
last modified time | relevance | path

Searched refs:m_current_allocated_memory (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/mps/
H A DMPSAllocator.h307 size_t getCurrentAllocatedMemory() const { return m_current_allocated_memory; } in getCurrentAllocatedMemory()
340 size_t m_current_allocated_memory = 0; variable
H A DMPSAllocator.mm299 m_current_allocated_memory += buffer_block->size;
312 TORCH_INTERNAL_ASSERT_DEBUG_ONLY(m_current_allocated_memory >= buffer_block->size);
313 m_current_allocated_memory -= buffer_block->size;