Searched defs:ExecutableRunOptions (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
H A Dexecutable_run_options.h19 // Intentionally forward declared so that ExecutableRunOptions can be linked
47 class ExecutableRunOptions { class in namespace:xla
50 ExecutableRunOptions& set_allocator(DeviceMemoryAllocator* allocator);
58 ExecutableRunOptions& set_device_ordinal(int device_ordinal);
64 ExecutableRunOptions& set_stream(perftools::gputools::Stream* stream);
69 ExecutableRunOptions& set_inter_op_thread_pool(
75 ExecutableRunOptions& set_intra_op_thread_pool(
81 ExecutableRunOptions& set_execution_profile(ExecutionProfile* profile);
83 ExecutableRunOptions& set_device_assignment(

Completed in 276 milliseconds