Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DTensorShape.cpp3640 bool swap_first_and_second_tensors = false; in meshgrid() local
3644 swap_first_and_second_tensors = size >= 2; in meshgrid()
3645 if (swap_first_and_second_tensors) { in meshgrid()
3672 if (swap_first_and_second_tensors) { in meshgrid()
/aosp_15_r20/external/pytorch/torch/_refs/
H A D__init__.py5251 swap_first_and_second_tensors = False
5253 swap_first_and_second_tensors = len(tensors) >= 2
5254 if swap_first_and_second_tensors:
5281 if swap_first_and_second_tensors: