Home
last modified time | relevance | path

Searched refs:fn_test_scalar_arg (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/
H A Dtest_jit_fuser.py574 def fn_test_scalar_arg(x: torch.Tensor, p: float) -> torch.Tensor: function
579 scripted = self.checkScript(fn_test_scalar_arg, (x, p))
H A Dtest_jit_fuser_te.py898 def fn_test_scalar_arg(x: torch.Tensor, p: float) -> torch.Tensor: function
903 scripted = self.checkScript(fn_test_scalar_arg, (x, p))