From fe7bce67725699dea690a99d48cda1ab3e0d3557 Mon Sep 17 00:00:00 2001 From: Roman Kazantsev Date: Sat, 30 Mar 2024 22:41:29 +0400 Subject: [PATCH] Update src/frontends/tensorflow_common/src/op/cumsum.cpp --- src/frontends/tensorflow_common/src/op/cumsum.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/frontends/tensorflow_common/src/op/cumsum.cpp b/src/frontends/tensorflow_common/src/op/cumsum.cpp index f2b222fb70dc22..9e0343a2529a57 100644 --- a/src/frontends/tensorflow_common/src/op/cumsum.cpp +++ b/src/frontends/tensorflow_common/src/op/cumsum.cpp @@ -32,7 +32,7 @@ OutputVector translate_cumsum_op(const NodeContext& node) { auto const_one = create_same_type_const_scalar(axis, 1); auto axis_update = make_shared(less_than_zero, const_one, zero); - auto new_axis = make_shared(axis, axis_update); + axis = make_shared(axis, axis_update)->output(0); } auto cum_sum = make_shared(x, axis, exclusive, reverse);