Searched refs:mesh_mapping (Results 1 – 2 of 2) sorted by relevance
119 std::vector<MeshDimAssignment> mesh_mapping) in ShardingSpec() argument121 mesh_mapping_(std::move(mesh_mapping)) {} in ShardingSpec()
89 std::vector<MeshDimAssignment> mesh_mapping) { in CppMeshMappingToPy() argument90 py::tuple result(mesh_mapping.size()); in CppMeshMappingToPy()92 for (auto& value : mesh_mapping) { in CppMeshMappingToPy()