Home
last modified time | relevance | path

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

/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/
H A DSplitReduction.cpp302 SmallVector<Attribute> neutralElements = llvm::to_vector<4>( in splitReductionByScaling() local
306 if (!llvm::all_of(neutralElements, [](Attribute attr) { return attr; })) in splitReductionByScaling()
310 if (op.getNumOutputs() != static_cast<int64_t>(neutralElements.size())) in splitReductionByScaling()
334 for (auto it : llvm::zip(op.outputs(), neutralElements)) { in splitReductionByScaling()