Skip to content

Commit ba89ef8

Browse files
committed
apply resource
1 parent 45c6171 commit ba89ef8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

hls4ml/backends/vivado/passes/resource_strategy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ class ApplyResourceStrategy(OptimizerPass):
77
''' Transposes the weights to use the dense_resource matrix multiply routine '''
88
def match(self, node):
99

10-
node_matches = isinstance(node, (Dense, Conv1D, SeparableConv1D, Conv2D, SeparableConv2D, LSTM, GRU))
10+
node_matches = isinstance(node, (Dense, Conv1D, SeparableConv1D, Conv2D, SeparableConv2D, LSTM, GRU, Conv1DTranspose, Conv2DTranspose))
1111

1212
is_resource_strategy = node.get_attr('strategy', '').lower() == 'resource'
1313
already_transformed = node.get_attr('_weights_transposed', False) == True

0 commit comments

Comments
 (0)