Home
last modified time | relevance | path

Searched refs:execute_options (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
Dmultiple_devices_on_host_test.cc40 xla::ExecutableRunOptions execute_options; in CompileAndExecute() local
41 execute_options.set_intra_op_thread_pool( in CompileAndExecute()
43 execute_options.set_device_ordinal(device_ordinal); in CompileAndExecute()
44 execute_options.set_allocator( in CompileAndExecute()
49 executable->Run(absl::Span<const ShapedBuffer* const>(), execute_options); in CompileAndExecute()
/external/tensorflow/tensorflow/compiler/xla/python/
Doutfeed_receiver.cc410 ExecuteOptions execute_options; in SendShutdownOutfeedHeader() local
413 executable->Execute({{}}, execute_options)); in SendShutdownOutfeedHeader()
Doutfeed_receiver_test.cc48 ExecuteOptions execute_options; in CompileAndExecute() local
51 executable->Execute({{}}, execute_options)); in CompileAndExecute()