Searched refs:AllocateStream (Results 1 – 14 of 14) sorted by relevance
116 bool AllocateStream(Stream *stream) override;
194 bool HostExecutor::AllocateStream(Stream *stream) { return true; } in AllocateStream() function in stream_executor::host::HostExecutor
151 bool AllocateStream(Stream *stream) override { return true; } in AllocateStream() function
73 bool AllocateStream(Stream* stream) override;
89 bool TpuExecutor::AllocateStream(Stream* stream) { in AllocateStream() function in tensorflow::tpu::TpuExecutor
326 streams_[core_id] = AllocateStream(core_id); in GrpcTpuDriver()331 host_stream_ = AllocateStream(-1); in GrpcTpuDriver()430 std::unique_ptr<GrpcTpuStream> AllocateStream(int32_t core_id);989 std::unique_ptr<GrpcTpuStream> GrpcTpuDriver::AllocateStream(int32_t id) { in AllocateStream() function in tpu_driver::__anon4073f9c10111::GrpcTpuDriver
254 virtual bool AllocateStream(Stream *stream) = 0;
762 bool StreamExecutor::AllocateStream(Stream *stream) { in AllocateStream() function in stream_executor::StreamExecutor764 if (!implementation_->AllocateStream(stream)) { in AllocateStream()
613 bool AllocateStream(Stream *stream);
305 if (parent_->AllocateStream(this)) { in Init()
159 bool AllocateStream(Stream* stream) override;
607 bool GpuExecutor::AllocateStream(Stream* stream) { in AllocateStream() function in stream_executor::gpu::GpuExecutor
478 bool AllocateStream(Stream* stream) override { in AllocateStream() function in stream_executor::__anon00950a620111::CStreamExecutor
699 bool GpuExecutor::AllocateStream(Stream* stream) { in AllocateStream() function in stream_executor::gpu::GpuExecutor