Lines Matching refs:ArrayRef
22 SmallVector<ExprType> computeSuffixProductImpl(ArrayRef<ExprType> sizes,
33 SmallVector<ExprType> computeElementwiseMulImpl(ArrayRef<ExprType> v1,
34 ArrayRef<ExprType> v2) {
45 ExprType linearizeImpl(ArrayRef<ExprType> offsets, ArrayRef<ExprType> basis,
56 ArrayRef<ExprType> strides,
71 SmallVector<int64_t> mlir::computeSuffixProduct(ArrayRef<int64_t> sizes) {
78 SmallVector<int64_t> mlir::computeElementwiseMul(ArrayRef<int64_t> v1,
79 ArrayRef<int64_t> v2) {
83 int64_t mlir::computeSum(ArrayRef<int64_t> basis) {
91 int64_t mlir::computeProduct(ArrayRef<int64_t> basis) {
100 int64_t mlir::linearize(ArrayRef<int64_t> offsets, ArrayRef<int64_t> basis) {
108 ArrayRef<int64_t> strides) {
116 mlir::computeShapeRatio(ArrayRef<int64_t> shape, ArrayRef<int64_t> subShape) {
147 SmallVector<AffineExpr> mlir::computeSuffixProduct(ArrayRef<AffineExpr> sizes) {
154 SmallVector<AffineExpr> mlir::computeElementwiseMul(ArrayRef<AffineExpr> v1,
155 ArrayRef<AffineExpr> v2) {
159 AffineExpr mlir::computeSum(MLIRContext *ctx, ArrayRef<AffineExpr> basis) {
167 AffineExpr mlir::computeProduct(MLIRContext *ctx, ArrayRef<AffineExpr> basis) {
175 AffineExpr mlir::linearize(MLIRContext *ctx, ArrayRef<AffineExpr> offsets,
176 ArrayRef<AffineExpr> basis) {
181 AffineExpr mlir::linearize(MLIRContext *ctx, ArrayRef<AffineExpr> offsets,
182 ArrayRef<int64_t> basis) {
188 ArrayRef<AffineExpr> strides) {
195 ArrayRef<int64_t> strides) {
205 mlir::invertPermutationVector(ArrayRef<int64_t> permutation) {
215 bool mlir::isIdentityPermutation(ArrayRef<int64_t> permutation) {
222 bool mlir::isPermutationVector(ArrayRef<int64_t> interchange) {
235 mlir::computePermutationVector(int64_t permSize, ArrayRef<int64_t> positions,
236 ArrayRef<int64_t> desiredPositions) {
255 SmallVector<int64_t> mlir::dropDims(ArrayRef<int64_t> inputPerm,
256 ArrayRef<int64_t> dropPositions) {
305 ArrayRef<OpFoldResult> strides,
306 ArrayRef<OpFoldResult> indices) {
335 mlir::computeLinearIndex(OpFoldResult sourceOffset, ArrayRef<int64_t> strides,
336 ArrayRef<Value> indices) {
347 static SmallVector<int64_t> padTileShapeToSize(ArrayRef<int64_t> tileShape,
359 ArrayRef<int64_t> shape, ArrayRef<int64_t> tileShape,
360 ArrayRef<int64_t> loopOrder)