Skip to content

Commit dd021ec

Browse files
committed
fix n_in/n_out
1 parent d56dc73 commit dd021ec

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

hls4ml/backends/vivado/passes/convolution_templates.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -108,8 +108,8 @@ def format(self, node):
108108

109109
mult_params = self._default_config_params(node)
110110
if is_pointwise_parallel_latency:
111-
mult_params['n_in'] = node.get_attr('n_chan') * node.get_attr('filt_width') / mult_params['reuse']
112-
mult_params['n_out'] = node.get_attr('n_filt') / mult_params['reuse']
111+
mult_params['n_in'] = int(node.get_attr('in_width') * node.get_attr('n_chan') * node.get_attr('filt_width') / mult_params['reuse'])
112+
mult_params['n_out'] = int(node.get_attr('in_width') * node.get_attr('n_filt') / mult_params['reuse'])
113113
else:
114114
mult_params['n_in'] = node.get_attr('n_chan') * node.get_attr('filt_width')
115115
mult_params['n_out'] = node.get_attr('n_filt')

0 commit comments

Comments
 (0)