We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent f57d0a3 commit 8ab4260Copy full SHA for 8ab4260
src/frontends/tensorflow_common/src/op/cumsum.cpp
@@ -29,7 +29,7 @@ OutputVector translate_cumsum_op(const NodeContext& node) {
29
x = complex_type_mark->input_value(0);
30
auto zero = create_same_type_const_scalar<int32_t>(axis, 0);
31
auto less_than_zero = make_shared<v1::Less>(axis, zero);
32
- auto const_one = make_shared<v0::Constant>(element::i32, Shape{}, 1);
+ auto const_one = create_same_type_const_scalar<int32_t>(axis, 1);
33
34
auto axis_update = make_shared<v1::Select>(less_than_zero, const_one, zero);
35
auto new_axis = make_shared<v1::Subtract>(axis, axis_update);
0 commit comments