HomeSort by relevance Sort by last modified time
    Searched refs:host_to_device_stream_ (Results 1 - 9 of 9) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
infeed_manager.cc 27 host_to_device_stream_ = absl::make_unique<se::Stream>(executor);
28 host_to_device_stream_->Init();
37 return host_to_device_stream_.get();
infeed_manager.h 73 // Mutex for serializing the creation of host_to_device_stream_.
77 std::unique_ptr<se::Stream> host_to_device_stream_
  /external/tensorflow/tensorflow/core/common_runtime/
gpu_device_context.h 38 host_to_device_stream_(host_to_device_stream),
45 se::Stream* host_to_device_stream() const { return host_to_device_stream_; }
76 se::Stream* host_to_device_stream_; member in class:tensorflow::GPUDeviceContext
  /external/tensorflow/tensorflow/compiler/jit/
xla_device_context.cc 80 host_to_device_stream_(std::move(host_to_device_stream)),
87 CHECK(host_to_device_stream_ != nullptr);
158 host_to_device_stream_->ThenWaitFor(stream_.get());
162 host_to_device_stream_.get(), literal, xla_tensor->shaped_buffer()));
167 host_to_device_stream_->ThenRecordEvent(event.get());
169 host_to_device_stream_.get());
189 host_to_device_stream_->ThenDoHostCallback([ref]() { ref.Unref(); });
192 host_to_device_stream_->ThenDoHostCallback([ref, done]() {
xla_device_context.h 76 return host_to_device_stream_.get();
90 bool UseMultipleStreams() const { return stream_ != host_to_device_stream_; }
97 std::shared_ptr<se::Stream> host_to_device_stream_; member in class:tensorflow::XlaDeviceContext
xla_device.h 220 std::shared_ptr<se::Stream> host_to_device_stream_ GUARDED_BY(mu_);
xla_device.cc 291 &host_to_device_stream_,
298 host_to_device_stream = host_to_device_stream_;
  /external/tensorflow/tensorflow/compiler/xla/
executable_run_options.cc 50 host_to_device_stream_ = stream;
55 return host_to_device_stream_;
executable_run_options.h 91 stream_executor::Stream* host_to_device_stream_ = nullptr; member in class:xla::ExecutableRunOptions

Completed in 175 milliseconds