Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_tracer.py441 def arange_start_end(x): function
456 traced_arange_start_end = torch.jit.trace(arange_start_end, x)
457 self.assertEqual(traced_arange_start_end(y), arange_start_end(y))
458 self.assertEqual(traced_arange_start_end(x), arange_start_end(x))