Support none reduction for scatterND (#1013)

Signed-off-by: Kevin Chen <[email protected]>

Author: kevinch-nv

docs/operators.md

@@ -168,7 +168,7 @@ TensorRT supports the following ONNX data types: DOUBLE, FLOAT32, FLOAT16, BFLOA
 | Scan                      | Y          | FP32, FP16, BF16|
 | Scatter                   | Y          | FP32, FP16, BF16, INT32, INT64 |
 | ScatterElements           | Y          | FP32, FP16, BF16, INT32, INT64 |
-| ScatterND                 | Y          | FP32, FP16, BF16, INT32, INT64 | `reduction` is not supported
+| ScatterND                 | Y          | FP32, FP16, BF16, INT32, INT64 | `reduction` other than `none` is not supported
 | Selu                      | Y          | FP32, FP16, BF16, |
 | SequenceAt                | N          |
 | SequenceConstruct         | N          |

onnxOpImporters.cpp

@@ -5334,8 +5334,9 @@ DEFINE_BUILTIN_OP_IMPORTER(GridSample)
 DEFINE_BUILTIN_OP_IMPORTER(ScatterND)
 {
     OnnxAttrs attrs(node, ctx);
-    ONNXTRT_CHECK_NODE(!attrs.count("reduction"), "Attribute reduction is not supported.", node, nodeIdx,
-        ErrorCode::kUNSUPPORTED_NODE_ATTR);
+    auto mode = attrs.get<std::string>("reduction", "none");
+    ONNXTRT_CHECK_NODE(
+        mode == "none", "ScatterND with reduction is not supported.", node, nodeIdx, ErrorCode::kUNSUPPORTED_NODE_ATTR);
     return addScatterLayer(ctx, node, nodeIdx, inputs, nvinfer1::ScatterMode::kND);
 }