Add missing BatchNormalization include for ApplyAlpha layer

thesps · Jun 3, 2020 · d7e53de · d7e53de
1 parent d7ff7b1
commit d7e53de
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/hls4ml/model/optimizer/passes/qkeras.py b/hls4ml/model/optimizer/passes/qkeras.py
@@ -79,7 +79,7 @@ def add_weights(self, scale, bias):
 register_layer('ApplyAlpha', ApplyAlpha)
 # TODO ideally: for backend in backends
 temps = templates.get_backend('Vivado')
-temps.register_templates('ApplyAlpha', temps.get_function_template('BatchNormalization'), temps.get_config_template('BatchNormalization'))
+temps.register_templates('ApplyAlpha', temps.get_function_template('BatchNormalization'), temps.get_config_template('BatchNormalization'), temps.get_include_list('BatchNormalization'))
 
 class QKerasFactorizeAlpha(OptimizerPass):
     '''OptimizerPass for extracting alpha "scale" from QKeras quantized layer.