Searched refs:test_scalar_cast (Results 1 – 1 of 1) sorted by relevance
229 def test_scalar_cast(x): function233 graph = torch.jit.script(test_scalar_cast).graph235 self.checkScript(test_scalar_cast, (torch.tensor(1.0),))236 self.checkScript(test_scalar_cast, (torch.tensor(1),))