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