Searched refs:set_num_replicas (Results 1 – 4 of 4) sorted by relevance
73 ExecutableBuildOptions& set_num_replicas(int num_replicas);
61 ExecutableBuildOptions& ExecutableBuildOptions::set_num_replicas( in set_num_replicas() function in xla::ExecutableBuildOptions
113 execution_options.set_num_replicas(build_options.num_replicas()); in CreateExecutionOptions()
249 build_options.set_num_replicas(num_replicas);