diff --git a/src/frontends/pytorch/src/op/is_nonzero.cpp b/src/frontends/pytorch/src/op/is_nonzero.cpp index e35b901ed8d426..e486016d290e09 100644 --- a/src/frontends/pytorch/src/op/is_nonzero.cpp +++ b/src/frontends/pytorch/src/op/is_nonzero.cpp @@ -4,6 +4,7 @@ #include "openvino/frontend/pytorch/node_context.hpp" #include "openvino/op/constant.hpp" +#include "openvino/op/convert_like.hpp" #include "openvino/op/not_equal.hpp" #include "pt_framework_node.hpp" #include "utils.hpp" @@ -21,7 +22,7 @@ OutputVector translate_is_nonzero(const NodeContext& context) { Output zero_tensor = context.mark_node(v0::Constant::create(element::boolean, Shape{1}, {false})); - align_eltwise_input_types(context, input, zero_tensor); + zero_tensor = context.mark_node(std::make_shared(zero_tensor, input)); auto result = context.mark_node(std::make_shared(input, zero_tensor)); return {result};