Skip to content

Commit

Permalink
Update dequantize_linear.cpp
Browse files Browse the repository at this point in the history
  • Loading branch information
gkrivor authored Oct 31, 2024
1 parent b9a94c3 commit e41317d
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/frontends/onnx/frontend/src/op/dequantize_linear.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@
#include "openvino/op/subtract.hpp"
#include "openvino/op/transpose.hpp"
#include "openvino/op/unsqueeze.hpp"
#include "transformations/rt_info/disable_constant_folding.hpp"
#include "utils/common.hpp"
#include "utils/reshape.hpp"
using namespace ov::op;
Expand Down Expand Up @@ -241,6 +242,7 @@ ov::OutputVector dequantize_linear(const ov::frontend::onnx::Node& node) {
zp = inputs[2];
if (zp.get_element_type() != scale.get_element_type()) {
zp = std::make_shared<v0::Convert>(zp, scale_type);
disable_constant_folding(zp.get_node_shared_ptr());
}
zp = std::make_shared<v0::Unsqueeze>(zp, unsqueezed_axes);
}
Expand Down

0 comments on commit e41317d

Please sign in to comment.