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

  /external/tensorflow/tensorflow/compiler/xla/
executable_run_options.h 91 stream_executor::Stream* host_to_device_stream_ = nullptr; member in class:xla::ExecutableRunOptions
  /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.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

Completed in 979 milliseconds