Home
last modified time | relevance | path

Searched refs:AllocateStream (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.h151 bool AllocateStream(Stream *stream) override { return true; } in AllocateStream() function
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h129 bool AllocateStream(Stream *stream) override;
Dhost_gpu_executor.cc161 bool HostExecutor::AllocateStream(Stream *stream) { return true; } in AllocateStream() function in stream_executor::host::HostExecutor
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_internal.h250 virtual bool AllocateStream(Stream *stream) = 0;
Dstream_executor_pimpl.cc774 bool StreamExecutor::AllocateStream(Stream *stream) { in AllocateStream() function in stream_executor::StreamExecutor
776 if (!implementation_->AllocateStream(stream)) { in AllocateStream()
Dstream_executor_pimpl.h587 bool AllocateStream(Stream *subject);
Dstream.cc298 if (parent_->AllocateStream(this)) { in Init()
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h158 bool AllocateStream(Stream* stream) override;
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc608 bool GpuExecutor::AllocateStream(Stream* stream) { in AllocateStream() function in stream_executor::gpu::GpuExecutor
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc727 bool GpuExecutor::AllocateStream(Stream* stream) { in AllocateStream() function in stream_executor::gpu::GpuExecutor