Fix logic for NCHW

Pranshu-S · Mar 23, 2024 · 245b986 · 245b986
1 parent 91e1d48
commit 245b986
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/frontends/tensorflow_common/src/op/bias_add.cpp b/src/frontends/tensorflow_common/src/op/bias_add.cpp
@@ -47,14 +47,11 @@ OutputVector translate_bias_add_op(const NodeContext& node) {
         TENSORFLOW_OP_VALIDATION(node,
                                  value_shape.rank().is_static(),
                                  "Value of dynamic rank for BiasAdd in NCHW layout is not supported.");
-        auto value_rank = complex_type_inputs ? value_shape.rank().get_length() : value_shape.rank().get_length();
+        auto value_rank = complex_type_inputs ? value_shape.rank().get_length() - 1 : value_shape.rank().get_length();
 
         std::vector<int64_t> axes_unsqueeze;
         for (int64_t dim_ind = 0; dim_ind < value_rank; ++dim_ind) {
-            if (!complex_type_inputs && dim_ind != 1) {
-                axes_unsqueeze.push_back(dim_ind);
-            }
-            if (complex_type_inputs && dim_ind != 2){
+            if (dim_ind != 1) {
                 axes_unsqueeze.push_back(dim_ind);
             }
         }