Searched refs:segment_ids_rank (Results 1 – 1 of 1) sorted by relevance
4877 int64_t segment_ids_rank = segment_ids_type.getRank(); in matchAndRewrite() local4887 auto suffix = data_type.getShape().drop_front(segment_ids_rank); in matchAndRewrite()4903 int64_t index_vector_dim = segment_ids_rank; in matchAndRewrite()4907 GetI64ElementsAttrForSeq(segment_ids_rank, data_rank, &rewriter), in matchAndRewrite()