Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_activation.h30 class StreamExecutor; variable
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_activation.h30 class StreamExecutor; variable
/external/tensorflow/tensorflow/compiler/xla/stream_executor/gpu/
Dgpu_activation.h30 class StreamExecutor; variable
Dgpu_executor.h45 class StreamExecutor; variable
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dtimer.h29 class StreamExecutor; variable
Devent.h30 class StreamExecutor; variable
Dkernel.h90 class StreamExecutor; variable
Dplatform.h34 class StreamExecutor; variable
Ddevice_memory.h45 class StreamExecutor; variable
Dstream.h74 class StreamExecutor; variable
Dstream_executor_pimpl.cc139 StreamExecutor::StreamExecutor( in StreamExecutor() function in stream_executor::StreamExecutor
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dxla_executor_state.h31 explicit GpuExecutorXLAState(stream_executor::StreamExecutor *) {} in GpuExecutorXLAState()
Dthunk.h100 se::StreamExecutor* /*executor*/) { in Initialize()
/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_event_mgr.h35 class StreamExecutor; variable
/external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/
Dc_api_defn.h31 class StreamExecutor; variable
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dcompiler.cc102 std::unique_ptr<HloModule> hlo_module, se::StreamExecutor* /*stream_exec*/, in RunHloPasses()
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor_test.cc34 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
/external/tensorflow/tensorflow/compiler/xla/tests/
Dllvm_compiler_test.cc58 GpuVersion GetGpuVersion(se::StreamExecutor*) override { in GetGpuVersion()
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device.cc496 stream->ThenEnqueueOnBackgroundThread([stream, done](se::StreamExecutor*) { in Sync()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc820 std::unique_ptr<HloModule> module, se::StreamExecutor* /*stream_exec*/, in RunHloPasses()