Skip to content

Commit 970ee1c

Browse files
committed
fix accum_t; no transpose for resource?
1 parent ba89ef8 commit 970ee1c

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

hls4ml/backends/fpga/fpga_backend.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,9 @@
1414
Activation,
1515
BatchNormalization,
1616
Conv1D,
17+
Conv1DTranspose,
1718
Conv2D,
19+
Conv2DTranspose,
1820
Dense,
1921
Dot,
2022
Embedding,
@@ -52,7 +54,9 @@ def __init__(self, name):
5254
accum_layers = [
5355
Dense,
5456
Conv1D,
57+
Conv1DTranspose,
5558
Conv2D,
59+
Conv2DTranspose,
5660
SeparableConv1D,
5761
SeparableConv2D,
5862
Pooling1D,

hls4ml/backends/vivado/passes/resource_strategy.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,13 @@
11
import numpy as np
22

33
from hls4ml.model.optimizer import OptimizerPass
4-
from hls4ml.model.layers import Conv1D, Conv2D, Conv1DTranspose, Conv2DTranspose, Dense, SeparableConv1D, SeparableConv2D, LSTM, GRU
4+
from hls4ml.model.layers import Conv1D, Conv2D, Dense, SeparableConv1D, SeparableConv2D, LSTM, GRU
55

66
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, Conv1DTranspose, Conv2DTranspose))
11-
10+
node_matches = isinstance(node, (Dense, Conv1D, SeparableConv1D, Conv2D, SeparableConv2D, LSTM, GRU))
1211
is_resource_strategy = node.get_attr('strategy', '').lower() == 'resource'
1312
already_transformed = node.get_attr('_weights_transposed', False) == True
1413

0 commit comments

Comments
 (0)