Searched refs:if_function (Results 1 – 1 of 1) sorted by relevance
165 def if_function(inp: torch.Tensor) -> Any: function171 self.checkScript(if_function, (torch.randn(5),))