| /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/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 | 302 std::shared_ptr<se::Stream> host_to_device_stream; in GetDeviceContextLocked() local
314 host_to_device_stream = host_to_device_stream_; in GetDeviceContextLocked()
321 host_to_device_stream = stream_; in GetDeviceContextLocked()
344 stream_, host_to_device_stream, device_to_host_stream, in GetDeviceContextLocked()
351 stream_, std::move(host_to_device_stream), in GetDeviceContextLocked()
|
| D | xla_tpu_device.cc | 227 dst_xla_context->host_to_device_stream()->ThenWaitFor( in TpuDeviceToDeviceCopy()
249 dst_xla_context->host_to_device_stream(), in TpuDeviceToDeviceCopy()
260 dst_xla_context->host_to_device_stream()); in TpuDeviceToDeviceCopy()
|
| D | xla_device_context.cc | 78 std::shared_ptr<se::Stream> host_to_device_stream, in XlaDeviceContext() argument
85 host_to_device_stream_(std::move(host_to_device_stream)), in XlaDeviceContext()
|
| D | xla_launch_util.cc | 376 gpu_device_context->host_to_device_stream()); in SetOutputForConstant()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_on_demand_compiler.cc | 45 if (options.run_options().host_to_device_stream() != nullptr) { in ToC()
46 se_options.host_to_device_stream = in ToC()
48 options.run_options().host_to_device_stream()->implementation()) in ToC()
51 se_options.host_to_device_stream = nullptr; in ToC()
|
| D | tpu_execute.h | 49 stream_executor::Stream* host_to_device_stream,
|
| D | tpu_execute.cc | 421 stream_executor::Stream* host_to_device_stream, in TPUExecute() argument
445 run_options.set_host_to_device_stream(host_to_device_stream); in TPUExecute()
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | executable_run_options.cc | 70 stream_executor::Stream* ExecutableRunOptions::host_to_device_stream() const { in host_to_device_stream() function in xla::ExecutableRunOptions
|
| D | executable_run_options.h | 108 stream_executor::Stream* host_to_device_stream() const;
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | pjrt_stream_executor_client.cc | 421 se::Stream* tuple_table_stream = local_device->host_to_device_stream(); in AllocateDestinationBuffer()
723 local_device->host_to_device_stream(), in BufferFromHostBuffer()
792 local_device->host_to_device_stream(), literal, buffer)); in BufferFromHostBuffer()
797 local_device->host_to_device_stream(), literal, buffer)); in BufferFromHostBuffer()
804 local_device->host_to_device_stream())); in BufferFromHostBuffer()
807 local_device->host_to_device_stream()); in BufferFromHostBuffer()
876 local_device->host_to_device_stream(), in BufferFromHostLiteral()
900 se::Stream* h2d_stream = local_device->host_to_device_stream(); in BufferFromHostLiteral()
1501 se::Stream* stream = local_device->host_to_device_stream(); in MakeTupleHelper()
1750 run_options.set_host_to_device_stream(device_state->host_to_device_stream()); in EnqueueExecution()
|
| D | local_device_state.h | 107 se::Stream* host_to_device_stream() const { in host_to_device_stream() function
|
| /external/tensorflow/tensorflow/stream_executor/tpu/ |
| D | c_api_decl.h | 166 SE_Stream* host_to_device_stream; member
|
| D | tpu_executable_interface.cc | 189 run_options->run_options().host_to_device_stream())); in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 315 ->host_to_device_stream(); in CopyCPUTensorToGPU()
|