Home
last modified time | relevance | path

Searched refs:service_run_options (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_runner.cc208 ServiceExecutableRunOptions service_run_options = in ExecuteWithDeviceBuffers() local
216 executable->ExecuteOnStreamWrapper(&service_run_options, in ExecuteWithDeviceBuffers()
245 ServiceExecutableRunOptions service_run_options = in ExecuteWithDeviceBuffers() local
251 executable->ExecuteOnStreamWrapper(&service_run_options, in ExecuteWithDeviceBuffers()
279 std::vector<ServiceExecutableRunOptions> service_run_options; in ExecuteReplicated() local
299 service_run_options.emplace_back(GetServiceRunOptionsForDevice( in ExecuteReplicated()
373 executable->ExecuteOnStreams(service_run_options, in ExecuteReplicated()
387 &service_run_options[i], argument_buffer_slices[i], nullptr); in ExecuteReplicated()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dbuffer_comparator.cc145 ServiceExecutableRunOptions service_run_options(run_options); in CompareEqualImpl() local
147 &service_run_options, {&ref_buffer_, &shaped_test_buffer}, nullptr); in CompareEqualImpl()