| /aosp_15_r20/external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| H A D | stream_executor_test.cc | 34 TEST(StreamExecutor, SuccessfulRegistration) { in TEST() argument
56 TEST(StreamExecutor, NameNotSet) { in TEST() argument
71 TEST(StreamExecutor, InvalidNameWithSemicolon) { in TEST() argument
88 TEST(StreamExecutor, InvalidNameWithSlash) { in TEST() argument
104 TEST(StreamExecutor, CreateDeviceNotSet) { in TEST() argument
120 TEST(StreamExecutor, UnifiedMemoryAllocateNotSet) { in TEST() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_activation.h | 30 class StreamExecutor; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_activation.h | 30 class StreamExecutor; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/ |
| H A D | gpu_activation.h | 30 class StreamExecutor; variable
|
| H A D | gpu_executor.h | 45 class StreamExecutor; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | timer.h | 29 class StreamExecutor; variable
|
| H A D | event.h | 30 class StreamExecutor; variable
|
| H A D | kernel.h | 90 class StreamExecutor; variable
|
| H A D | platform.h | 34 class StreamExecutor; variable
|
| H A D | device_memory.h | 45 class StreamExecutor; variable
|
| H A D | stream.h | 74 class StreamExecutor; variable
|
| H A D | stream_executor_pimpl.cc | 139 StreamExecutor::StreamExecutor( in StreamExecutor() function in stream_executor::StreamExecutor
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | xla_executor_state.h | 31 explicit GpuExecutorXLAState(stream_executor::StreamExecutor *) {} in GpuExecutorXLAState()
|
| H A D | thunk.h | 100 se::StreamExecutor* /*executor*/) { in Initialize()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/device/ |
| H A D | device_event_mgr.h | 35 class StreamExecutor; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| H A D | c_api_defn.h | 31 class StreamExecutor; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| H A D | compiler.cc | 102 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* /*stream_exec*/, in RunHloPasses()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | llvm_compiler_test.cc | 58 GpuVersion GetGpuVersion(se::StreamExecutor*) override { in GetGpuVersion()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_device.cc | 496 stream->ThenEnqueueOnBackgroundThread([stream, done](se::StreamExecutor*) { in Sync()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_compiler.cc | 820 std::unique_ptr<HloModule> module, se::StreamExecutor* /*stream_exec*/, in RunHloPasses()
|