Fix ComplexType Output for AddBias

Pranshu-S · Mar 10, 2024 · c778406 · c778406
1 parent 6027732
commit c778406
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/frontends/tensorflow_common/src/op/bias_add.cpp b/src/frontends/tensorflow_common/src/op/bias_add.cpp
@@ -62,6 +62,11 @@ OutputVector translate_bias_add_op(const NodeContext& node) {
 
     auto res = make_shared<v1::Add>(value, bias_reshaped);
     set_node_name(node.get_name(), res);
+
+    if (complex_type_inputs) {
+        auto complex_reshape = make_shared<ComplexTypeMark>(res, complex_type_mark_value->get_complex_part_type());
+        return {complex_reshape->output(0)};
+    }
     return res->outputs();
 }
 }  // namespace op