colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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>
This commit is contained in:
Chen Feiyue 2024-01-12 11:49:46 +08:00 committed by GitHub
parent 0d8ca44377
commit b4b4f00f47
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
include/tim/vx/ops/scatternd_onnx_v16.json Normal file
View File

@ -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"]
}
]
}
}

2
src/tim/vx/ops/scatternd_onnx_v16.cc
View File

@ -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