Searched refs:sharded_axis (Results 1 – 1 of 1) sorted by relevance
379 py::class_<ShardedAxis> sharded_axis(pmap_lib, "ShardedAxis"); in BuildPmapSubmodule() local380 sharded_axis.def(py::init<int>()).def_readonly("axis", &ShardedAxis::axis); in BuildPmapSubmodule()381 sharded_axis in BuildPmapSubmodule()