Searched refs:set_auto_spmd_partitioning_mesh_ids (Results 1 – 7 of 7) sorted by relevance
110 ExecutableBuildOptions& set_auto_spmd_partitioning_mesh_ids(
94 ExecutableBuildOptions::set_auto_spmd_partitioning_mesh_ids( in set_auto_spmd_partitioning_mesh_ids() function in xla::ExecutableBuildOptions
111 config->set_auto_spmd_partitioning_mesh_ids(mesh_ids); in CreateModuleConfig()
182 void set_auto_spmd_partitioning_mesh_ids(std::vector<int64_t> mesh_ids) { in set_auto_spmd_partitioning_mesh_ids() function
549 module_config.set_auto_spmd_partitioning_mesh_ids(mesh_ids); in CreateModuleConfigFromShape()
516 config.set_auto_spmd_partitioning_mesh_ids(mesh_ids); in FromC()
730 &ExecutableBuildOptions::set_auto_spmd_partitioning_mesh_ids) in BuildXlaCompilerSubmodule()