Home
last modified time | relevance | path

Searched refs:currentDim (Results 1 – 1 of 1) sorted by relevance

/external/llvm-project/mlir/lib/Dialect/Linalg/IR/
DLinalgOps.cpp744 unsigned currentDim = 0; in computeReshapeCollapsedType() local
748 AffineExpr stride = strides[currentDim + dim - 1]; in computeReshapeCollapsedType()
749 if (!isReshapableDimBand(currentDim, dim, sizes, strides)) { in computeReshapeCollapsedType()
754 size *= sizes[currentDim + d]; in computeReshapeCollapsedType()
758 currentDim += dim; in computeReshapeCollapsedType()
938 unsigned currentDim = 0; in computeTensorReshapeCollapsedType() local
941 auto band = shape.slice(currentDim, dim); in computeTensorReshapeCollapsedType()
947 size *= shape[currentDim + d]; in computeTensorReshapeCollapsedType()
949 currentDim += dim; in computeTensorReshapeCollapsedType()