Searched refs:check_dynamic_indexing (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/jit/ |
| H A D | test_python_builtins.py | 214 def check_dynamic_indexing(indexing, tensor, value1, value2): function
280 check_dynamic_indexing("[i + j]", consec((3, 3)), 0, 1)
281 check_dynamic_indexing("[i:j, i]", consec((3, 3, 2)), 0, 2)
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_jit.py | 4989 def check_dynamic_indexing(indexing, tensor, value1, value2): function
5071 check_dynamic_indexing("[i + j]", consec((3, 3)), 0, 1)
5072 check_dynamic_indexing("[i:j, i]", consec((3, 3, 2)), 0, 2)
|