tfl.lattice_lib.assert_constraints

View source on GitHub

Asserts that weights satisfy constraints.

weights Lattice weights tensor of shape: (prod(lattice_sizes), units).
lattice_sizes List or tuple of integers which represents lattice sizes.
monotonicities Monotonicity constraints.
edgeworth_trusts Edgeworth trust constraints.
trapezoid_trusts Trapezoid trust constraints.
monotonic_dominances Monotonic dominance constraints.
range_dominances Range dominance constraints.
joint_monotonicities Joint monotonicity constraints.
joint_unimodalities Joint unimodality constraints.
output_min None or lower bound constraints.
output_max None or upper bound constraints.
eps Allowed constraints violation.

List of assertion ops in graph mode or directly executes assertions in eager mode.