Searched refs:ExecutorForTfGpuId (Results 1 – 3 of 3) sorted by relevance
41 static se::port::StatusOr<se::StreamExecutor*> ExecutorForTfGpuId( in ExecutorForTfGpuId() function
73 GpuIdUtil::ExecutorForTfGpuId(tf_gpu_id).ValueOrDie(); in BusIdForGPU()217 se = GpuIdUtil::ExecutorForTfGpuId(TfGpuId(i)).ValueOrDie(); in GetGpuHostAllocator()
373 auto executor_status = GpuIdUtil::ExecutorForTfGpuId(tf_gpu_id_); in Init()