Searched defs:mesh_ids (Results 1 – 5 of 5) sorted by relevance
107 std::vector<int64_t> mesh_ids; in CreateModuleConfig() local
182 void set_auto_spmd_partitioning_mesh_ids(std::vector<int64_t> mesh_ids) { in set_auto_spmd_partitioning_mesh_ids()
545 std::vector<int64_t> mesh_ids; in CreateModuleConfigFromShape() local
95 std::vector<int64_t> mesh_ids) { in set_auto_spmd_partitioning_mesh_ids()
515 std::vector<int64_t> mesh_ids(mesh_ids_span.begin(), mesh_ids_span.end()); in FromC() local