saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-04 04:07:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Enable some ops for QDQ Node Unit support (#14701)

Browse source 
### Description
Enable some ops for QDQ Node Unit support: Flatten, Split,
GlobalAveragePool, ReduceMean, Relu, Sigmoid, Sqrt, Div, Mul, Pow, Sub.
This commit is contained in:
Hector Li 2023-02-16 17:14:31 -08:00 committed by GitHub
parent 892f59b31a
commit 0dda42b46c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
onnxruntime/core/optimizer/qdq_transformer/selectors_actions/shared/utils.cc
View file

@ -30,21 +30,31 @@ void Selectors::RegisterSelector(const OpVersionsAndSelector::OpVersionsMap& ops
static const OpVersionsAndSelector::OpVersionsMap GetMiscOpVersionsMap() {
return {{"Gather", {}},
{"Reshape", {}},
{"Flatten", {}},
{"Transpose", {}},
{"MaxPool", {12}},
{"Resize", {}},
{"Split", {}},
{"Squeeze", {}},
{"Unsqueeze", {}}};
}
static const OpVersionsAndSelector::OpVersionsMap GetUnaryOpVersionsMap() {
return {{"AveragePool", {}},
{"GlobalAveragePool", {}},
{"LeakyRelu", {}},
{"ReduceMean", {}},
{"Relu", {}},
{"Sigmoid", {}},
{"Softmax", {}},
{"LeakyRelu", {}}};
{"Sqrt", {}}};
}
static const OpVersionsAndSelector::OpVersionsMap GetBinaryOpVersionsMap() {
return {{"Add", {}},
{"Mul", {}}};
{"Div", {}},
{"Mul", {}},
{"Pow", {}},
{"Sub", {}}};
}
static const OpVersionsAndSelector::OpVersionsMap GetVariadicOpVersionsMap() {
return {{"Concat", {}}};