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

  /external/tensorflow/tensorflow/core/common_runtime/
gpu_device_context.h 33 se::Stream* host_to_device_stream,
38 host_to_device_stream_(host_to_device_stream),
45 se::Stream* host_to_device_stream() const { return host_to_device_stream_; } function in class:tensorflow::GPUDeviceContext
  /external/tensorflow/tensorflow/compiler/xla/
executable_run_options.cc 54 stream_executor::Stream* ExecutableRunOptions::host_to_device_stream() const { function in class:xla::ExecutableRunOptions
  /external/tensorflow/tensorflow/compiler/jit/
xla_device_context.h 56 std::shared_ptr<se::Stream> host_to_device_stream,
75 se::Stream* host_to_device_stream() const { function in class:tensorflow::XlaDeviceContext
xla_device.cc 286 std::shared_ptr<se::Stream> host_to_device_stream; local
290 TF_RETURN_IF_ERROR(EnsureStreamOkLocked(backend, "host_to_device_stream",
298 host_to_device_stream = host_to_device_stream_;
305 host_to_device_stream = stream_;
325 stream_, std::move(host_to_device_stream),

Completed in 451 milliseconds