Skip to content

ApplyAlpha layer inherits Trace setting from parent

Javier Duarte requested to merge github/fork/thesps/trace_dev into master

Created by: thesps

The QKeras optimization pass may insert an ApplyAlpha (BatchNormalization) layer for QKeras Dense or Conv layers with alpha != 1. Currently, these can only be traced by adding the layer name to the config dictionary. Since the layer name is generated by the optimizer pass, it's a bit unwieldy for users (who may not anyway be aware that it's being added).

This PR copies the Trace setting of the parent layer into the ApplyAlpha layer. I had to adjust some of the handling of the Trace settings to make it usable, specifically: rather than retrieving a layer's trace setting from the config, it's now retrieved from the layer attributes. Layer config entries were already made layer attributes, so this doesn't interfere with the existing functionality.

I also added a 'convert_precision_string' method, continuing the push to remove precision strings everywhere, which is then used in the profiling to query the type width/integer bits.

Merge request reports