Searched refs:loopBounds (Results 1 – 6 of 6) sorted by relevance
133 SmallVector<int64_t, 4> loopBounds = linalgOp.computeStaticLoopSizes(); in matchAndRewrite() local164 SmallVector<uint64_t> indices(loopBounds.size(), 0); in matchAndRewrite()165 SmallVector<uint64_t> dstIndices(loopBounds.size(), 0); in matchAndRewrite()167 numInputs, SmallVector<uint64_t>(loopBounds.size(), 0)); in matchAndRewrite()185 for (int dim = loopBounds.size() - 1; dim >= 0; --dim) { in matchAndRewrite()186 indices[dim] = totalCount % loopBounds[dim]; in matchAndRewrite()187 totalCount /= loopBounds[dim]; in matchAndRewrite()190 for (int dim = loopBounds.size() - 1; dim >= 0; --dim) { in matchAndRewrite()
96 inline constexpr auto loopBounds(decltype(scalarExpr) &p) { in loopBounds() function100 template <typename PA> inline constexpr auto loopBounds(const PA &p) { in loopBounds() function
273 (construct<LoopControl>(loopBounds(scalarExpr)) ||
63 maybe(integerTypeSpec / "::"), loopBounds(scalarIntExpr)))
273 constexpr auto ioImpliedDoControl{loopBounds(scalarIntExpr)};
825 maybe(integerTypeSpec / "::"), loopBounds(scalarIntConstantExpr))))