Searched refs:ReduceScatterNet (Results 1 – 1 of 1) sorted by relevance
95 class ReduceScatterNet(nn.Cell): class99 super(ReduceScatterNet, self).__init__()203 network = ReduceScatterNet(2, 1, op)