Searched refs:AllReduceNet (Results 1 – 5 of 5) sorted by relevance
31 class AllReduceNet(nn.Cell): class33 super(AllReduceNet, self).__init__()51 net = AllReduceNet()
32 class AllReduceNet(nn.Cell): class34 super(AllReduceNet, self).__init__()51 net = AllReduceNet()
35 class AllReduceNet(nn.Cell): class37 super(AllReduceNet, self).__init__()65 net = AllReduceNet()
35 class AllReduceNet(nn.Cell): class37 super(AllReduceNet, self).__init__()55 net = AllReduceNet()
45 class AllReduceNet(nn.Cell): class49 super(AllReduceNet, self).__init__()150 network = AllReduceNet(2, 1, op)