From 520deeeaeaea7191d56951d3cb451a4eb95a22fe Mon Sep 17 00:00:00 2001 From: siddhant-0707 Date: Wed, 25 Oct 2023 13:11:23 +0530 Subject: [PATCH] Update is_nonzero.cpp --- src/frontends/pytorch/src/op/is_nonzero.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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};