Searched refs:AlltoAllNet (Results 1 – 1 of 1) sorted by relevance
110 class AlltoAllNet(nn.Cell): class114 super(AlltoAllNet, self).__init__()239 network = AlltoAllNet(2, 1)