[ONNX] 🛠️🕸️ Fix softmax opsets (#5959)

This commit is contained in:
Tomasz Socha authored and GitHub committed 2021-06-07 14:01:20 +03:00
1 parent 9fd73b4f1a
commit f9b27c3714
5 files changed
+9 -9

No files matched your search

@@ -69,7 +69,7 @@ namespace ngraph
return {result};
}
} // namespace set_1
namespace set_7
namespace set_11
{
OutputVector softmax(const Node& node)
{
@@ -109,7 +109,7 @@ namespace ngraph
return {result};
}
} // namespace set_7
} // namespace set_11
namespace set_13
{
OutputVector softmax(const Node& node)
@@ -19,11 +19,11 @@ namespace ngraph
} // namespace set_1
namespace set_7
namespace set_11
{
OutputVector softmax(const Node& node);
} // namespace set_7
} // namespace set_11
namespace set_13
{
@@ -437,9 +437,7 @@ namespace ngraph
REGISTER_OPERATOR("Slice", 1, slice);
REGISTER_OPERATOR("Slice", 10, slice);
REGISTER_OPERATOR("Softmax", 1, softmax);
// Softmax v7 should be in the 11th opset but,
// other frameworks(mxnet and onnxruntime) already use for older models.
REGISTER_OPERATOR("Softmax", 7, softmax);
REGISTER_OPERATOR("Softmax", 11, softmax);
REGISTER_OPERATOR("Softmax", 13, softmax);
REGISTER_OPERATOR("Softplus", 1, softplus);
REGISTER_OPERATOR("Softsign", 1, softsign);
+1 -1
View File
@@ -149,5 +149,5 @@ xfail_issue_49753 = xfail_test(reason="RuntimeError: Unsupported dynamic ops: v1
xfail_issue_49754 = xfail_test(reason="RuntimeError: Unsupported dynamic ops: v1::TopKIE")
xfail_issue_52463 = xfail_test(reason="test_operator_add_size1_singleton_broadcast_cpu - "
"Not equal to tolerance")
xfail_issue_45432 = xfail_test(reason="Einsum is not implemented in CPU plugin.")
xfail_issue_onnx_models_140 = xfail_test(reason="https://github.com/onnx/models/issues/140")
@@ -25,7 +25,8 @@ from tests import (
xfail_issue_47430,
xfail_issue_47495,
xfail_issue_48145,
xfail_issue_48190)
xfail_issue_48190,
xfail_issue_onnx_models_140)
MODELS_ROOT_DIR = tests.MODEL_ZOO_DIR
@@ -170,6 +171,7 @@ if len(zoo_models) > 0:
(xfail_issue_47430, "test_onnx_model_zoo_vision_object_detection_segmentation_fcn_model_fcn_resnet101_11_fcn_resnet101_11_model_cpu"),
(xfail_issue_48145, "test_onnx_model_zoo_text_machine_comprehension_bert_squad_model_bertsquad_8_download_sample_8_bertsquad8_cpu"),
(xfail_issue_48190, "test_onnx_model_zoo_text_machine_comprehension_roberta_model_roberta_base_11_roberta_base_11_roberta_base_11_cpu"),
(xfail_issue_onnx_models_140, "test_onnx_model_zoo_vision_object_detection_segmentation_duc_model_ResNet101_DUC_7_ResNet101_DUC_HDC_ResNet101_DUC_HDC_cpu"),
# Model MSFT
(xfail_issue_37973, "test_MSFT_opset7_tf_inception_v2_model_cpu"),