Searched defs:set_auto_spmd_partitioning_mesh_ids (Results 1 – 2 of 2) sorted by relevance
94 ExecutableBuildOptions::set_auto_spmd_partitioning_mesh_ids( in set_auto_spmd_partitioning_mesh_ids() function in xla::ExecutableBuildOptions
182 void set_auto_spmd_partitioning_mesh_ids(std::vector<int64_t> mesh_ids) { in set_auto_spmd_partitioning_mesh_ids() function