Update src/frontends/tensorflow_common/src/op/cumsum.cpp

Pranshu-S · Mar 30, 2024 · fe7bce6 · fe7bce6
1 parent 5068c00
commit fe7bce6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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<int32_t>(axis, 1);
 
         auto axis_update = make_shared<v1::Select>(less_than_zero, const_one, zero);
-        auto new_axis = make_shared<v1::Subtract>(axis, axis_update);
+        axis = make_shared<v1::Subtract>(axis, axis_update)->output(0);
     }
 
     auto cum_sum = make_shared<v0::CumSum>(x, axis, exclusive, reverse);