Searched refs:device_to_host_stream (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 74 std::shared_ptr<se::Stream> device_to_host_stream, in XlaDeviceContext() argument
81 device_to_host_stream_(std::move(device_to_host_stream)), in XlaDeviceContext()
216 std::shared_ptr<se::Stream> device_to_host_stream; in CopyDeviceTensorToCPU() local
218 device_to_host_stream = device_to_host_stream_; in CopyDeviceTensorToCPU()
227 device_to_host_stream = in CopyDeviceTensorToCPU()
232 xla_tensor->WaitForDefinitionEventOnStream(device_to_host_stream.get()); in CopyDeviceTensorToCPU()
248 device_to_host_stream.get(), xla_tensor->shaped_buffer(), literal, in CopyDeviceTensorToCPU()
249 [ref, xla_tensor, done, device_to_host_stream](xla::Status status) { in CopyDeviceTensorToCPU()
|
| D | xla_device.cc | 287 std::shared_ptr<se::Stream> device_to_host_stream; in GetDeviceContextLocked() local
303 device_to_host_stream = nullptr; in GetDeviceContextLocked()
306 device_to_host_stream = stream_; in GetDeviceContextLocked()
326 std::move(device_to_host_stream), std::move(device_to_device_streams), in GetDeviceContextLocked()
|
| D | xla_device_context.h | 57 std::shared_ptr<se::Stream> device_to_host_stream,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | gpu_device_context.h | 34 se::Stream* device_to_host_stream, in GPUDeviceContext() argument
39 device_to_host_stream_(device_to_host_stream), in GPUDeviceContext()
46 se::Stream* device_to_host_stream() const { return device_to_host_stream_; } in device_to_host_stream() function
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 131 ->device_to_host_stream(); in SetProtoFromGPU()
271 ->device_to_host_stream(); in CopyGPUTensorToCPU()
|