Home
last modified time | relevance | path

Searched defs:AdvancedIndex (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/
H A DIndexingUtils.h148 struct AdvancedIndex { struct
151 Tensor src;
152 std::vector<Tensor> indices;
153 DimVector indexed_sizes;
154 DimVector indexed_strides;
155 int64_t dims_before;
156 int64_t dims_after;
H A DTensorAdvancedIndexing.cpp553 AdvancedIndex::AdvancedIndex(const Tensor& src, TensorList indices_list) in AdvancedIndex() function in at::native::AdvancedIndex