From b4b4f00f474e3b2cc2b33783b45a6a53e017580c Mon Sep 17 00:00:00 2001
From: Chen Feiyue <69809761+chenfeiyue-cfy@users.noreply.github.com>
Date: Fri, 12 Jan 2024 11:49:46 +0800
Subject: [PATCH] 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 <Feiyue.Chen@verisilicon.com>
---
 include/tim/vx/ops/scatternd_onnx_v16.json | 17 +++++++++++++++++
 src/tim/vx/ops/scatternd_onnx_v16.cc       |  2 +-
 2 files changed, 18 insertions(+), 1 deletion(-)
 create mode 100644 include/tim/vx/ops/scatternd_onnx_v16.json

diff --git a/include/tim/vx/ops/scatternd_onnx_v16.json b/include/tim/vx/ops/scatternd_onnx_v16.json
new file mode 100644
index 0000000..f84625d
--- /dev/null
+++ 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"]
+                }
+            ]
+    }
+}
\ No newline at end of file
diff --git a/src/tim/vx/ops/scatternd_onnx_v16.cc b/src/tim/vx/ops/scatternd_onnx_v16.cc
index 6c56697..7e9d737 100644
--- 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