Home
last modified time | relevance | path

Searched refs:set_host_to_device_stream (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dexecutable_run_options.cc64 ExecutableRunOptions& ExecutableRunOptions::set_host_to_device_stream( in set_host_to_device_stream() function in xla::ExecutableRunOptions
Dexecutable_run_options.h106 ExecutableRunOptions& set_host_to_device_stream(
/external/tensorflow/tensorflow/core/tpu/
Dtpu_execute.cc445 run_options.set_host_to_device_stream(host_to_device_stream); in TPUExecute()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc1931 run_options.set_host_to_device_stream(device_state->host_to_device_stream()); in EnqueueExecution()