Searched refs:AllocatorParts (Results 1 – 2 of 2) sorted by relevance
144 struct AllocatorParts { struct151 std::vector<AllocatorParts> gpu_allocators_ TF_GUARDED_BY(mu_); argument155 std::vector<AllocatorParts> gpu_host_allocators_ TF_GUARDED_BY(mu_);
151 AllocatorParts& allocator_parts = gpu_allocators_[tf_gpu_id.value()]; in GetGPUAllocator()234 AllocatorParts& allocator_parts = gpu_allocators_[tf_gpu_id.value()]; in GPUAllocatorCounter()325 AllocatorParts& allocator_parts = gpu_host_allocators_.back(); in GetGpuHostAllocator()