Searched refs:DiagNet (Results 1 – 1 of 1) sorted by relevance
291 class DiagNet(nn.Cell): class295 super(DiagNet, self).__init__()402 'block': DiagNet(),