Searched refs:current_allocated_memory (Results 1 – 4 of 4) sorted by relevance
139 op_name, current_allocated_memory = self.memories_allocated[i]140 op_diff[op_name] = current_allocated_memory - previous_allocated_memory141 previous_allocated_memory = current_allocated_memory
18 current_allocated_memory
113 def current_allocated_memory() -> int: function
1141 caching_allocator_mem_allocated = torch.mps.current_allocated_memory()1147 self.caching_allocator_before = torch.mps.current_allocated_memory()1157 caching_allocator_mem_allocated = torch.mps.current_allocated_memory()1172 caching_allocator_mem_allocated = torch.mps.current_allocated_memory()8008 current_alloc_before = torch.mps.current_allocated_memory()8017 current_alloc_after = torch.mps.current_allocated_memory()