Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DElementwiseOpFusion.cpp1109 ArrayRef<ReassociationIndices> foldedIterationDims) { in initialize() argument
1112 for (ReassociationIndicesRef foldedIterationDim : foldedIterationDims) { in initialize()
1333 GenericOp genericOp, ArrayRef<ReassociationIndices> foldedIterationDims, in collapseGenericOpIterationDims() argument
1336 if (genericOp.getNumLoops() <= 1 || foldedIterationDims.empty() || in collapseGenericOpIterationDims()
1337 llvm::all_of(foldedIterationDims, [](ReassociationIndicesRef foldedDims) { in collapseGenericOpIterationDims()
1344 foldedIterationDims))) { in collapseGenericOpIterationDims()