Home
last modified time | relevance | path

Searched defs:StreamExecutor (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_activation.h30 class StreamExecutor; variable
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_activation.h30 class StreamExecutor; variable
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_activation.h30 class StreamExecutor; variable
/external/tensorflow/tensorflow/stream_executor/
Dtimer.h29 class StreamExecutor; variable
Devent.h30 class StreamExecutor; variable
Dplatform.h35 class StreamExecutor; variable
Dkernel.h89 class StreamExecutor; variable
Ddevice_memory.h45 class StreamExecutor; variable
Dstream.h74 class StreamExecutor; variable
Dstream_executor_pimpl.cc138 StreamExecutor::StreamExecutor( in StreamExecutor() function in stream_executor::StreamExecutor
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc93 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* /*stream_exec*/, in RunHloPasses()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_transfer_manager.cc148 se::StreamExecutor* /*executor*/, const Shape& literal_shape, in TransferLiteralFromOutfeed()
Dthunk.h88 se::StreamExecutor* /*executor*/) { in Initialize()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_event_mgr.h35 class StreamExecutor; variable
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.cc466 stream->ThenEnqueueOnBackgroundThread([stream, done](se::StreamExecutor*) { in Sync()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc510 std::unique_ptr<HloModule> module, se::StreamExecutor* /*stream_exec*/, in RunHloPasses()