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

Pranshu-S · Mar 27, 2024 · 9c7b5af · 9c7b5af
1 parent a14a1ec
commit 9c7b5af
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/frontends/tensorflow_common/src/op/pack.cpp b/src/frontends/tensorflow_common/src/op/pack.cpp
@@ -21,6 +21,7 @@ namespace op {
 
 OutputVector translate_pack_op(const NodeContext& node) {
     default_op_checks(node, 1, {"Pack", "PACK"}, true);
+    auto num_size = static_cast<int>(node.get_input_size());
 
     auto axis = node.get_attribute<int64_t>("axis", 0);