Searched refs:ReshapeNet6 (Results 1 – 1 of 1) sorted by relevance
286 class ReshapeNet6(nn.Cell): class288 super(ReshapeNet6, self).__init__()378 reshape_net2(_VirtualDatasetCell(ReshapeNet6(((1, 8), (8, 1)))))382 reshape_net2(_VirtualDatasetCell(ReshapeNet6(((1, 8), (8, 2)))))