Searched refs:can_use_expanded_index_path (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | TensorAdvancedIndexing.cpp | 1572 static bool can_use_expanded_index_path( in can_use_expanded_index_path() function
1642 if (can_use_expanded_index_path(result, dim, index, self, /*is_scatter_like=*/false)) { in TORCH_IMPL_FUNC()
1908 if (can_use_expanded_index_path(mut_out, dim, index, src, /*is_scatter_like*/true)) { in TORCH_IMPL_FUNC()
1933 if (can_use_expanded_index_path(out, dim, index, src, /*is_scatter_like*/true)) { in TORCH_IMPL_FUNC()
|