Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings

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, range_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.
  • range_dominances: Range 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.