Searched refs:set_num_replicas (Results 1 – 12 of 12) sorted by relevance
62 ExecutableBuildOptions& ExecutableBuildOptions::set_num_replicas( in set_num_replicas() function in xla::ExecutableBuildOptions125 execution_options.set_num_replicas(build_options.num_replicas()); in CreateExecutionOptions()
81 ExecutableBuildOptions& set_num_replicas(int num_replicas);
77 execution_options.set_num_replicas(options.replica_count()); in CompileAheadOfTime()
97 metadata.set_num_replicas(num_replicas); in CompilationCacheKey()
162 build_options.set_num_replicas(num_replicas); in Compile()
464 options.executable_build_options.set_num_replicas(num_replicas); in BuildXlaCompilerSubmodule()538 &ExecutableBuildOptions::set_num_replicas) in BuildXlaCompilerSubmodule()
400 compile_options.executable_build_options.set_num_replicas(1); in SendShutdownOutfeedHeader()
39 compile_options.executable_build_options.set_num_replicas(1); in CompileAndExecute()
354 r.mutable_compile()->set_num_replicas(num_replicas); in CompileProgram()
843 req->mutable_compile()->set_num_replicas(num_replicas); in CompileProgram()
306 metadata->set_num_replicas(num_replicas); in SetMetadataProtoFromClusterFuncOp()
2322 proto.set_num_replicas(params_info.NumReplicas()); in BuildCompileNode()