Searched refs:GradWrapSub (Results 1 – 1 of 1) sorted by relevance
235 class GradWrapSub(nn.Cell): class239 super(GradWrapSub, self).__init__()407 'block': GradWrapSub(NetWithLossSub(SubNet())),