Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/
H A Dtest_nestedtensor.py116 def noncontiguous_to_padded_tensor(input, shape=None): function
1160 noncontiguous_to_padded_tensor(nt_noncontiguous),
2306 ptT_from_ntT = noncontiguous_to_padded_tensor(ntT)
2377 ptT_from_ntT = noncontiguous_to_padded_tensor(ntT)
2386 ptT_from_ntT = noncontiguous_to_padded_tensor(ntT)
2435 self.assertEqual(noncontiguous_to_padded_tensor(nt1), pt1)
2454 ptT_from_ntT = noncontiguous_to_padded_tensor(ntT)
2464 ptT_from_ntT = noncontiguous_to_padded_tensor(ntT)
2513 self.assertEqual(noncontiguous_to_padded_tensor(nt1), pt1)