We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
add
sum
1 parent b1413a6 commit 9cf12e1Copy full SHA for 9cf12e1
1 file changed
lib/Conversion/TorchOnnxToTorch/DefaultDomainQtoZ.cpp
@@ -635,10 +635,12 @@ void mlir::torch::onnx_c::populateDefaultDomainQtoZ(
635
636
// TODO: Implement max and min cases
637
if (reduction == "mul") {
638
- reduction = "multiply";
+ reduction = "prod";
639
} else if (reduction == "max" || reduction == "min") {
640
return rewriter.notifyMatchFailure(
641
binder.op, "max/min reduction unsupported for scatter elements");
642
+ } else if (reduction == "add") {
643
+ reduction = "sum";
644
}
645
646
Value cstStrReduction =
0 commit comments