Skip to content

Commit

Permalink
Fixed some translator's registration calls
Browse files Browse the repository at this point in the history
  • Loading branch information
gkrivor committed Jun 22, 2024
1 parent 82e30b6 commit c45d0cc
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 5 deletions.
7 changes: 4 additions & 3 deletions src/frontends/onnx/frontend/src/op/quantize_linear.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -153,7 +153,7 @@ static bool registered = register_translator("QuantizeLinear", VersionRange{1, 1
} // namespace opset_1

namespace opset_13 {
namespace {
namespace detail {
ov::OutputVector quantize_linear(ov::Output<ov::Node> x,
ov::Output<ov::Node> y_scale,
ov::Output<ov::Node> y_zero_point,
Expand Down Expand Up @@ -204,7 +204,7 @@ ov::OutputVector quantize_linear(ov::Output<ov::Node> x,

return {detail::make_fake_quantize(y_scale, y_zero_point, x)};
}
} // namespace
} // namespace detail

ov::OutputVector quantize_linear(const ov::frontend::onnx::Node& node) {
const ov::OutputVector inputs{node.get_ov_inputs()};
Expand All @@ -224,8 +224,9 @@ ov::OutputVector quantize_linear(const ov::frontend::onnx::Node& node) {
return ai_onnx::opset_1::quantize_linear(node);
}

return quantize_linear(x, scale, zero_point, node.get_attribute_value<int64_t>("axis", 1), node);
return detail::quantize_linear(x, scale, zero_point, node.get_attribute_value<int64_t>("axis", 1), node);
}
static bool registered = register_translator("QuantizeLinear", VersionRange::since(13), quantize_linear);
} // namespace opset_13
} // namespace ai_onnx
} // namespace onnx
Expand Down
2 changes: 1 addition & 1 deletion src/frontends/onnx/frontend/src/op/split.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -44,7 +44,7 @@ ov::OutputVector split(const ov::frontend::onnx::Node& node) {
}
}

static bool registered = register_translator("Split", VersionRange{13, 12}, split);
static bool registered = register_translator("Split", VersionRange::since(13), split);
} // namespace opset_13
} // namespace ai_onnx
} // namespace onnx
Expand Down
2 changes: 1 addition & 1 deletion src/frontends/onnx/frontend/src/op/topk.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -60,7 +60,7 @@ ov::OutputVector topk(const ov::frontend::onnx::Node& node) {

return {top_k->output(0), top_k->output(1)};
}
static bool registered = register_translator("TopK", VersionRange{10, 10}, topk);
static bool registered = register_translator("TopK", VersionRange::in(10), topk);
} // namespace opset_10

namespace opset_11 {
Expand Down

0 comments on commit c45d0cc

Please sign in to comment.