Searched refs:GradWrapCumSum (Results 1 – 1 of 1) sorted by relevance
316 class GradWrapCumSum(nn.Cell): class320 super(GradWrapCumSum, self).__init__()397 'block': GradWrapCumSum(NetWithLossCumSum(NetCumSum())),