Searched refs:get_set_tensor (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_indexing.py | 631 def get_set_tensor(indexed, indexer): function
666 assert_set_eq(reference, indexer, get_set_tensor(reference, indexer))
716 assert_set_eq(reference, indexer, get_set_tensor(reference, indexer))
787 assert_set_eq(reference, indexer, get_set_tensor(reference, indexer))
|