Searched refs:py_mesh_mapping (Results 1 – 1 of 1) sorted by relevance
180 pybind11::iterable py_mesh_mapping) in ShardingSpec() argument183 xla::IterableToVector<MeshDimAssignment>(py_mesh_mapping)) {} in ShardingSpec()