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

Added support for Pad-2 operator in OpenVINO-EP (#2405)

Browse source
This commit is contained in:
Patrick Foley 2019-11-18 15:57:27 -08:00 committed by Hector Li
parent 95e8c3377e
commit 1cb6bdc33c
1 changed files with 35 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
onnxruntime/core/providers/openvino/openvino_execution_provider.cc
View file

@ -177,6 +177,7 @@ bool IsOpSupported(std::string name) {
"Transpose",
"Identity",
"MatMul",
"Pad",
"Unsqueeze",
"ImageScaler",
"LeakyRelu",
@ -244,6 +245,17 @@ void CheckGraphSupported(const onnxruntime::GraphViewer& graph_viewer, std::stri
if (it == initializers.end() && node_inputs[i]->Shape() != nullptr) {
if (node_inputs[i]->Shape()->dim_size() == 0) {
throw "Node_input is zero dimension";
} else {
auto num_dims = node_inputs[i]->Shape()->dim_size();
int v = 0;
//Batch size for NCHW can be 0
if (num_dims > 3) {
v = 1;
}
for (int j = v; j < num_dims; j++) {
if (node_inputs[i]->Shape()->dim(j).dim_value() == 0)
throw "Node_input has zero dimension";
}
}
}
}
@ -255,6 +267,29 @@ void CheckGraphSupported(const onnxruntime::GraphViewer& graph_viewer, std::stri
}
}
//Pad doesn't support negative values.
//If all pads are 0, the node gets removed
if (node->OpType() == "Pad") {
auto attributes = node->GetAttributes();
auto it = attributes.find("pads");
if (it != attributes.end()) {
auto pads_ints = attributes["pads"].ints();
auto min_val = *std::min_element(pads_ints.begin(), pads_ints.end());
auto max_val = *std::max_element(pads_ints.begin(), pads_ints.end());
if (min_val < 0) {
throw "Pad: Negative values are not supported";
}
if (min_val == 0 && max_val == 0) {
auto iter = node->OutputNodesBegin();
if (iter == node->OutputNodesEnd()) {
throw "Pad: Must extend tensor in one or more dimensions";
}
}
} else {
throw "Pad-1 and Pad (opset 11) operators are not supported";
}
}
//Conv cannot take more than 1 input
if (node->OpType() == "Conv") {
if (GetInputCount(node, initializers) > 1) {