| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_activation.h | 30 class StreamExecutor; variable
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_activation.h | 30 class StreamExecutor; variable
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_activation.h | 30 class StreamExecutor; variable
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | timer.h | 29 class StreamExecutor; variable
|
| D | event.h | 30 class StreamExecutor; variable
|
| D | platform.h | 35 class StreamExecutor; variable
|
| D | kernel.h | 90 class StreamExecutor; variable
|
| D | device_memory.h | 45 class StreamExecutor; variable
|
| D | stream.h | 75 class StreamExecutor; variable
|
| D | stream_executor_pimpl.cc | 139 StreamExecutor::StreamExecutor( in StreamExecutor() function in stream_executor::StreamExecutor
|
| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_event_mgr.h | 35 class StreamExecutor; variable
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | c_api_defn.h | 31 class StreamExecutor; variable
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_transfer_manager.cc | 117 se::StreamExecutor* /*executor*/, MutableBorrowingLiteral literal) { in TransferLiteralFromOutfeed()
|
| D | thunk.h | 99 se::StreamExecutor* /*executor*/) { in Initialize()
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | compiler.cc | 97 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* /*stream_exec*/, in RunHloPasses()
|
| /external/tensorflow/tensorflow/c/experimental/stream_executor/ |
| D | stream_executor_test.cc | 210 TEST(StreamExecutor, SuccessfulRegistration) { in TEST() argument
232 TEST(StreamExecutor, NameNotSet) { in TEST() argument
247 TEST(StreamExecutor, InvalidNameWithSemicolon) { in TEST() argument
264 TEST(StreamExecutor, InvalidNameWithSlash) { in TEST() argument
280 TEST(StreamExecutor, CreateDeviceNotSet) { in TEST() argument
296 TEST(StreamExecutor, UnifiedMemoryAllocateNotSet) { in TEST() argument
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | llvm_compiler_test.cc | 56 GpuVersion GetGpuVersion(se::StreamExecutor*) override { in GetGpuVersion()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device.cc | 466 stream->ThenEnqueueOnBackgroundThread([stream, done](se::StreamExecutor*) { in Sync()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 575 std::unique_ptr<HloModule> module, se::StreamExecutor* /*stream_exec*/, in RunHloPasses()
|