Searched refs:underlying_device_ (Results 1 – 2 of 2) sorted by relevance
40 return underlying_device_->UnderlyingDevice(); in UnderlyingDevice()43 return underlying_device_->UnderlyingDevice(); in UnderlyingDevice()50 return underlying_device_->tensorflow_cpu_worker_threads(); in tensorflow_cpu_worker_threads()54 return underlying_device_->tensorflow_gpu_device_info(); in tensorflow_gpu_device_info()58 return underlying_device_->GetAllocator(attr); in GetAllocator()63 return underlying_device_->GetScopedAllocator(attr, step_id); in GetScopedAllocator()67 return underlying_device_->GetScopedAllocatorMgr(); in GetScopedAllocatorMgr()73 if (underlying_threadpool_ && underlying_device_->has_eigen_cpu_device()) { in eigen_cpu_device()76 return underlying_device_->eigen_cpu_device(); in eigen_cpu_device()84 underlying_device_->tensorflow_device_thread_pool() != nullptr) { in tensorflow_device_thread_pool()[all …]
56 underlying_device_(underlying), in RenamedDevice()74 delete underlying_device_; in ~RenamedDevice()