Home
last modified time | relevance | path

Searched defs:ExecutorForDevice (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/host/
Dhost_platform.cc48 port::StatusOr<StreamExecutor*> HostPlatform::ExecutorForDevice(int ordinal) { in ExecutorForDevice() function in stream_executor::host::HostPlatform
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dplatform.cc50 port::StatusOr<StreamExecutor*> XlaInterpreterPlatform::ExecutorForDevice( in ExecutorForDevice() function in stream_executor::interpreter::XlaInterpreterPlatform
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_platform.cc112 port::StatusOr<StreamExecutor*> ROCmPlatform::ExecutorForDevice(int ordinal) { in ExecutorForDevice() function in stream_executor::gpu::ROCmPlatform
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_platform.cc149 port::StatusOr<StreamExecutor*> CudaPlatform::ExecutorForDevice(int ordinal) { in ExecutorForDevice() function in stream_executor::gpu::CudaPlatform
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_jit_compiled_cpu_function_test.cc303 se::port::StatusOr<se::StreamExecutor*> ExecutorForDevice( in TEST() function in tensorflow::__anon2c9a16e00111::TEST::FakePlatform