Searched refs:BaseTorchFunction (Results 1 – 1 of 1) sorted by relevance
149 class BaseTorchFunction(torch.Tensor): class407 BaseTorchFunction,506 input = torch.ones(2, 2).as_subclass(BaseTorchFunction)509 self.assertIsInstance(out, BaseTorchFunction)