Searched defs:dims_indexed (Results 1 – 3 of 3) sorted by relevance
530 static Tensor restride_src(const Tensor& src, int64_t dims_before, int64_t dims_indexed, in restride_src()556 int64_t dims_before = 0, dims_after = 0, dims_indexed = 0; in AdvancedIndex() local
351 int64_t dims_before = 0, dims_indexed = 0; in compute_indexed_shape() local
2289 int64_t dims_indexed = 0; in getsetitem() local