Searched refs:NetFlatten (Results 1 – 1 of 1) sorted by relevance
25 class NetFlatten(nn.Cell): class27 super(NetFlatten, self).__init__()84 flatten = NetFlatten()89 flatten = NetFlatten()