Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/host/
H A Dhost_platform.cc48 port::StatusOr<StreamExecutor*> HostPlatform::ExecutorForDevice(int ordinal) { in ExecutorForDevice() function in stream_executor::host::HostPlatform
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
H A Dplatform.cc49 port::StatusOr<StreamExecutor*> XlaInterpreterPlatform::ExecutorForDevice( in ExecutorForDevice() function in stream_executor::interpreter::XlaInterpreterPlatform
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
H A Dtpu_platform.h85 StatusOr<::stream_executor::StreamExecutor*> ExecutorForDevice( in ExecutorForDevice() function
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_platform.cc113 port::StatusOr<StreamExecutor*> ROCmPlatform::ExecutorForDevice(int ordinal) { in ExecutorForDevice() function in stream_executor::gpu::ROCmPlatform
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_platform.cc143 port::StatusOr<StreamExecutor*> CudaPlatform::ExecutorForDevice(int ordinal) { in ExecutorForDevice() function in stream_executor::gpu::CudaPlatform
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dxla_jit_compiled_cpu_function_test.cc300 se::port::StatusOr<se::StreamExecutor*> ExecutorForDevice( in TEST() function in tensorflow::__anon8ca9750b0111::TEST::FakePlatform
/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/stream_executor/
H A Dstream_executor.cc691 port::StatusOr<StreamExecutor*> CPlatform::ExecutorForDevice(int ordinal) { in ExecutorForDevice() function in stream_executor::CPlatform