Add CUMSUM node name

"CUMSUM" operation name is used along with "cumsum", so add both names in cusum translator op. Signed-off-by: Pratikkumar V Prajapati <[email protected]>
openvinotoolkit · Sep 19, 2024 · 4543fe8 · 4543fe8
1 parent 3e8557a
commit 4543fe8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/frontends/tensorflow_common/src/op/cumsum.cpp b/src/frontends/tensorflow_common/src/op/cumsum.cpp
@@ -18,7 +18,8 @@ namespace tensorflow {
 namespace op {
 
 OutputVector translate_cumsum_op(const NodeContext& node) {
-    default_op_checks(node, 2, {"Cumsum"}, true);
+    default_op_checks(node, 2, {"Cumsum", "CUMSUM"}, true);
+
     auto x = node.get_input(0);
     auto axis = node.get_input(1);
     auto exclusive = node.get_attribute<bool>("exclusive", false);