Skip to content

Commit

Permalink
add precision check
Browse files Browse the repository at this point in the history
  • Loading branch information
itikhono committed Nov 17, 2024
1 parent 1202ab2 commit c93f7a7
Showing 1 changed file with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -200,7 +200,8 @@ KeepConstsPrecision::KeepConstsPrecision(const element::TypeVector& precisions,
for (const auto& pattern_node : keep_const_precisions) {
if (pt_map.count(pattern_node.first)) {
auto node = pt_map.at(pattern_node.first).get_node_shared_ptr();
if (ov::as_type_ptr<v0::Constant>(node)) {
if (ov::as_type_ptr<v0::Constant>(node) &&
check_precision(node->get_output_element_type(0), precisions)) {
if (pattern_node.second) {
ov::disable_keep_const_precision(node);
} else {
Expand Down

0 comments on commit c93f7a7

Please sign in to comment.