diff --git a/src/finn/custom_op/fpgadataflow/vectorvectoractivation.py b/src/finn/custom_op/fpgadataflow/vectorvectoractivation.py index 24cb3101fa0b0f475fb623e226867270ed3f8388..77fed5e3abb169684ab9ccd3f03ddc30643d87a4 100644 --- a/src/finn/custom_op/fpgadataflow/vectorvectoractivation.py +++ b/src/finn/custom_op/fpgadataflow/vectorvectoractivation.py @@ -670,7 +670,9 @@ class VectorVectorActivation(HLSCustomOp): wei = npy_to_rtlsim_input( "{}/weights.npy".format(code_gen_dir), export_wdt, wnbits ) - num_w_reps = 1 + dim_h, dim_w = self.get_nodeattr("Dim") + num_w_reps = dim_h * dim_w + io_dict = { "inputs": {"in0": inp, "weights": wei * num_w_reps}, "outputs": {"out": []},