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

tfl.categorical_calibration_lib.assert_constraints

View source on GitHub

Asserts that weights satisfiy constraints.

tfl.categorical_calibration_lib.assert_constraints(
    weights, output_min, output_max, monotonicities, debug_tensors=None, eps=1e-06
)

Args:

  • weights: Tensor which represents weights of Categorical calibration layer.
  • output_min: Lower bound constraint on weights.
  • output_max: Upper bound constraint on weights.
  • monotonicities: List of pair of indices (i, j), indicating constraint weight[i] <= weight[j].
  • debug_tensors: None or list of anything convertible to tensor (for example tensors or strings) which will be printed in case of constraints violation.
  • eps: Allowed constraints violation.

Returns:

List of assertion ops in graph mode or immideately asserts in eager mode.