Searched refs:py_mesh_mapping (Results 1 – 1 of 1) sorted by relevance
84 pybind11::tuple py_mesh_mapping) { in PyMeshShardingToCpp() argument85 return py::cast<std::vector<MeshDimAssignment>>(py_mesh_mapping); in PyMeshShardingToCpp()