Searched refs:NetAllFlatten (Results 1 – 1 of 1) sorted by relevance
34 class NetAllFlatten(nn.Cell): class36 super(NetAllFlatten, self).__init__()101 flatten = NetAllFlatten()106 flatten = NetAllFlatten()