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.h34 class StreamExecutor; variable
Dkernel.h89 class StreamExecutor; variable
Dstream.h74 class StreamExecutor; variable
Ddevice_memory.h44 class StreamExecutor; variable
Dstream_executor_pimpl.cc157 StreamExecutor::StreamExecutor(PlatformKind platform_kind, in StreamExecutor() function in stream_executor::StreamExecutor
180 StreamExecutor::StreamExecutor( in StreamExecutor() function in stream_executor::StreamExecutor
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dthunk.h84 se::StreamExecutor* /*executor*/) { in Initialize()
Dgpu_transfer_manager.cc148 se::StreamExecutor* /*executor*/, const Shape& literal_shape, in TransferLiteralFromOutfeed()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc98 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* /*stream_exec*/, in RunHloPasses()
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_event_mgr.h35 class StreamExecutor; variable
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.cc432 [stream, done](se::StreamExecutor*) { in Sync()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc525 std::unique_ptr<HloModule> module, se::StreamExecutor* /*stream_exec*/, in RunHloPasses()