Searched refs:AllocatorParts (Results 1 – 2 of 2) sorted by relevance
137 struct AllocatorParts { struct143 std::vector<AllocatorParts> gpu_allocators_ GUARDED_BY(mu_); argument146 std::vector<AllocatorParts> gpu_host_allocators_ GUARDED_BY(mu_);
93 AllocatorParts& allocator_parts = gpu_allocators_[tf_gpu_id.value()]; in GetGPUAllocator()173 AllocatorParts& allocator_parts = gpu_allocators_[tf_gpu_id.value()]; in GPUAllocatorCounter()256 AllocatorParts& allocator_parts = gpu_host_allocators_.back(); in GetGpuHostAllocator()