tfl.premade_lib.build_multi_unit_calibration_layers

Creates a mapping from feature names to calibration outputs.

tfl.premade_lib.build_multi_unit_calibration_layers(
    calibration_input_layer,
    calibration_output_units,
    model_config,
    layer_output_range,
    output_single_tensor,
    dtype
)

calibration_input_layer A mapping from feature name to keras.Input.
calibration_output_units A mapping from feature name to units.
model_config Model configuration object describing model architecture. Should be one of the model configs in tfl.configs.
layer_output_range A tfl.premade_lib.LayerOutputRange enum.
output_single_tensor If output for each feature should be a single tensor.
dtype dtype

A mapping from feature name to calibration output Tensors.