tfl.premade_lib.build_rtl_layer

Creates a tfl.layers.RTL layer.

Main aliases

tfl.premade.premade_lib.build_rtl_layer

tfl.premade_lib.build_rtl_layer(
    calibration_outputs, model_config, submodel_index, average_outputs, dtype
)

This function expects that all features defined in model_config.feature_configs are used and present in calibration_outputs.

Args
`calibration_outputs`	A mapping from feature name to calibration output.
`model_config`	Model configuration object describing model architecture. Should be one of the model configs in `tfl.configs`.
`submodel_index`	Corresponding index into submodels.
`average_outputs`	Whether to average the outputs of this layer.
`dtype`	dtype

Returns
A `tfl.layers.RTL` instance.

Raises
`ValueError`	If `model_config.parameterization` is not one of `'all_vertices'` or `'kronecker_factored'`.

Args