Home
last modified time | relevance | path

Searched defs:device_to_host_stream (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dgpu_device_context.h37 se::Stream* device_to_host_stream, in GPUDeviceContext()
56 se::Stream* device_to_host_stream() const { return device_to_host_stream_; } in device_to_host_stream() function
/external/tensorflow/tensorflow/compiler/jit/
Dxla_device_context.cc79 std::shared_ptr<se::Stream> device_to_host_stream, in XlaDeviceContext()
224 std::shared_ptr<se::Stream> device_to_host_stream; in CopyDeviceTensorToCPU() local
Dxla_device.cc303 std::shared_ptr<se::Stream> device_to_host_stream; in GetDeviceContextLocked() local