Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_id_utils.h41 static se::port::StatusOr<se::StreamExecutor*> ExecutorForTfGpuId( in ExecutorForTfGpuId() function
Dgpu_process_state.cc73 GpuIdUtil::ExecutorForTfGpuId(tf_gpu_id).ValueOrDie(); in BusIdForGPU()
217 se = GpuIdUtil::ExecutorForTfGpuId(TfGpuId(i)).ValueOrDie(); in GetGpuHostAllocator()
Dgpu_device.cc373 auto executor_status = GpuIdUtil::ExecutorForTfGpuId(tf_gpu_id_); in Init()