| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_context.h | 33 int stream_id, se::Stream* stream, se::Stream* host_to_device_stream, in PluggableDeviceContext() argument
38 host_to_device_stream_(host_to_device_stream), in PluggableDeviceContext()
45 se::Stream* host_to_device_stream() const { return host_to_device_stream_; } in host_to_device_stream() function
|
| D | pluggable_device_util.cc | 237 ->host_to_device_stream(); in CopyCPUTensorToPluggableDevice()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | gpu_device_context.h | 36 se::Stream* host_to_device_stream, in GPUDeviceContext() argument
45 host_to_device_stream_(host_to_device_stream), in GPUDeviceContext()
57 se::Stream* host_to_device_stream() const { return host_to_device_stream_; } in host_to_device_stream() function
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/tpu/ |
| D | tpu_executable.cc | 30 if (options.run_options().host_to_device_stream() != nullptr) { in ToC()
31 se_options.host_to_device_stream = in ToC()
33 options.run_options().host_to_device_stream()->implementation()) in ToC()
36 se_options.host_to_device_stream = nullptr; in ToC()
|
| D | c_api_decl.h | 163 SE_Stream* host_to_device_stream; member
|
| D | tpu_executable_interface.cc | 204 run_options->run_options().host_to_device_stream())); in ExecuteAsyncOnStream()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_device_context.h | 58 std::shared_ptr<se::Stream> host_to_device_stream,
77 se::Stream* host_to_device_stream() const { in host_to_device_stream() function
|
| D | xla_device.cc | 331 std::shared_ptr<se::Stream> host_to_device_stream; in GetDeviceContextLocked() local
343 host_to_device_stream = host_to_device_stream_; in GetDeviceContextLocked()
350 host_to_device_stream = stream_; in GetDeviceContextLocked()
371 stream_, host_to_device_stream, device_to_host_stream, in GetDeviceContextLocked()
|
| D | xla_tpu_device.cc | 235 dst_xla_context->host_to_device_stream()->ThenWaitFor( in TpuDeviceToDeviceCopy()
257 dst_xla_context->host_to_device_stream(), in TpuDeviceToDeviceCopy()
268 dst_xla_context->host_to_device_stream()); in TpuDeviceToDeviceCopy()
|
| D | xla_device_context.cc | 89 std::shared_ptr<se::Stream> host_to_device_stream, in XlaDeviceContext() argument
96 host_to_device_stream_(std::move(host_to_device_stream)), in XlaDeviceContext()
|
| D | xla_launch_util.cc | 411 gpu_device_context->host_to_device_stream()); in SetOutputForConstant()
|
| /external/tensorflow/tensorflow/core/tpu/ |
| D | tpu_execute.h | 49 stream_executor::Stream* host_to_device_stream,
|
| D | tpu_execute.cc | 425 stream_executor::Stream* host_to_device_stream, in TPUExecute() argument
449 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 | local_device_state.h | 108 se::Stream* host_to_device_stream() const { in host_to_device_stream() function
|
| D | pjrt_stream_executor_client.cc | 431 se::Stream* tuple_table_stream = local_device->host_to_device_stream(); in AllocateDestinationBuffer()
779 local_device->host_to_device_stream(), in BufferFromHostBuffer()
871 local_device->host_to_device_stream(), literal, buffer)); in BufferFromHostBuffer()
878 local_device->host_to_device_stream(), literal, buffer)); in BufferFromHostBuffer()
885 local_device->host_to_device_stream())); in BufferFromHostBuffer()
888 local_device->host_to_device_stream(), in BufferFromHostBuffer()
955 local_device->host_to_device_stream(), in BufferFromHostLiteral()
979 se::Stream* h2d_stream = local_device->host_to_device_stream(); in BufferFromHostLiteral()
1695 se::Stream* stream = local_device->host_to_device_stream(); in MakeTupleHelper()
1977 run_options.set_host_to_device_stream(device_state->host_to_device_stream()); in EnqueueExecution()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_util.cc | 327 ->host_to_device_stream(); in CopyCPUTensorToGPU()
|