Searched refs:arange_start_end (Results 1 – 1 of 1) sorted by relevance
441 def arange_start_end(x): function456 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))