Skip to content

Commit c778406

Browse files
committed
Fix ComplexType Output for AddBias
1 parent 6027732 commit c778406

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/frontends/tensorflow_common/src/op/bias_add.cpp

+5
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,11 @@ OutputVector translate_bias_add_op(const NodeContext& node) {
6262

6363
auto res = make_shared<v1::Add>(value, bias_reshaped);
6464
set_node_name(node.get_name(), res);
65+
66+
if (complex_type_inputs) {
67+
auto complex_reshape = make_shared<ComplexTypeMark>(res, complex_type_mark_value->get_complex_part_type());
68+
return {complex_reshape->output(0)};
69+
}
6570
return res->outputs();
6671
}
6772
} // namespace op

0 commit comments

Comments
 (0)