Searched refs:cpp_sharding (Results 1 – 1 of 1) sorted by relevance
44 std::vector<AvalDimSharding> cpp_sharding; in PyShardingToCpp() local45 cpp_sharding.reserve(py_sharding.size()); in PyShardingToCpp()48 cpp_sharding.push_back(NoSharding()); in PyShardingToCpp()50 cpp_sharding.push_back(py::cast<Chunked>(value)); in PyShardingToCpp()52 cpp_sharding.push_back(py::cast<Unstacked>(value)); in PyShardingToCpp()61 return cpp_sharding; in PyShardingToCpp()