Searched refs:host_to_device_stream (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | gpu_device_context.h | 36 se::Stream* host_to_device_stream, in GPUDeviceContext() argument
44 host_to_device_stream_(host_to_device_stream), in GPUDeviceContext()
55 se::Stream* host_to_device_stream() const { return host_to_device_stream_; } in host_to_device_stream() function
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | local_client.cc | 330 local_device->host_to_device_stream()->parent(), scoped_buffer)) { in FromHostBuffer()
331 local_device->host_to_device_stream()->ThenWaitFor( in FromHostBuffer()
352 local_device->host_to_device_stream(), buffer)); in FromHostBuffer()
368 local_device->host_to_device_stream(), literal, buffer)); in FromHostBuffer()
373 local_device->host_to_device_stream(), literal, buffer)); in FromHostBuffer()
378 .ThenAllocateAndRecordEvent(local_device->host_to_device_stream()) in FromHostBuffer()
385 std::move(event), local_device->host_to_device_stream()); in FromHostBuffer()
388 local_device->ThenRelease(local_device->host_to_device_stream(), in FromHostBuffer()
393 local_device->host_to_device_stream(), in FromHostBuffer()
448 local_device->host_to_device_stream()->parent(), shaped_buffer)) { in MakeTuple()
[all …]
|
| D | local_device_state.h | 61 se::Stream* host_to_device_stream() const { in host_to_device_stream() function
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.h | 56 std::shared_ptr<se::Stream> host_to_device_stream,
75 se::Stream* host_to_device_stream() const { in host_to_device_stream() function
|
| D | xla_device.cc | 301 std::shared_ptr<se::Stream> host_to_device_stream; in GetDeviceContextLocked() local
313 host_to_device_stream = host_to_device_stream_; in GetDeviceContextLocked()
320 host_to_device_stream = stream_; in GetDeviceContextLocked()
343 stream_, host_to_device_stream, device_to_host_stream, in GetDeviceContextLocked()
350 stream_, std::move(host_to_device_stream), in GetDeviceContextLocked()
|
| D | xla_device_context.cc | 76 std::shared_ptr<se::Stream> host_to_device_stream, in XlaDeviceContext() argument
83 host_to_device_stream_(std::move(host_to_device_stream)), in XlaDeviceContext()
|
| D | xla_launch_util.cc | 451 gpu_device_context->host_to_device_stream()); in PopulateOutputs()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | executable_run_options.cc | 68 stream_executor::Stream* ExecutableRunOptions::host_to_device_stream() const { in host_to_device_stream() function in xla::ExecutableRunOptions
|
| D | executable_run_options.h | 94 stream_executor::Stream* host_to_device_stream() const;
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 315 ->host_to_device_stream(); in CopyCPUTensorToGPU()
|