Searched defs:unpackFn (Results 1 – 1 of 1) sorted by relevance
/llvm-project/mlir/lib/Dialect/GPU/Transforms/ |
H A D | SubgroupReduceLowering.cpp | 154 createSubgroupShuffleReduction(OpBuilder & builder,Location loc,Value input,gpu::AllReduceOperation mode,unsigned subgroupSize,function_ref<Value (Value)> packFn,function_ref<Value (Value)> unpackFn) createSubgroupShuffleReduction() argument 211 auto unpackFn = [loc, &rewriter, equivIntType, matchAndRewrite() local 281 auto unpackFn = [loc, &rewriter, shuffleVecType, matchAndRewrite() local
|