Skip to content

Commit 8ab4260

Browse files
authored
Update src/frontends/tensorflow_common/src/op/cumsum.cpp
1 parent f57d0a3 commit 8ab4260

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

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

+1-1
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ OutputVector translate_cumsum_op(const NodeContext& node) {
2929
x = complex_type_mark->input_value(0);
3030
auto zero = create_same_type_const_scalar<int32_t>(axis, 0);
3131
auto less_than_zero = make_shared<v1::Less>(axis, zero);
32-
auto const_one = make_shared<v0::Constant>(element::i32, Shape{}, 1);
32+
auto const_one = create_same_type_const_scalar<int32_t>(axis, 1);
3333

3434
auto axis_update = make_shared<v1::Select>(less_than_zero, const_one, zero);
3535
auto new_axis = make_shared<v1::Subtract>(axis, axis_update);

0 commit comments

Comments
 (0)