Searched refs:cuda_allocators_ (Results 1 – 2 of 2) sorted by relevance
99 XRTGenericDeviceAccessor::ScopedRef::cuda_allocators_ = member in tensorflow::XRTGenericDeviceAccessor::ScopedRef114 if (!cuda_allocators_->count(stream)) { in Acquire()116 if (!cuda_allocators_->count(stream)) { in Acquire()120 (*cuda_allocators_)[stream] = in Acquire()125 (*cuda_allocators_)[stream].get()); in Acquire()
72 cuda_allocators_; variable