diff --git a/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py b/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py index 0cd3471ed18d4877c45f514a235f32fb4974faef..c34a7ef23d60a41f80922fcd48ad5db0bf1d2bf7 100644 --- a/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py +++ b/src/finn/transformation/fpgadataflow/convert_to_hls_layers.py @@ -577,6 +577,7 @@ class InferChannelwiseLinearLayer(Transformation): else: odt = DataType.get_smallest_possible(idt.max() * idt.max()) model.set_initializer(ll_const, ll_cinit.reshape(ch)) + model.set_tensor_datatype(ll_output, odt) # create and insert node new_node = helper.make_node( "ChannelwiseOp_Batch",