Skip to content

Commit

Permalink
fix: updated Expand Op properties
Browse files Browse the repository at this point in the history
  • Loading branch information
ankitm3k committed Oct 1, 2024
1 parent b766127 commit 40b3d78
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions onnxruntime/core/providers/openvino/ov_versions/data_ops.cc
Original file line number Diff line number Diff line change
Expand Up @@ -127,6 +127,7 @@ std::vector<SupportedOp> supported_op_mode = {
{"Erf", V_2020_4, {"CPU", "GPU"}},
{"Exp", V_2020_4, {"CPU", "GPU"}},
{"Expand", V_2022_1, {"CPU", "GPU"}},
{"Expand", V_2024_3, {"CPU", "GPU"}},
{"EyeLike", V_2022_1, {"CPU"}},
{"Flatten", V_2020_4, {"CPU", "GPU"}},
{"Floor", V_2020_4, {"CPU", "GPU"}},
Expand Down Expand Up @@ -328,6 +329,7 @@ void DataOps::populate_op_mode_supported() {
no_dimension_supported_.push_back({"Equal", V_2022_1, {"CPU"}});
no_dimension_supported_.push_back({"Equal", V_2023_0, {"GPU"}});
no_dimension_supported_.push_back({"Expand", V_2023_3, {"CPU"}});
no_dimension_supported_.push_back({"Expand", V_2024_3, {"CPU", "GPU"}});
no_dimension_supported_.push_back({"Floor", V_2020_4, {"All"}});
no_dimension_supported_.push_back({"Gather", V_2020_4, {"All"}});
no_dimension_supported_.push_back({"Identity", V_2023_0, {"All"}});
Expand Down

0 comments on commit 40b3d78

Please sign in to comment.