Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_process_state.h137 struct AllocatorParts { struct
143 std::vector<AllocatorParts> gpu_allocators_ GUARDED_BY(mu_); argument
146 std::vector<AllocatorParts> gpu_host_allocators_ GUARDED_BY(mu_);
Dgpu_process_state.cc93 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()