Home
last modified time | relevance | path

Searched defs:reduc (Results 1 – 4 of 4) sorted by relevance

/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/
H A DLoopEmitter.cpp473 MutableArrayRef<Value> reduc, bool isParallel) { in emitForLoopOverTensorAtLvl() argument
526 MutableArrayRef<Value> reduc, boo in emitWhileLoopOverTensorsAtLvls() argument
620 enterCoIterationOverTensorsAtLvls(OpBuilder & builder,Location loc,ArrayRef<TensorLevel> tidLvls,MutableArrayRef<Value> reduc,bool tryParallel,bool needsUniv) enterCoIterationOverTensorsAtLvls() argument
732 exitForLoop(RewriterBase & rewriter,Location loc,MutableArrayRef<Value> reduc) exitForLoop() argument
808 exitWhileLoop(OpBuilder & builder,Location loc,MutableArrayRef<Value> reduc) exitWhileLoop() argument
865 exitCurrentLoop(RewriterBase & rewriter,Location loc,MutableArrayRef<Value> reduc) exitCurrentLoop() argument
[all...]
H A DSparseTensorIterator.cpp1218 OpBuilder &b, Location l, ValueRange reduc, TraverseBuilder builder) const { in inflateSubSectTree()
1272 ValueRange reduc) { in inflateSubSectTree()
1335 SmallVector<Value, 3> reduc = { in genInitImpl() local
1344 ValueRange reduc) -> scf::ValueVector { in genInitImpl()
/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseTensorRewriting.cpp158 SmallVector<Value> reduc = op.getInitArgs(); in genForeachOnSparseConstant() local
924 __anon1bc5ecbb0a02(OpBuilder &builder, Location loc, ValueRange srcLcvs, Value v, ValueRange reduc) matchAndRewrite() argument
1029 __anon1bc5ecbb0b02(OpBuilder &builder, Location loc, ValueRange srcLcvs, Value v, ValueRange reduc) matchAndRewrite() argument
1216 __anon1bc5ecbb0d02(OpBuilder &builder, Location loc, ValueRange dcvs, Value v, ValueRange reduc) matchAndRewrite() argument
1303 __anon1bc5ecbb0e02(OpBuilder &builder, Location loc, ValueRange dcvs, Value v, ValueRange reduc) matchAndRewrite() argument
1370 SmallVector<Value> reduc = op.getInitArgs(); matchAndRewrite() local
1540 __anon1bc5ecbb0f02(OpBuilder &builder, Location loc, ValueRange dcvs, Value v, ValueRange reduc) matchAndRewrite() argument
[all...]
H A DSparsification.cpp836 __anon93cfebb60502(MutableArrayRef<Value> reduc) genCoIteration() argument
1213 __anon93cfebb60a02(MutableArrayRef<Value> reduc) endLoop() argument