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