tfl.kronecker_factored_lattice_lib.finalize_scale_constraints

Clips scale to strictly satisfy all constraints.

scale Scale variable of shape: (units, num_terms).
output_min None or minimum layer output.
output_max None or maximum layer output.

Clipped scale tensor of same shape as scale.