Searched refs:device_to_host_stream (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.cc | 79 std::shared_ptr<se::Stream> device_to_host_stream, in XlaDeviceContext() argument
86 device_to_host_stream_(std::move(device_to_host_stream)), in XlaDeviceContext()
224 std::shared_ptr<se::Stream> device_to_host_stream; in CopyDeviceTensorToCPU() local
226 device_to_host_stream = device_to_host_stream_; in CopyDeviceTensorToCPU()
235 device_to_host_stream = in CopyDeviceTensorToCPU()
240 xla_tensor->WaitForDefinitionEventOnStream(device_to_host_stream.get()); in CopyDeviceTensorToCPU()
258 device_to_host_stream.get(), xla_tensor->shaped_buffer(), literal, in CopyDeviceTensorToCPU()
259 [this, ref, xla_tensor, done, device_to_host_stream, in CopyDeviceTensorToCPU()
278 !device_to_host_stream->RefreshStatus().ok()) { in CopyDeviceTensorToCPU()
283 [device_to_host_stream] {}); in CopyDeviceTensorToCPU()
|
| D | xla_device.cc | 303 std::shared_ptr<se::Stream> device_to_host_stream; in GetDeviceContextLocked() local
319 device_to_host_stream = nullptr; in GetDeviceContextLocked()
322 device_to_host_stream = stream_; in GetDeviceContextLocked()
344 stream_, host_to_device_stream, device_to_host_stream, in GetDeviceContextLocked()
352 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 | 37 se::Stream* device_to_host_stream, in GPUDeviceContext() argument
45 device_to_host_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/core/common_runtime/gpu/ |
| D | gpu_util.cc | 129 ->device_to_host_stream(); in SetProtoFromGPU()
270 ->device_to_host_stream(); in CopyGPUTensorToCPU()
|