Home
last modified time | relevance | path

Searched refs:AccumulatorFactory (Results 1 – 1 of 1) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/GPU/Transforms/
H A DAllReduceLowering.cpp29 using AccumulatorFactory = std::function<Value(Value, Value)>; typedef
99 AccumulatorFactory accumFactory = getFactory(); in rewrite()
170 AccumulatorFactory getFactory() { in getFactory()
177 return AccumulatorFactory(); in getFactory()
183 AccumulatorFactory getFactory(Region &body) { in getFactory()
184 return AccumulatorFactory([&](Value lhs, Value rhs) { in getFactory()
216 AccumulatorFactory getFactory(gpu::AllReduceOperation opName) { in getFactory()
249 AccumulatorFactory getFactory() { in getFactory()
258 AccumulatorFactory getCmpFactory() const { in getCmpFactory()
324 AccumulatorFactory &accumFactory) { in createSubgroupReduce()