View source on GitHub |
Creates a calibration layer followed by a lattice ensemble layer.
tfl.premade_lib.build_calibrated_lattice_ensemble_layer(
calibration_input_layer, model_config, average_outputs, dtype
)
Args | |
---|---|
calibration_input_layer
|
A mapping from feature name to keras.Input .
|
model_config
|
Model configuration object describing model architecture.
Should be one of the model configs in tfl.configs .
|
average_outputs
|
Whether to average the outputs of this layer. |
dtype
|
dtype |
Returns | |
---|---|
A tfl.layers.RTL instance if model_config.lattices is 'rtl_layer.
Otherwise a list of tfl.layers.Lattice instances.
|