Searched refs:auto_spmd_partitioning_mesh_ids (Results 1 – 12 of 12) sorted by relevance
464 CreateVector(config.auto_spmd_partitioning_mesh_ids(), in ToC()465 &hlo_config.auto_spmd_partitioning_mesh_ids); in ToC()514 MakeSpan(c_config.auto_spmd_partitioning_mesh_ids); in FromC()
311 Int64List auto_spmd_partitioning_mesh_ids; member
107 std::vector<int64_t> auto_spmd_partitioning_mesh_ids() const { in auto_spmd_partitioning_mesh_ids() function
204 for (auto t : build_options.auto_spmd_partitioning_mesh_ids()) { in CreateExecutionOptions()
97 for (auto t : options.auto_spmd_partitioning_mesh_ids()) { in CompileAheadOfTime()
108 for (auto t : execution_options->auto_spmd_partitioning_mesh_ids()) { in CreateModuleConfig()
185 std::vector<int64_t> auto_spmd_partitioning_mesh_ids() const { in auto_spmd_partitioning_mesh_ids() function
97 virtual std::vector<int64_t> auto_spmd_partitioning_mesh_ids() const { in auto_spmd_partitioning_mesh_ids() function
546 for (auto t : execution_options->auto_spmd_partitioning_mesh_ids()) { in CreateModuleConfigFromShape()
254 auto_spmd_partitioning_mesh_ids: List[int]
729 &ExecutableBuildOptions::auto_spmd_partitioning_mesh_ids, in BuildXlaCompilerSubmodule()
502 repeated int64 auto_spmd_partitioning_mesh_ids = 17; field