Home
last modified time | relevance | path

Searched defs:xPerm (Results 1 – 3 of 3) sorted by relevance

/llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/
H A DSparseBufferRewriting.cpp55 StringRef namePrefix, AffineMap xPerm, in getMangledSortHelperFuncName() argument
80 StringRef namePrefix, AffineMap xPerm, uint64_t ny, ValueRange operands, in getMangledSortHelperFunc() argument
110 OpBuilder &builder, Location loc, ValueRange args, AffineMap xPerm, in forEachIJPairInXs() argument
130 forEachIJPairInAllBuffers(OpBuilder & builder,Location loc,ValueRange args,AffineMap xPerm,uint64_t ny,function_ref<void (uint64_t,Value,Value,Value)> bodyBuilder) forEachIJPairInAllBuffers() argument
167 createSwap(OpBuilder & builder,Location loc,ValueRange args,AffineMap xPerm,uint64_t ny) createSwap() argument
181 createInlinedCompareImplementation(OpBuilder & builder,Location loc,ValueRange args,AffineMap xPerm,uint64_t ny,function_ref<Value (OpBuilder &,Location,Value,Value,Value,bool,bool)> compareBuilder) createInlinedCompareImplementation() argument
251 createInlinedEqCompare(OpBuilder & builder,Location loc,ValueRange args,AffineMap xPerm,uint64_t ny,uint32_t nTrailingP=0) createInlinedEqCompare() argument
305 createInlinedLessThan(OpBuilder & builder,Location loc,ValueRange args,AffineMap xPerm,uint64_t ny,uint32_t nTrailingP=0) createInlinedLessThan() argument
328 createBinarySearchFunc(OpBuilder & builder,ModuleOp module,func::FuncOp func,AffineMap xPerm,uint64_t ny,uint32_t nTrailingP=0) createBinarySearchFunc() argument
393 createScanLoop(OpBuilder & builder,ModuleOp module,func::FuncOp func,ValueRange xs,Value i,Value p,AffineMap xPerm,uint64_t ny,int step) createScanLoop() argument
436 createCompareThenSwap(OpBuilder & builder,Location loc,AffineMap xPerm,uint64_t ny,SmallVectorImpl<Value> & swapOperands,SmallVectorImpl<Value> & compareOperands,Value a,Value b) createCompareThenSwap() argument
453 createInsert3rd(OpBuilder & builder,Location loc,AffineMap xPerm,uint64_t ny,SmallVectorImpl<Value> & swapOperands,SmallVectorImpl<Value> & compareOperands,Value v0,Value v1,Value v2) createInsert3rd() argument
465 createSort3(OpBuilder & builder,Location loc,AffineMap xPerm,uint64_t ny,SmallVectorImpl<Value> & swapOperands,SmallVectorImpl<Value> & compareOperands,Value v0,Value v1,Value v2) createSort3() argument
480 createSort5(OpBuilder & builder,Location loc,AffineMap xPerm,uint64_t ny,SmallVectorImpl<Value> & swapOperands,SmallVectorImpl<Value> & compareOperands,Value v0,Value v1,Value v2,Value v3,Value v4) createSort5() argument
511 createChoosePivot(OpBuilder & builder,ModuleOp module,func::FuncOp func,AffineMap xPerm,uint64_t ny,Value lo,Value hi,Value mi,ValueRange args) createChoosePivot() argument
579 createPartitionFunc(OpBuilder & builder,ModuleOp module,func::FuncOp func,AffineMap xPerm,uint64_t ny,uint32_t nTrailingP=0) createPartitionFunc() argument
730 createShiftDownFunc(OpBuilder & builder,ModuleOp module,func::FuncOp func,AffineMap xPerm,uint64_t ny,uint32_t nTrailingP) createShiftDownFunc() argument
861 createHeapSortFunc(OpBuilder & builder,ModuleOp module,func::FuncOp func,AffineMap xPerm,uint64_t ny,uint32_t nTrailingP) createHeapSortFunc() argument
920 createQuickSort(OpBuilder & builder,ModuleOp module,func::FuncOp func,ValueRange args,AffineMap xPerm,uint64_t ny,uint32_t nTrailingP) createQuickSort() argument
1000 createSortStableFunc(OpBuilder & builder,ModuleOp module,func::FuncOp func,AffineMap xPerm,uint64_t ny,uint32_t nTrailingP) createSortStableFunc() argument
1115 createQuickSortFunc(OpBuilder & builder,ModuleOp module,func::FuncOp func,AffineMap xPerm,uint64_t ny,uint32_t nTrailingP) createQuickSortFunc() argument
1221 matchAndRewriteSortOp(OpTy op,ValueRange xys,AffineMap xPerm,uint64_t ny,PatternRewriter & rewriter) matchAndRewriteSortOp() argument
1411 auto xPerm = op.getPermMap(); matchAndRewrite() local
[all...]
H A DSparseTensorCodegen.cpp1553 auto xPerm = rewriter.getMultiDimIdentityMap(lvlRank); matchAndRewrite() local
/llvm-project/mlir/lib/Dialect/SparseTensor/IR/
H A DSparseTensorDialect.cpp2036 AffineMap xPerm = getPermMap(); verify() local