Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_bfc_allocator.h46 gpu_id_(gpu_id), in GPUMemAllocator()
60 VisitAlloc(ptr, gpu_id_.value(), num_bytes); in Alloc()
67 VisitFree(ptr, gpu_id_.value(), num_bytes); in Free()
79 const PlatformGpuId gpu_id_; variable