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