Home
last modified time | relevance | path

Searched refs:set_launch_id (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_module_util.cc102 config->set_launch_id(execution_options->launch_id()); in CreateModuleConfig()
Dhlo_module_config.h122 void set_launch_id(uint64 launch_id) { launch_id_ = launch_id; } in set_launch_id() function
/external/tensorflow/tensorflow/compiler/xla/
Dexecutable_run_options.h133 ExecutableRunOptions& set_launch_id(int32 launch_id) { in set_launch_id() function
/external/tensorflow/tensorflow/core/tpu/kernels/
Dtpu_compile_op_support.cc105 config->set_launch_id(*launch_id); in CreateModuleConfig()
/external/tensorflow/tensorflow/stream_executor/tpu/
Dc_api_conversions.cc457 config.set_launch_id(c_config.launch_id); in FromC()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dpjrt_stream_executor_client.cc1758 run_options.set_launch_id(options.launch_id); in EnqueueExecution()