Searched defs:dims_before (Results 1 – 4 of 4) sorted by relevance
155 int64_t dims_before; member
530 static Tensor restride_src(const Tensor& src, int64_t dims_before, int64_t dims_indexed, in restride_src()544 static Tensor reshape_indexer(const Tensor& index, int64_t dims_before, int64_t dims_after) { in reshape_indexer()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
3842 std::vector<int64_t> dims_before, dims_after; in TEST_F() local