Skip to content

Commit 805005d

Browse files
ethansfngfacebook-github-bot
authored andcommitted
Replace conv_transpose optional bias with zero bias
Differential Revision: D83517548
1 parent 73b3303 commit 805005d

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

backends/cadence/aot/replace_ops.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -438,11 +438,17 @@ class ReplaceConvolutionOptionalArgsWithConcreteArgsPass(ExportPass):
438438
"""
439439

440440
def call_operator(self, op, args, kwargs, meta):
441-
if get_edge_overload_packet(op) != exir_ops.edge.cadence.convolution:
441+
op_packet = get_edge_overload_packet(op)
442+
if op_packet not in {
443+
exir_ops.edge.cadence.convolution,
444+
exir_ops.edge.cadence.transposed_convolution,
445+
}:
442446
return super().call_operator(op, args, kwargs, meta)
443447

444448
# Check if the bias is already concrete
445-
assert len(args) == 8
449+
is_transposed = op_packet == exir_ops.edge.cadence.transposed_convolution
450+
expected_args = 9 if is_transposed else 8
451+
assert len(args) == expected_args
446452
if args[2] is not None:
447453
return super().call_operator(op, args, kwargs, meta)
448454

0 commit comments

Comments
 (0)