Searched refs:TotalAllocatedBytes (Results 1 – 2 of 2) sorted by relevance
46 int64 TotalAllocatedBytes() { return total_allocated_bytes_; } in TotalAllocatedBytes() function
63 int64 TotalAllocatedBytes() { return total_allocated_bytes_; } in TotalAllocatedBytes() function in xla::gpu::__anon6437d7ed0111::ScratchAllocator744 int64 scratch_bytes_used = scratch_allocator.TotalAllocatedBytes(); in PickBestAlgorithmNoCacheRocm()