Home
last modified time | relevance | path

Searched defs:AllocateStream (Results 1 – 8 of 8) 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/tpu/
Dtpu_executor.cc89 bool TpuExecutor::AllocateStream(Stream* stream) { in AllocateStream() function in tensorflow::tpu::TpuExecutor
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.cc194 bool HostExecutor::AllocateStream(Stream *stream) { return true; } in AllocateStream() function in stream_executor::host::HostExecutor
/external/tensorflow/tensorflow/c/experimental/stream_executor/
Dstream_executor.cc478 bool AllocateStream(Stream* stream) override { in AllocateStream() function in stream_executor::__anon00950a620111::CStreamExecutor
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc699 bool GpuExecutor::AllocateStream(Stream* stream) { in AllocateStream() function in stream_executor::gpu::GpuExecutor
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc607 bool GpuExecutor::AllocateStream(Stream* stream) { in AllocateStream() function in stream_executor::gpu::GpuExecutor
/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc762 bool StreamExecutor::AllocateStream(Stream *stream) { in AllocateStream() function in stream_executor::StreamExecutor
/external/tensorflow/tensorflow/compiler/xla/python/tpu_driver/
Dgrpc_tpu_driver.cc989 std::unique_ptr<GrpcTpuStream> GrpcTpuDriver::AllocateStream(int32_t id) { in AllocateStream() function in tpu_driver::__anon4073f9c10111::GrpcTpuDriver