Added opjson and bug fix for scatternd_onnx_v16 (#678)

Fixed typing error of missed param in this op's clone function Type: Code Improvement Signed-off-by: Feiyue Chen <[email protected]>
VeriSilicon · Jan 12, 2024 · b4b4f00 · b4b4f00
1 parent 0d8ca44
commit b4b4f00
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/include/tim/vx/ops/scatternd_onnx_v16.json b/include/tim/vx/ops/scatternd_onnx_v16.json
@@ -0,0 +1,17 @@
+{
+    "ScatterND_ONNX_V16":{
+        "parameters":
+            [
+                {"name":"reduction",
+                    "dtype": "tim::vx::ops::ScatterND_ONNX_V16::ReductionType",
+                    "Optional": "true",
+                    "default":"tim::vx::ops::ScatterND_ONNX_V16::REDUCTION_NONE",
+                    "range": ["tim::vx::ops::ScatterND_ONNX_V16::REDUCTION_NONE",
+                              "tim::vx::ops::ScatterND_ONNX_V16::REDUCTION_ADD",
+                              "tim::vx::ops::ScatterND_ONNX_V16::REDUCTION_MUL",
+                              "tim::vx::ops::ScatterND_ONNX_V16::REDUCTION_MAX",
+                              "tim::vx::ops::ScatterND_ONNX_V16::REDUCTION_MIN"]
+                }
+            ]
+    }
+}
diff --git a/src/tim/vx/ops/scatternd_onnx_v16.cc b/src/tim/vx/ops/scatternd_onnx_v16.cc
@@ -55,7 +55,7 @@ ScatterND_ONNX_V16::ScatterND_ONNX_V16(Graph* graph, ReductionType reduction)
 }
 
 std::shared_ptr<Operation> ScatterND_ONNX_V16::Clone(std::shared_ptr<Graph>& graph) const {
-  return graph->CreateOperation<ScatterND_ONNX_V16>();
+  return graph->CreateOperation<ScatterND_ONNX_V16>(this->reduction_);
 }
 
 }  // namespace ops