Searched refs:ModelWithDifferentTensorSizes (Results 1 – 1 of 1) sorted by relevance
88 class ModelWithDifferentTensorSizes(torch.nn.Module): class90 super(ModelWithDifferentTensorSizes, self).__init__()356 ModelWithDifferentTensorSizes383 ModelWithDifferentTensorSizes,