Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/include/mlir/Dialect/Vector/Utils/
H A DVectorUtils.h45 SmallVector<int64_t, 4> computeStrides(ArrayRef<int64_t> shape,
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Utils/
H A DVectorUtils.cpp54 SmallVector<int64_t, 4> mlir::computeStrides(ArrayRef<int64_t> shape, in computeStrides() function in mlir
/llvm-project-15.0.7/mlir/lib/Conversion/MathToLibm/
H A DMathToLibm.cpp77 SmallVector<int64_t> strides = computeStrides(shape, ones); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/
H A DVectorUnrollDistribute.cpp34 computeStrides(originalShape, targetShape); in getVectorOffset()
H A DVectorTransforms.cpp436 auto prunedInStrides = computeStrides(prunedInShape, ones); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/Dialect/Math/Transforms/
H A DPolynomialApproximation.cpp133 auto strides = computeStrides(iterationDims, ones); in handleMultidimensionalVectors()