Searched refs:ReshapeNet4 (Results 1 – 1 of 1) sorted by relevance
256 class ReshapeNet4(nn.Cell): class258 super(ReshapeNet4, self).__init__()349 reshape_net2(_VirtualDatasetCell(ReshapeNet4(((1, 8), (8, 1)))))360 reshape_net2(_VirtualDatasetCell(ReshapeNet4(((1, 8), (8, 2)))))