tfl.lattice_lib.assert_constraints

View source on GitHub

Asserts that weights satisfy constraints.

tfl.lattice_lib.assert_constraints(
    weights,
    lattice_sizes,
    monotonicities,
    edgeworth_trusts,
    trapezoid_trusts,
    monotonic_dominances,
    joint_monotonicities,
    output_min=None,
    output_max=None,
    eps=1e-06
)

Args:

  • 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.
  • joint_monotonicities: Joint monotonicity constraints.
  • output_min: None or lower bound constraints.
  • output_max: None or upper bound constraints.
  • eps: Allowed constraints violation.

Returns:

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