Searched refs:TensorSplitModel (Results 1 – 1 of 1) sorted by relevance
6397 class TensorSplitModel(torch.nn.Module): class6411 self.run_test(TensorSplitModel(), torch.randn(5, 4, 3))6415 class TensorSplitModel(torch.nn.Module): class6419 self.run_test(TensorSplitModel(), torch.randn(1, 2, 3))6423 class TensorSplitModel(torch.nn.Module): class6430 TensorSplitModel(),