SYMBOL INDEX (628 symbols across 42 files)

FILE: docs/build_docs.py
  function local_definitions_filter (line 55) | def local_definitions_filter(path, parent, children):
  function main (line 62) | def main(_):

FILE: examples/keras_functional_uci_heart.py
  function main (line 86) | def main(_):

FILE: examples/keras_sequential_uci_heart.py
  function main (line 80) | def main(_):

FILE: tensorflow_lattice/python/aggregation_layer.py
  class Aggregation (line 36) | class Aggregation(keras.layers.Layer):
    method __init__ (line 61) | def __init__(self, model, **kwargs):
    method call (line 78) | def call(self, x):
    method get_config (line 82) | def get_config(self):
    method from_config (line 91) | def from_config(cls, config, custom_objects=None):

FILE: tensorflow_lattice/python/aggregation_test.py
  class AggregationTest (line 40) | class AggregationTest(tf.test.TestCase):
    method testAggregationLayer (line 42) | def testAggregationLayer(self):

FILE: tensorflow_lattice/python/categorical_calibration_layer.py
  class CategoricalCalibration (line 40) | class CategoricalCalibration(keras.layers.Layer):
    method __init__ (line 97) | def __init__(self,
    method build (line 168) | def build(self, input_shape):
    method call (line 209) | def call(self, inputs):
    method compute_output_shape (line 240) | def compute_output_shape(self, input_shape):
    method get_config (line 248) | def get_config(self):
    method assert_constraints (line 268) | def assert_constraints(self, eps=1e-6):
  class CategoricalCalibrationConstraints (line 289) | class CategoricalCalibrationConstraints(keras.constraints.Constraint):
    method __init__ (line 302) | def __init__(self, output_min=None, output_max=None, monotonicities=No...
    method __call__ (line 318) | def __call__(self, w):
    method get_config (line 326) | def get_config(self):

FILE: tensorflow_lattice/python/categorical_calibration_lib.py
  function project (line 24) | def project(weights, output_min, output_max, monotonicities):
  function assert_constraints (line 64) | def assert_constraints(weights,
  function verify_hyperparameters (line 127) | def verify_hyperparameters(num_buckets=None,

FILE: tensorflow_lattice/python/categorical_calibration_test.py
  class CategoricalCalibrationLayerTest (line 39) | class CategoricalCalibrationLayerTest(parameterized.TestCase, tf.test.Te...
    method setUp (line 41) | def setUp(self):
    method _ResetAllBackends (line 48) | def _ResetAllBackends(self):
    method _ScatterXUniformly (line 52) | def _ScatterXUniformly(self, units, num_points, num_buckets,
    method _SetDefaults (line 73) | def _SetDefaults(self, config):
    method _TrainModel (line 87) | def _TrainModel(self, config):
    method testUnconstrainedNoMissingValue (line 176) | def testUnconstrainedNoMissingValue(self, y_function):
    method testUnconstrainedWithMissingValue (line 201) | def testUnconstrainedWithMissingValue(self, y_function):
    method testConstraints (line 237) | def testConstraints(self, output_min, output_max, monotonicities,
    method testCircularMonotonicites (line 275) | def testCircularMonotonicites(self):
    method testRegularizers (line 301) | def testRegularizers(self, regularizer):
    method testOutputShape (line 321) | def testOutputShape(self):

FILE: tensorflow_lattice/python/cdf_layer.py
  class CDF (line 35) | class CDF(keras.layers.Layer):
    method __init__ (line 84) | def __init__(self,
    method build (line 158) | def build(self, input_shape):
    method call (line 204) | def call(self, inputs):
    method get_config (line 245) | def get_config(self):
  function create_kernel_initializer (line 272) | def create_kernel_initializer(kernel_initializer_id):

FILE: tensorflow_lattice/python/cdf_test.py
  class CdfLayerTest (line 33) | class CdfLayerTest(parameterized.TestCase, tf.test.TestCase):
    method setUp (line 35) | def setUp(self):
    method _ResetAllBackends (line 42) | def _ResetAllBackends(self):
    method _SetDefaults (line 46) | def _SetDefaults(self, config):
    method _ScatterXUniformly (line 59) | def _ScatterXUniformly(self, num_points, input_dims):
    method _ScatterXUniformlyExtendedRange (line 70) | def _ScatterXUniformlyExtendedRange(self, num_points, input_dims):
    method _TwoDMeshGrid (line 81) | def _TwoDMeshGrid(self, num_points, input_dims):
    method _TwoDMeshGridExtendedRange (line 89) | def _TwoDMeshGridExtendedRange(self, num_points, input_dims):
    method _Sin (line 97) | def _Sin(self, x):
    method _SinPlusX (line 100) | def _SinPlusX(self, x):
    method _SinPlusXNd (line 103) | def _SinPlusXNd(self, x):
    method _SinOfSum (line 106) | def _SinOfSum(self, x):
    method _Square (line 109) | def _Square(self, x):
    method _ScaledSum (line 112) | def _ScaledSum(self, x):
    method _GetTrainingInputsAndLabels (line 118) | def _GetTrainingInputsAndLabels(self, config):
    method _TrainModel (line 142) | def _TrainModel(self, config):
    method test1Dim (line 199) | def test1Dim(self, activation, reduction, input_scaling_type, expected...
    method test2Dim (line 231) | def test2Dim(self, activation, reduction, input_scaling_type, expected...
    method test5DimScaledSum (line 263) | def test5DimScaledSum(self, activation, reduction, input_scaling_type,
    method test5DimSinOfSum (line 296) | def test5DimSinOfSum(self, activation, reduction, input_scaling_type,
    method test1DimInputOutOfBounds (line 329) | def test1DimInputOutOfBounds(self, activation, reduction, input_scalin...
    method test2DimInputOutOfBounds (line 362) | def test2DimInputOutOfBounds(self, activation, reduction, input_scalin...
    method testMultiUnitOutputSparsity (line 395) | def testMultiUnitOutputSparsity(self, input_dims, units, activation,
    method testInputScalingInit (line 434) | def testInputScalingInit(self, activation, reduction, input_scaling_init,
    method testGraphSize (line 480) | def testGraphSize(self, input_dims, num_keypoints, units, activation,

FILE: tensorflow_lattice/python/conditional_cdf.py
  function _verify_cdf_params (line 40) | def _verify_cdf_params(
  function cdf_fn (line 119) | def cdf_fn(

FILE: tensorflow_lattice/python/conditional_cdf_test.py
  class CdfFnTest (line 23) | class CdfFnTest(parameterized.TestCase, tf.test.TestCase):
    method assertAllClose (line 25) | def assertAllClose(self, x, y):
    method test_compute_sigmoid (line 122) | def test_compute_sigmoid(
    method test_compute_relu6 (line 270) | def test_compute_relu6(
    method test_scaling_exp_transformation (line 309) | def test_scaling_exp_transformation(
    method test_gradient (line 476) | def test_gradient(
    method test_raise (line 701) | def test_raise(

FILE: tensorflow_lattice/python/conditional_pwl_calibration.py
  function _front_pad (line 62) | def _front_pad(x: tf.Tensor, constant_values: float) -> tf.Tensor:
  function default_keypoint_output_parameters (line 66) | def default_keypoint_output_parameters(
  function default_keypoint_input_parameters (line 114) | def default_keypoint_input_parameters(
  function _verify_pwl_calibration (line 150) | def _verify_pwl_calibration(
  function _compute_interpolation_weights (line 249) | def _compute_interpolation_weights(inputs, keypoints, lengths):
  function pwl_calibration_fn (line 270) | def pwl_calibration_fn(

FILE: tensorflow_lattice/python/conditional_pwl_calibration_test.py
  class PwlCalibrationFnTest (line 24) | class PwlCalibrationFnTest(tf.test.TestCase):
    method assertAllClose (line 26) | def assertAllClose(self, x, y):
    method assertAllGreaterEqual (line 29) | def assertAllGreaterEqual(self, a, comparison_target):
    method assertAllLessEqual (line 32) | def assertAllLessEqual(self, a, comparison_target):
    method assertAllEqual (line 35) | def assertAllEqual(self, a, comparison_target):
    method setUp (line 40) | def setUp(self):
    method test_suite_none_monotonic (line 87) | def test_suite_none_monotonic(self):
    method test_suite_increasing_monotonic (line 240) | def test_suite_increasing_monotonic(self):
    method test_gradient_step (line 399) | def test_gradient_step(self):
    method test_suite_raises (line 440) | def test_suite_raises(self):

FILE: tensorflow_lattice/python/configs.py
  class _Config (line 80) | class _Config(object):
    method __init__ (line 83) | def __init__(self, kwargs):
    method __repr__ (line 90) | def __repr__(self):
    method get_config (line 93) | def get_config(self):
    method deserialize_nested_configs (line 125) | def deserialize_nested_configs(cls, config, custom_objects=None):
  class _HasFeatureConfigs (line 161) | class _HasFeatureConfigs(object):
    method feature_config_by_name (line 164) | def feature_config_by_name(self, feature_name):
  class _HasRegularizerConfigs (line 176) | class _HasRegularizerConfigs(object):
    method regularizer_config_by_name (line 179) | def regularizer_config_by_name(self, regularizer_name):
  class CalibratedLatticeEnsembleConfig (line 194) | class CalibratedLatticeEnsembleConfig(_Config, _HasFeatureConfigs,
    method __init__ (line 275) | def __init__(self,
    method from_config (line 392) | def from_config(cls, config, custom_objects=None):
  class CalibratedLatticeConfig (line 397) | class CalibratedLatticeConfig(_Config, _HasFeatureConfigs,
    method __init__ (line 422) | def __init__(self,
    method from_config (line 504) | def from_config(cls, config, custom_objects=None):
  class CalibratedLinearConfig (line 509) | class CalibratedLinearConfig(_Config, _HasFeatureConfigs,
    method __init__ (line 536) | def __init__(self,
    method from_config (line 580) | def from_config(cls, config, custom_objects=None):
  class AggregateFunctionConfig (line 586) | class AggregateFunctionConfig(_Config, _HasFeatureConfigs,
    method __init__ (line 609) | def __init__(self,
    method from_config (line 685) | def from_config(cls, config, custom_objects=None):
  class FeatureConfig (line 690) | class FeatureConfig(_Config, _HasRegularizerConfigs):
    method __init__ (line 747) | def __init__(self,
    method from_config (line 838) | def from_config(cls, config, custom_objects=None):
  class RegularizerConfig (line 843) | class RegularizerConfig(_Config):
    method __init__ (line 922) | def __init__(self, name, l1=0.0, l2=0.0):
    method from_config (line 933) | def from_config(cls, config, custom_objects=None):
  class TrustConfig (line 938) | class TrustConfig(_Config):
    method __init__ (line 1030) | def __init__(self,
    method from_config (line 1046) | def from_config(cls, config, custom_objects=None):
  class DominanceConfig (line 1051) | class DominanceConfig(_Config):
    method __init__ (line 1085) | def __init__(self, feature_name, dominance_type='monotonic'):
    method from_config (line 1097) | def from_config(cls, config, custom_objects=None):
  class _TypeDict (line 1102) | class _TypeDict(collections.defaultdict):
    method __init__ (line 1105) | def __init__(self, hparams):
    method __contains__ (line 1110) | def __contains__(self, _):
  function apply_updates (line 1114) | def apply_updates(model_config, updates):
  function _apply_update (line 1155) | def _apply_update(node, k, v):

FILE: tensorflow_lattice/python/configs_test.py
  class ConfigsTest (line 60) | class ConfigsTest(tf.test.TestCase):
    method test_from_config (line 62) | def test_from_config(self):
    method test_updates (line 150) | def test_updates(self):

FILE: tensorflow_lattice/python/internal_utils.py
  function _topological_sort (line 28) | def _topological_sort(key_less_than_values):
  function _min_projection (line 65) | def _min_projection(weights, sorted_indices, key_less_than_values, step):
  function _max_projection (line 96) | def _max_projection(weights, sorted_indices, key_greater_than_values, st...
  function approximately_project_categorical_partial_monotonicities (line 127) | def approximately_project_categorical_partial_monotonicities(

FILE: tensorflow_lattice/python/internal_utils_test.py
  class InternalUtilsTest (line 26) | class InternalUtilsTest(parameterized.TestCase, tf.test.TestCase):
    method _ResetAllBackends (line 28) | def _ResetAllBackends(self):
    method testApproximatelyProjectCategoricalPartialMonotonicities (line 36) | def testApproximatelyProjectCategoricalPartialMonotonicities(

FILE: tensorflow_lattice/python/kronecker_factored_lattice_layer.py
  class KroneckerFactoredLattice (line 51) | class KroneckerFactoredLattice(keras.layers.Layer):
    method __init__ (line 121) | def __init__(self,
    method build (line 194) | def build(self, input_shape):
    method call (line 280) | def call(self, inputs):
    method compute_output_shape (line 292) | def compute_output_shape(self, input_shape):
    method get_config (line 302) | def get_config(self):
    method finalize_constraints (line 323) | def finalize_constraints(self):
    method assert_constraints (line 339) | def assert_constraints(self, eps=1e-6):
  function create_kernel_initializer (line 362) | def create_kernel_initializer(kernel_initializer_id,
  function create_scale_initializer (line 418) | def create_scale_initializer(scale_initializer_id, output_min, output_max):
  class KFLRandomMonotonicInitializer (line 449) | class KFLRandomMonotonicInitializer(keras.initializers.Initializer):
    method __init__ (line 454) | def __init__(self, monotonicities, init_min=0.5, init_max=1.5, seed=No...
    method __call__ (line 469) | def __call__(self, shape, scale, dtype=None, **kwargs):
    method get_config (line 489) | def get_config(self):
  class ScaleInitializer (line 500) | class ScaleInitializer(keras.initializers.Initializer):
    method __init__ (line 512) | def __init__(self, output_min, output_max):
    method __call__ (line 522) | def __call__(self, shape, dtype=None, **kwargs):
    method get_config (line 538) | def get_config(self):
  class BiasInitializer (line 547) | class BiasInitializer(keras.initializers.Initializer):
    method __init__ (line 558) | def __init__(self, output_min, output_max):
    method __call__ (line 568) | def __call__(self, shape, dtype=None, **kwargs):
    method get_config (line 583) | def get_config(self):
  class KroneckerFactoredLatticeConstraints (line 592) | class KroneckerFactoredLatticeConstraints(keras.constraints.Constraint):
    method __init__ (line 604) | def __init__(self,
    method __call__ (line 631) | def __call__(self, w):
    method get_config (line 651) | def get_config(self):
  class ScaleConstraints (line 662) | class ScaleConstraints(keras.constraints.Constraint):
    method __init__ (line 676) | def __init__(self, output_min=None, output_max=None):
    method __call__ (line 688) | def __call__(self, scale):
    method get_config (line 703) | def get_config(self):

FILE: tensorflow_lattice/python/kronecker_factored_lattice_lib.py
  function custom_reduce_prod (line 25) | def custom_reduce_prod(t, axis):
  function evaluate_with_hypercube_interpolation (line 73) | def evaluate_with_hypercube_interpolation(inputs, scale, bias, kernel, u...
  function default_init_params (line 152) | def default_init_params(output_min, output_max):
  function kfl_random_monotonic_initializer (line 165) | def kfl_random_monotonic_initializer(shape,
  function scale_initializer (line 224) | def scale_initializer(units, num_terms, output_min, output_max):
  function bias_initializer (line 256) | def bias_initializer(units, output_min, output_max, dtype=tf.float32):
  function _approximately_project_monotonicity (line 287) | def _approximately_project_monotonicity(weights, units, scale, monotonic...
  function _approximately_project_bounds (line 346) | def _approximately_project_bounds(weights, units, output_min, output_max):
  function finalize_weight_constraints (line 394) | def finalize_weight_constraints(weights, units, scale, monotonicities,
  function finalize_scale_constraints (line 447) | def finalize_scale_constraints(scale, output_min, output_max):
  function verify_hyperparameters (line 472) | def verify_hyperparameters(lattice_sizes=None,
  function _assert_monotonicity_constraints (line 552) | def _assert_monotonicity_constraints(weights, units, scale, monotonicities,
  function _assert_bound_constraints (line 600) | def _assert_bound_constraints(weights, units, scale, output_min, output_...
  function assert_constraints (line 697) | def assert_constraints(weights,

FILE: tensorflow_lattice/python/kronecker_factored_lattice_test.py
  class KroneckerFactoredLatticeTest (line 38) | class KroneckerFactoredLatticeTest(parameterized.TestCase, tf.test.TestC...
    method setUp (line 40) | def setUp(self):
    method _ResetAllBackends (line 49) | def _ResetAllBackends(self):
    method _ScatterXUniformly (line 53) | def _ScatterXUniformly(self, num_points, lattice_sizes, input_dims):
    method _ScatterXUniformlyExtendedRange (line 65) | def _ScatterXUniformlyExtendedRange(self, num_points, lattice_sizes,
    method _SameValueForAllDims (line 79) | def _SameValueForAllDims(self, num_points, lattice_sizes, input_dims):
    method _TwoDMeshGrid (line 90) | def _TwoDMeshGrid(self, num_points, lattice_sizes, input_dims):
    method _TwoDMeshGridExtendedRange (line 102) | def _TwoDMeshGridExtendedRange(self, num_points, lattice_sizes, input_...
    method _Sin (line 114) | def _Sin(self, x):
    method _SinPlusX (line 117) | def _SinPlusX(self, x):
    method _SinPlusLargeX (line 120) | def _SinPlusLargeX(self, x):
    method _SinPlusXNd (line 123) | def _SinPlusXNd(self, x):
    method _SinOfSum (line 126) | def _SinOfSum(self, x):
    method _Max (line 129) | def _Max(self, x):
    method _ScaledSum (line 132) | def _ScaledSum(self, x):
    method _GetNonMonotonicInitializer (line 138) | def _GetNonMonotonicInitializer(self, weights):
    method _GetTrainingInputsAndLabels (line 156) | def _GetTrainingInputsAndLabels(self, config):
    method _SetDefaults (line 181) | def _SetDefaults(self, config):
    method _TestEnsemble (line 194) | def _TestEnsemble(self, config):
    method _TrainModel (line 212) | def _TrainModel(self, config):
    method testMonotonicityOneD (line 276) | def testMonotonicityOneD(self):
    method testMonotonicityTwoD (line 338) | def testMonotonicityTwoD(self):
    method testMonotonicity5d (line 417) | def testMonotonicity5d(self):
    method testMonotonicityEquivalence (line 481) | def testMonotonicityEquivalence(self, monotonicities):
    method testMonotonicity10dAlmostMonotone (line 500) | def testMonotonicity10dAlmostMonotone(self):
    method testMonotonicity10dSinOfSum (line 530) | def testMonotonicity10dSinOfSum(self):
    method testInitializerType (line 567) | def testInitializerType(self, initializer, expected_loss):
    method testAssertMonotonicity (line 588) | def testAssertMonotonicity(self):
    method testAssertBounds (line 663) | def testAssertBounds(self, output_min, output_max, kernel_initializer,
    method testOutputBounds (line 697) | def testOutputBounds(self, units, input_dims, output_min, output_max,
    method testConstraints (line 736) | def testConstraints(self, lattice_sizes, units, dims, num_terms, outpu...
    method testInputOutOfBounds (line 810) | def testInputOutOfBounds(self):
    method testHighDimensionsStressTest (line 845) | def testHighDimensionsStressTest(self):
    method testGraphSize (line 877) | def testGraphSize(self, lattice_sizes, input_dims, num_terms,
    method testCreateKernelInitializer (line 900) | def testCreateKernelInitializer(self, kernel_initializer_id, expected_...
    method testSavingLoadingScale (line 915) | def testSavingLoadingScale(self):
    method testOutputShapeForDifferentInputTypes (line 965) | def testOutputShapeForDifferentInputTypes(self, batch_size, dims, units):

FILE: tensorflow_lattice/python/lattice_layer.py
  class Lattice (line 41) | class Lattice(keras.layers.Layer):
    method __init__ (line 162) | def __init__(self,
    method build (line 370) | def build(self, input_shape):
    method call (line 442) | def call(self, inputs):
    method compute_output_shape (line 466) | def compute_output_shape(self, input_shape):
    method get_config (line 476) | def get_config(self):
    method finalize_constraints (line 505) | def finalize_constraints(self):
    method assert_constraints (line 519) | def assert_constraints(self, eps=1e-6):
  function create_kernel_initializer (line 547) | def create_kernel_initializer(kernel_initializer_id,
  class LinearInitializer (line 653) | class LinearInitializer(keras.initializers.Initializer):
    method __init__ (line 674) | def __init__(self,
    method __call__ (line 708) | def __call__(self, shape, dtype=None, partition_info=None):
    method get_config (line 728) | def get_config(self):
  class RandomMonotonicInitializer (line 740) | class RandomMonotonicInitializer(keras.initializers.Initializer):
    method __init__ (line 754) | def __init__(self, lattice_sizes, output_min, output_max, unimodalitie...
    method __call__ (line 778) | def __call__(self, shape, dtype=None, partition_info=None):
    method get_config (line 794) | def get_config(self):
  class LatticeConstraints (line 805) | class LatticeConstraints(keras.constraints.Constraint):
    method __init__ (line 817) | def __init__(self,
    method __call__ (line 883) | def __call__(self, w):
    method get_config (line 919) | def get_config(self):
  class TorsionRegularizer (line 938) | class TorsionRegularizer(keras.regularizers.Regularizer):
    method __init__ (line 967) | def __init__(self, lattice_sizes, l1=0.0, l2=0.0):
    method __call__ (line 985) | def __call__(self, x):
    method get_config (line 992) | def get_config(self):
  class LaplacianRegularizer (line 1001) | class LaplacianRegularizer(keras.regularizers.Regularizer):
    method __init__ (line 1036) | def __init__(self, lattice_sizes, l1=0.0, l2=0.0):
    method __call__ (line 1061) | def __call__(self, x):
    method get_config (line 1068) | def get_config(self):

FILE: tensorflow_lattice/python/lattice_lib.py
  function evaluate_with_simplex_interpolation (line 32) | def evaluate_with_simplex_interpolation(inputs, kernel, units, lattice_s...
  function evaluate_with_hypercube_interpolation (line 150) | def evaluate_with_hypercube_interpolation(inputs, kernel, units, lattice...
  function compute_interpolation_weights (line 193) | def compute_interpolation_weights(inputs, lattice_sizes, clip_inputs=True):
  function batch_outer_operation (line 275) | def batch_outer_operation(list_of_tensors, operation="auto"):
  function _clip_onto_lattice_range (line 337) | def _clip_onto_lattice_range(inputs, lattice_sizes):
  function _bucketize_consequtive_equal_dims (line 372) | def _bucketize_consequtive_equal_dims(inputs, lattice_sizes):
  function default_init_params (line 416) | def default_init_params(output_min, output_max):
  function linear_initializer (line 441) | def linear_initializer(lattice_sizes,
  function _linspace (line 524) | def _linspace(start, stop, num):
  function random_monotonic_initializer (line 531) | def random_monotonic_initializer(lattice_sizes,
  function _approximately_project_monotonicity (line 620) | def _approximately_project_monotonicity(weights, lattice_sizes, monotoni...
  function _approximately_project_edgeworth (line 710) | def _approximately_project_edgeworth(weights, lattice_sizes, units,
  function _approximately_project_trapezoid (line 812) | def _approximately_project_trapezoid(weights, lattice_sizes, units,
  function _trapezoid_violation_update (line 949) | def _trapezoid_violation_update(differences, units, any_edgeworth,
  function _approximately_project_bounds (line 986) | def _approximately_project_bounds(weights, units, output_min, output_max):
  function finalize_constraints (line 1039) | def finalize_constraints(weights,
  function _project_partial_monotonicity (line 1109) | def _project_partial_monotonicity(weights, lattice_sizes, monotonicities,
  function _project_partial_edgeworth (line 1218) | def _project_partial_edgeworth(weights, lattice_sizes, edgeworth_trust,
  function _project_partial_trapezoid (line 1317) | def _project_partial_trapezoid(weights, lattice_sizes, trapezoid_trust,
  function _project_partial_monotonic_dominance (line 1419) | def _project_partial_monotonic_dominance(weights, lattice_sizes,
  function _project_partial_range_dominance (line 1510) | def _project_partial_range_dominance(weights, lattice_sizes, range_domin...
  function _project_partial_joint_monotonicity (line 1603) | def _project_partial_joint_monotonicity(weights, lattice_sizes,
  function _project_partial_joint_unimodality (line 1695) | def _project_partial_joint_unimodality(weights, lattice_sizes,
  function _project_onto_hyperplane (line 1766) | def _project_onto_hyperplane(weights, joint_unimodalities, hyperplane,
  function project_by_dykstra (line 1830) | def project_by_dykstra(weights,
  function laplacian_regularizer (line 2074) | def laplacian_regularizer(weights, lattice_sizes, l1=0.0, l2=0.0):
  function torsion_regularizer (line 2157) | def torsion_regularizer(weights, lattice_sizes, l1=0.0, l2=0.0):
  function _verify_dominances_hyperparameters (line 2245) | def _verify_dominances_hyperparameters(dominances, dominance_type,
  function verify_hyperparameters (line 2291) | def verify_hyperparameters(lattice_sizes,
  function assert_constraints (line 2524) | def assert_constraints(weights,
  function _unstack_nested_lists (line 2726) | def _unstack_nested_lists(tensor_or_list, axis):
  function _unstack_nd (line 2734) | def _unstack_nd(tensor, dims):
  function _stack_nested_lists (line 2749) | def _stack_nested_lists(tensor_or_list, axis):
  function _stack_nd (line 2757) | def _stack_nd(tensor, dims):
  function _get_element (line 2772) | def _get_element(lists, indices):
  function _set_element (line 2780) | def _set_element(lists, indices, value):
  function _reverse_second_list_dimension (line 2788) | def _reverse_second_list_dimension(layers):

FILE: tensorflow_lattice/python/lattice_test.py
  class LatticeTest (line 36) | class LatticeTest(parameterized.TestCase, tf.test.TestCase):
    method setUp (line 38) | def setUp(self):
    method _ResetAllBackends (line 46) | def _ResetAllBackends(self):
    method _ScatterXUniformly (line 50) | def _ScatterXUniformly(self, num_points, lattice_sizes):
    method _ScatterXUniformlyExtendedRange (line 64) | def _ScatterXUniformlyExtendedRange(self, num_points, lattice_sizes):
    method _SameValueForAllDims (line 78) | def _SameValueForAllDims(self, num_points, lattice_sizes):
    method _TwoDMeshGrid (line 93) | def _TwoDMeshGrid(self, num_points, lattice_sizes):
    method _TwoDMeshGridExtendedRange (line 105) | def _TwoDMeshGridExtendedRange(self, num_points, lattice_sizes):
    method _Sin (line 117) | def _Sin(self, x):
    method _SinPlusX (line 120) | def _SinPlusX(self, x):
    method _SinPlusLargeX (line 123) | def _SinPlusLargeX(self, x):
    method _SinPlusXNd (line 126) | def _SinPlusXNd(self, x):
    method _SinOfSum (line 129) | def _SinOfSum(self, x):
    method _Square (line 132) | def _Square(self, x):
    method _Max (line 135) | def _Max(self, x):
    method _WeightedSum (line 138) | def _WeightedSum(self, x):
    method _MixedSignWeightedSum (line 144) | def _MixedSignWeightedSum(self, x):
    method _PseudoLinear (line 151) | def _PseudoLinear(self, x):
    method _ScaledSum (line 160) | def _ScaledSum(self, x):
    method _GetMultiOutputInitializer (line 166) | def _GetMultiOutputInitializer(self, weights):
    method _GetTrainingInputsAndLabels (line 176) | def _GetTrainingInputsAndLabels(self, config):
    method _SetDefaults (line 200) | def _SetDefaults(self, config):
    method _TestEnsemble (line 223) | def _TestEnsemble(self, config):
    method _TrainModel (line 238) | def _TrainModel(self, config):
    method testMonotonicityOneD (line 308) | def testMonotonicityOneD(self):
    method testMonotonicityTwoD (line 360) | def testMonotonicityTwoD(self):
    method testMonotonicity5d (line 441) | def testMonotonicity5d(self):
    method testMonotonicityEquivalence (line 494) | def testMonotonicityEquivalence(self, monotonicities):
    method testMonotonicity10dAlmostMonotone (line 512) | def testMonotonicity10dAlmostMonotone(self):
    method testMonotonicity10dSinOfSum (line 541) | def testMonotonicity10dSinOfSum(self):
    method testSimpleTrustTwoD (line 573) | def testSimpleTrustTwoD(self, edgeworth_trusts, trapezoid_trusts,
    method testDenseTrustTwoD (line 602) | def testDenseTrustTwoD(self, edgeworth_trusts, trapezoid_trusts,
    method testSimpleTrust4D (line 632) | def testSimpleTrust4D(self, edgeworth_trusts, trapezoid_trusts,
    method testMultiDenseTrust4D (line 661) | def testMultiDenseTrust4D(self, edgeworth_trusts, trapezoid_trusts,
    method testEdgeworthTrustEquivalence (line 691) | def testEdgeworthTrustEquivalence(self, edgeworth_trusts):
    method testSimpleMonotonicDominance2D (line 717) | def testSimpleMonotonicDominance2D(self, monotonic_dominances, expecte...
    method testDenseMonotonicDominance2D (line 744) | def testDenseMonotonicDominance2D(self, monotonic_dominances, expected...
    method testDenseMonotonicDominance5D (line 771) | def testDenseMonotonicDominance5D(self, monotonic_dominances, expected...
    method testSimpleRangeDominance2D (line 799) | def testSimpleRangeDominance2D(self, range_dominances, expected_loss):
    method testDenseRangeDominance2D (line 826) | def testDenseRangeDominance2D(self, range_dominances, expected_loss, e...
    method testDenseRangeDominance5D (line 853) | def testDenseRangeDominance5D(self, range_dominances, expected_loss):
    method testSimpleJointMonotonicity2D (line 881) | def testSimpleJointMonotonicity2D(self, joint_monotonicities, expected...
    method testJointUnimodality1D (line 908) | def testJointUnimodality1D(self, joint_unimodalities, expected_loss):
    method testJointUnimodality2DSinOfSum (line 936) | def testJointUnimodality2DSinOfSum(self):
    method testJointUnimodality2DWshaped (line 966) | def testJointUnimodality2DWshaped(self, joint_unimodalities, expected_...
    method testJointUnimodality2OutOf4D (line 1008) | def testJointUnimodality2OutOf4D(self, joint_unimodalities):
    method testJointUnimodality3D (line 1062) | def testJointUnimodality3D(self):
    method testDenseJointMonotonicity2D (line 1088) | def testDenseJointMonotonicity2D(self, joint_monotonicities, expected_...
    method testDenseJointMonotonicity5D (line 1113) | def testDenseJointMonotonicity5D(self, joint_monotonicities, expected_...
    method testInitializerType (line 1144) | def testInitializerType(self, initializer, expected_loss):
    method _MergeDicts (line 1163) | def _MergeDicts(self, x, y):
    method testLinearMonotonicInitializer (line 1168) | def testLinearMonotonicInitializer(self):
    method testUnimodalInitializer (line 1248) | def testUnimodalInitializer(self):
    method testRandomMonotonicInitializer (line 1280) | def testRandomMonotonicInitializer(self):
    method testAssertMonotonicity (line 1326) | def testAssertMonotonicity(self):
    method testBounds (line 1353) | def testBounds(self):
    method testInputOutOfBounds (line 1403) | def testInputOutOfBounds(self):
    method testRegularizers2d (line 1447) | def testRegularizers2d(self, regularizer, pure_reg_loss, training_loss):
    method testRegularizersLargeLattice (line 1482) | def testRegularizersLargeLattice(self, regularizer, expected_loss):
    method testHighDimensionsStressTest (line 1498) | def testHighDimensionsStressTest(self):
    method testUnimodalityOneD (line 1539) | def testUnimodalityOneD(self, monotonicities, unimodalities, expected_...
    method testUnimodalityTwoD (line 1581) | def testUnimodalityTwoD(self, monotonicities, unimodalities, expected_...
    method testUnconstrained (line 1612) | def testUnconstrained(self):
    method testEqaulySizedDimsOptimization (line 1885) | def testEqaulySizedDimsOptimization(self, lattice_sizes, expected_loss):
    method testGraphSize (line 1907) | def testGraphSize(self, lattice_sizes, expected_graph_size):
    method testCreateKernelInitializer (line 1968) | def testCreateKernelInitializer(self, kernel_initializer_id, lattice_s...
    method testSimplexInterpolation (line 2060) | def testSimplexInterpolation(self, lattice_sizes, kernel, inputs,
    method testFinalizeConstraints (line 2117) | def testFinalizeConstraints(self, lattice_sizes, kernel, edgeworth_tru...

FILE: tensorflow_lattice/python/linear_layer.py
  class Linear (line 42) | class Linear(keras.layers.Layer):
    method __init__ (line 92) | def __init__(self,
    method build (line 197) | def build(self, input_shape):
    method call (line 273) | def call(self, inputs):
    method compute_output_shape (line 288) | def compute_output_shape(self, input_shape):
    method get_config (line 293) | def get_config(self):
    method assert_constraints (line 327) | def assert_constraints(self, eps=1e-4):
  class LinearConstraints (line 350) | class LinearConstraints(keras.constraints.Constraint):
    method __init__ (line 378) | def __init__(self, monotonicities, monotonic_dominances=None,
    method __call__ (line 403) | def __call__(self, w):
    method get_config (line 425) | def get_config(self):

FILE: tensorflow_lattice/python/linear_lib.py
  function project (line 28) | def project(weights,
  function assert_constraints (line 114) | def assert_constraints(weights,
  function verify_hyperparameters (line 211) | def verify_hyperparameters(num_input_dims=None,

FILE: tensorflow_lattice/python/linear_test.py
  class LinearTest (line 43) | class LinearTest(parameterized.TestCase, tf.test.TestCase):
    method setUp (line 46) | def setUp(self):
    method _ResetAllBackends (line 50) | def _ResetAllBackends(self):
    method _ScaterXUniformly (line 54) | def _ScaterXUniformly(self, num_points, num_dims, input_min, input_max):
    method _TwoDMeshGrid (line 68) | def _TwoDMeshGrid(self, num_points, num_dims, input_min, input_max):
    method _GenLinearFunction (line 80) | def _GenLinearFunction(self, weights, bias=0.0, noise=None):
    method _SinPlusXPlusD (line 96) | def _SinPlusXPlusD(self, x):
    method _SetDefaults (line 99) | def _SetDefaults(self, config):
    method _GetTrainingInputsAndLabels (line 116) | def _GetTrainingInputsAndLabels(self, config):
    method _TrainModel (line 142) | def _TrainModel(self, config):
    method _NegateAndTrain (line 214) | def _NegateAndTrain(self, config):
    method testOneDUnconstrained (line 235) | def testOneDUnconstrained(self, use_bias, expected_loss):
    method testTwoDUnconstrained (line 256) | def testTwoDUnconstrained(self, use_bias, expected_loss):
    method testInitializers (line 277) | def testInitializers(self):
    method testAssertConstraints (line 298) | def testAssertConstraints(self):
    method testOneDMonotonicities_MonotonicInput (line 334) | def testOneDMonotonicities_MonotonicInput(self, use_bias, expected_loss):
    method testOneDMonotonicities_AntiMonotonicInput (line 357) | def testOneDMonotonicities_AntiMonotonicInput(self, use_bias, expected...
    method testOneDNormalizationOrder (line 380) | def testOneDNormalizationOrder(self, norm_order, weight):
    method testOneDNormalizationOrderZeroWeights (line 401) | def testOneDNormalizationOrderZeroWeights(self):
    method testTwoDMonotonicity (line 433) | def testTwoDMonotonicity(self, expected_loss, monotonicities):
    method testTwoDNormalizationOrder (line 476) | def testTwoDNormalizationOrder(self, norm_order, weights, monotonicities,
    method testFiveDAllConstraints (line 507) | def testFiveDAllConstraints(self, weights, monotonicities, expected_lo...
    method testTwoDMonotonicDominance (line 533) | def testTwoDMonotonicDominance(self, expected_loss, dominances):
    method testTwoDRangeDominance (line 558) | def testTwoDRangeDominance(self, dominances, monotonicities, weights,
    method testRegularizers (line 587) | def testRegularizers(self, regularizer):

FILE: tensorflow_lattice/python/model_info.py
  class ModelGraph (line 28) | class ModelGraph(
  class InputFeatureNode (line 41) | class InputFeatureNode(
  class PWLCalibrationNode (line 53) | class PWLCalibrationNode(
  class CategoricalCalibrationNode (line 69) | class CategoricalCalibrationNode(
  class LinearNode (line 82) | class LinearNode(
  class LatticeNode (line 94) | class LatticeNode(
  class KroneckerFactoredLatticeNode (line 104) | class KroneckerFactoredLatticeNode(
  class MeanNode (line 119) | class MeanNode(collections.namedtuple('MeanNode', ['input_nodes'])):

FILE: tensorflow_lattice/python/parallel_combination_layer.py
  class ParallelCombination (line 39) | class ParallelCombination(keras.layers.Layer):
    method __init__ (line 79) | def __init__(self, calibration_layers=None, single_output=True, **kwar...
    method append (line 116) | def append(self, calibration_layer):
    method build (line 120) | def build(self, input_shape):
    method call (line 135) | def call(self, inputs):
    method compute_output_shape (line 155) | def compute_output_shape(self, input_shape):
    method get_config (line 161) | def get_config(self):

FILE: tensorflow_lattice/python/parallel_combination_test.py
  class ParallelCombinationTest (line 34) | class ParallelCombinationTest(parameterized.TestCase, tf.test.TestCase):
    method setUp (line 36) | def setUp(self):
    method testParallelCombinationSingleInput (line 41) | def testParallelCombinationSingleInput(self):
    method testParallelCombinationMultipleInputs (line 74) | def testParallelCombinationMultipleInputs(self):
    method testParallelCombinationClone (line 109) | def testParallelCombinationClone(self):

FILE: tensorflow_lattice/python/premade.py
  class CalibratedLatticeEnsemble (line 61) | class CalibratedLatticeEnsemble(keras.Model):
    method __init__ (line 86) | def __init__(self, model_config=None, dtype=tf.float32, **kwargs):
    method get_config (line 144) | def get_config(self):
    method from_config (line 153) | def from_config(cls, config, custom_objects=None):
  class CalibratedLattice (line 163) | class CalibratedLattice(keras.Model):
    method __init__ (line 187) | def __init__(self, model_config=None, dtype=tf.float32, **kwargs):
    method get_config (line 254) | def get_config(self):
    method from_config (line 263) | def from_config(cls, config, custom_objects=None):
  class CalibratedLinear (line 273) | class CalibratedLinear(keras.Model):
    method __init__ (line 297) | def __init__(self, model_config=None, dtype=tf.float32, **kwargs):
    method get_config (line 367) | def get_config(self):
    method from_config (line 376) | def from_config(cls, config, custom_objects=None):
  class AggregateFunction (line 388) | class AggregateFunction(keras.Model):
    method __init__ (line 410) | def __init__(self, model_config=None, dtype=tf.float32, **kwargs):
    method get_config (line 485) | def get_config(self):
    method from_config (line 494) | def from_config(cls, config, custom_objects=None):
  function get_custom_objects (line 504) | def get_custom_objects(custom_objects=None):

FILE: tensorflow_lattice/python/premade_lib.py
  function _input_calibration_regularizers (line 83) | def _input_calibration_regularizers(model_config, feature_config):
  function _middle_calibration_regularizers (line 93) | def _middle_calibration_regularizers(model_config):
  function _output_calibration_regularizers (line 102) | def _output_calibration_regularizers(model_config):
  function _lattice_regularizers (line 109) | def _lattice_regularizers(model_config, feature_configs):
  class LayerOutputRange (line 138) | class LayerOutputRange(enum.Enum):
  function _output_range (line 145) | def _output_range(layer_output_range, model_config, feature_config=None):
  function build_input_layer (line 176) | def build_input_layer(feature_configs, dtype, ragged=False):
  function build_multi_unit_calibration_layers (line 202) | def build_multi_unit_calibration_layers(calibration_input_layer,
  function build_calibration_layers (line 291) | def build_calibration_layers(calibration_input_layer, model_config,
  function build_aggregation_layer (line 360) | def build_aggregation_layer(aggregation_input_layer, model_config,
  function _monotonicities_from_feature_configs (line 441) | def _monotonicities_from_feature_configs(feature_configs):
  function _dominance_constraints_from_feature_configs (line 455) | def _dominance_constraints_from_feature_configs(feature_configs):
  function _canonical_feature_names (line 471) | def _canonical_feature_names(model_config, feature_names=None):
  function build_linear_layer (line 482) | def build_linear_layer(linear_input, feature_configs, model_config,
  function build_lattice_layer (line 535) | def build_lattice_layer(lattice_input, feature_configs, model_config,
  function build_lattice_ensemble_layer (line 654) | def build_lattice_ensemble_layer(submodels_inputs, model_config, dtype):
  function build_rtl_layer (line 689) | def build_rtl_layer(calibration_outputs, model_config, submodel_index,
  function build_calibrated_lattice_ensemble_layer (line 762) | def build_calibrated_lattice_ensemble_layer(calibration_input_layer,
  function build_linear_combination_layer (line 826) | def build_linear_combination_layer(ensemble_outputs, model_config, dtype):
  function build_output_calibration_layer (line 870) | def build_output_calibration_layer(output_calibration_input, model_config,
  function set_categorical_monotonicities (line 903) | def set_categorical_monotonicities(feature_configs):
  function set_random_lattice_ensemble (line 945) | def set_random_lattice_ensemble(model_config, feature_names=None):
  function _add_pair_to_ensemble (line 984) | def _add_pair_to_ensemble(lattices, lattice_rank, i, j):
  function _set_all_pairs_cover_lattices (line 1012) | def _set_all_pairs_cover_lattices(prefitting_model_config, feature_names):
  function construct_prefitting_model_config (line 1029) | def construct_prefitting_model_config(model_config, feature_names=None):
  function _verify_prefitting_model (line 1078) | def _verify_prefitting_model(prefitting_model, feature_names):
  function _get_lattice_weights (line 1112) | def _get_lattice_weights(prefitting_model, lattice_index):
  function _get_torsions_and_laplacians (line 1127) | def _get_torsions_and_laplacians(prefitting_model_config, prefitting_model,
  function _get_final_crystal_lattices (line 1171) | def _get_final_crystal_lattices(model_config, prefitting_model_config,
  function set_crystals_lattice_ensemble (line 1304) | def set_crystals_lattice_ensemble(model_config,
  function _weighted_quantile (line 1357) | def _weighted_quantile(sorted_values, quantiles, weights):
  function compute_keypoints (line 1392) | def compute_keypoints(values,
  function _feature_config_by_name (line 1489) | def _feature_config_by_name(feature_configs, feature_name, add_if_missing):
  function compute_feature_keypoints (line 1501) | def compute_feature_keypoints(feature_configs,
  function set_feature_keypoints (line 1536) | def set_feature_keypoints(feature_configs, feature_keypoints,
  function compute_label_keypoints (line 1547) | def compute_label_keypoints(model_config,
  function set_label_keypoints (line 1579) | def set_label_keypoints(model_config, label_keypoints):
  function _verify_ensemble_config (line 1584) | def _verify_ensemble_config(model_config):
  function _verify_kronecker_factored_config (line 1666) | def _verify_kronecker_factored_config(model_config):
  function _verify_aggregate_function_config (line 1717) | def _verify_aggregate_function_config(model_config):
  function _verify_feature_config (line 1739) | def _verify_feature_config(feature_config):
  function verify_config (line 1788) | def verify_config(model_config):

FILE: tensorflow_lattice/python/premade_test.py
  class PremadeTest (line 105) | class PremadeTest(parameterized.TestCase, tf.test.TestCase):
    method setUp (line 108) | def setUp(self):
    method _ResetAllBackends (line 266) | def _ResetAllBackends(self):
    class Encoder (line 270) | class Encoder(json.JSONEncoder):
      method default (line 272) | def default(self, o):
    method testSetRandomLattices (line 279) | def testSetRandomLattices(self):
    method testSetCategoricalMonotonicities (line 306) | def testSetCategoricalMonotonicities(self):
    method testVerifyConfig (line 312) | def testVerifyConfig(self):
    method testLatticeEnsembleFromConfig (line 346) | def testLatticeEnsembleFromConfig(self):
    method testLatticeFromConfig (line 370) | def testLatticeFromConfig(self):
    method testLatticeSimplexFromConfig (line 389) | def testLatticeSimplexFromConfig(self):
    method testLinearFromConfig (line 409) | def testLinearFromConfig(self):
    method testAggregateFromConfig (line 429) | def testAggregateFromConfig(self):
    method testCalibratedLatticeEnsembleCrystals (line 456) | def testCalibratedLatticeEnsembleCrystals(self, interpolation,
    method testCalibratedLatticeEnsembleRTL (line 529) | def testCalibratedLatticeEnsembleRTL(self, interpolation, parameteriza...
    method testCalibratedLattice (line 584) | def testCalibratedLattice(self, interpolation, parameterization, num_t...
    method testLearnedCalibrationInputKeypoints (line 628) | def testLearnedCalibrationInputKeypoints(self):
    method testLatticeEnsembleH5FormatSaveLoad (line 707) | def testLatticeEnsembleH5FormatSaveLoad(self, parameterization, num_te...
    method testLatticeEnsembleRTLH5FormatSaveLoad (line 752) | def testLatticeEnsembleRTLH5FormatSaveLoad(self, parameterization, num...
    method testLatticeH5FormatSaveLoad (line 797) | def testLatticeH5FormatSaveLoad(self, parameterization, num_terms):
    method testLinearH5FormatSaveLoad (line 833) | def testLinearH5FormatSaveLoad(self):
    method testAggregateH5FormatSaveLoad (line 860) | def testAggregateH5FormatSaveLoad(self):

FILE: tensorflow_lattice/python/pwl_calibration_layer.py
  class PWLCalibration (line 49) | class PWLCalibration(keras.layers.Layer):
    method __init__ (line 102) | def __init__(self,
    method build (line 300) | def build(self, input_shape):
    method call (line 396) | def call(self, inputs):
    method compute_output_shape (line 503) | def compute_output_shape(self, input_shape):
    method get_config (line 511) | def get_config(self):
    method assert_constraints (line 538) | def assert_constraints(self, eps=1e-6):
    method keypoints_outputs (line 582) | def keypoints_outputs(self):
    method keypoints_inputs (line 589) | def keypoints_inputs(self):
  class UniformOutputInitializer (line 614) | class UniformOutputInitializer(keras.initializers.Initializer):
    method __init__ (line 627) | def __init__(self, output_min, output_max, monotonicity, keypoints=None):
    method __call__ (line 655) | def __call__(self, shape, dtype=None, partition_info=None):
    method get_config (line 678) | def get_config(self):
  class PWLCalibrationConstraints (line 688) | class PWLCalibrationConstraints(keras.constraints.Constraint):
    method __init__ (line 699) | def __init__(
    method __call__ (line 752) | def __call__(self, w):
    method get_config (line 765) | def get_config(self):
  class NaiveBoundsConstraints (line 779) | class NaiveBoundsConstraints(keras.constraints.Constraint):
    method __init__ (line 789) | def __init__(self, lower_bound=None, upper_bound=None):
    method __call__ (line 799) | def __call__(self, w):
    method get_config (line 807) | def get_config(self):
  class LaplacianRegularizer (line 815) | class LaplacianRegularizer(keras.regularizers.Regularizer):
    method __init__ (line 832) | def __init__(self, l1=0.0, l2=0.0, is_cyclic=False):
    method __call__ (line 845) | def __call__(self, x):
    method get_config (line 873) | def get_config(self):
  class HessianRegularizer (line 882) | class HessianRegularizer(keras.regularizers.Regularizer):
    method __init__ (line 904) | def __init__(self, l1=0.0, l2=0.0, is_cyclic=False):
    method __call__ (line 917) | def __call__(self, x):
    method get_config (line 953) | def get_config(self):
  class WrinkleRegularizer (line 962) | class WrinkleRegularizer(keras.regularizers.Regularizer):
    method __init__ (line 984) | def __init__(self, l1=0.0, l2=0.0, is_cyclic=False):
    method __call__ (line 997) | def __call__(self, x):
    method get_config (line 1037) | def get_config(self):

FILE: tensorflow_lattice/python/pwl_calibration_lib.py
  class BoundConstraintsType (line 27) | class BoundConstraintsType(enum.Enum):
  function convert_all_constraints (line 39) | def convert_all_constraints(output_min, output_max, clamp_min, clamp_max):
  function _convert_constraints (line 73) | def _convert_constraints(value, clamp_to_value):
  function compute_interpolation_weights (line 95) | def compute_interpolation_weights(inputs, keypoints, lengths):
  function linear_initializer (line 129) | def linear_initializer(shape,
  function _approximately_project_bounds_only (line 196) | def _approximately_project_bounds_only(bias, heights, output_min, output...
  function _project_bounds_considering_monotonicity (line 244) | def _project_bounds_considering_monotonicity(bias, heights, monotonicity,
  function _project_convexity (line 360) | def _project_convexity(heights, lengths, convexity, constraint_group):
  function _project_monotonicity (line 462) | def _project_monotonicity(heights, monotonicity):
  function project_all_constraints (line 472) | def project_all_constraints(weights,
  function _squeeze_by_scaling (line 649) | def _squeeze_by_scaling(bias, heights, monotonicity, output_min, output_...
  function _approximately_project_convexity (line 700) | def _approximately_project_convexity(heights, lengths, convexity):
  function _finalize_constraints (line 729) | def _finalize_constraints(bias, heights, monotonicity, output_min, outpu...
  function assert_constraints (line 805) | def assert_constraints(outputs,
  function verify_hyperparameters (line 888) | def verify_hyperparameters(input_keypoints=None,

FILE: tensorflow_lattice/python/pwl_calibration_test.py
  class CalibrateWithSeparateMissing (line 46) | class CalibrateWithSeparateMissing(keras.layers.Layer):
    method __init__ (line 53) | def __init__(self, calibration_layer, missing_input_value):
    method call (line 58) | def call(self, x):
  class PwlCalibrationLayerTest (line 64) | class PwlCalibrationLayerTest(parameterized.TestCase, tf.test.TestCase):
    method setUp (line 66) | def setUp(self):
    method _ResetAllBackends (line 73) | def _ResetAllBackends(self):
    method _ScatterXUniformly (line 77) | def _ScatterXUniformly(self, units, num_points, input_min, input_max,
    method _ScatterXUniformlyIncludeBounds (line 94) | def _ScatterXUniformlyIncludeBounds(self, units, **kwargs):
    method _SmallWaves (line 101) | def _SmallWaves(self, x):
    method _SmallWavesPlusOne (line 105) | def _SmallWavesPlusOne(self, x):
    method _WavyParabola (line 108) | def _WavyParabola(self, x):
    method _SinCycle (line 112) | def _SinCycle(self, x):
    method _GenPWLFunction (line 116) | def _GenPWLFunction(self, input_keypoints, pwl_weights):
    method _SetDefaults (line 138) | def _SetDefaults(self, config):
    method _TrainModel (line 174) | def _TrainModel(self, config):
    method _InverseAndTrain (line 271) | def _InverseAndTrain(self, config):
    method _CreateTrainingData (line 293) | def _CreateTrainingData(self, config):
    method _CreateKerasLayer (line 306) | def _CreateKerasLayer(self, config):
    method testUnconstrainedNoMissingValue (line 339) | def testUnconstrainedNoMissingValue(self, units, one_d_input, expected...
    method testUnconstrainedWithMissingValue (line 373) | def testUnconstrainedWithMissingValue(self, units, missing_output_value,
    method testNonMonotonicFunction (line 419) | def testNonMonotonicFunction(self, units, output_min, output_max, opti...
    method testBoundsForMissing (line 458) | def testBoundsForMissing(self, units, missing_input_value, expected_lo...
    method testAllBoundsWithoutMonotonicityConstraints (line 522) | def testAllBoundsWithoutMonotonicityConstraints(self, units, output_min,
    method testMonotonicProperBounds (line 562) | def testMonotonicProperBounds(self, units, is_clamped, optimizer,
    method testMonotonicNarrowBounds (line 601) | def testMonotonicNarrowBounds(self, units, is_clamped, optimizer,
    method testMonotonicWideBounds (line 640) | def testMonotonicWideBounds(self, units, is_clamped, optimizer,
    method testAllBoundsAndMonotonicityDirection (line 798) | def testAllBoundsAndMonotonicityDirection(self, units, output_min, out...
    method testConvexitySimple (line 837) | def testConvexitySimple(self, units, convexity, expected_loss):
    method testConvexityNonUniformKeypoints (line 871) | def testConvexityNonUniformKeypoints(self, units, convexity, expected_...
    method testConvexityDifferentNumKeypoints (line 911) | def testConvexityDifferentNumKeypoints(self, units, num_keypoints,
    method testConvexityWithMonotonicityAndBounds (line 952) | def testConvexityWithMonotonicityAndBounds(self, units, monotonicity,
    method testInputKeypoints (line 989) | def testInputKeypoints(self, keypoints):
    method testIsCyclic (line 1017) | def testIsCyclic(self, units, regularizer, num_training_epoch, expecte...
    method testInitializer (line 1050) | def testInitializer(self, units, initializer, expected_loss):
    method testRegularizers (line 1094) | def testRegularizers(self, units, regularizer, pure_reg_loss, training...
    method testAssertMonotonicity (line 1130) | def testAssertMonotonicity(self):
    method testOutputShape (line 1172) | def testOutputShape(self):
    method testKeypointsInputs (line 1199) | def testKeypointsInputs(self, input_keypoints_type, input_dims, output...

FILE: tensorflow_lattice/python/rtl_layer.py
  class RTL (line 56) | class RTL(keras.layers.Layer):
    method __init__ (line 130) | def __init__(self,
    method build (line 295) | def build(self, input_shape):
    method call (line 377) | def call(self, x, **kwargs):
    method compute_output_shape (line 430) | def compute_output_shape(self, input_shape):
    method get_config (line 452) | def get_config(self):
    method finalize_constraints (line 478) | def finalize_constraints(self):
    method assert_constraints (line 492) | def assert_constraints(self, eps=1e-6):
    method _get_rtl_structure (line 509) | def _get_rtl_structure(self, input_shape):

FILE: tensorflow_lattice/python/rtl_lib.py
  function verify_hyperparameters (line 23) | def verify_hyperparameters(lattice_size,

FILE: tensorflow_lattice/python/rtl_test.py
  class RTLTest (line 35) | class RTLTest(parameterized.TestCase, tf.test.TestCase):
    method setUp (line 37) | def setUp(self):
    method testRTLInputShapes (line 42) | def testRTLInputShapes(self):
    method testRTLOutputShape (line 142) | def testRTLOutputShape(self):
    method testRTLSaveLoad (line 161) | def testRTLSaveLoad(self):

FILE: tensorflow_lattice/python/test_utils.py
  class TimeTracker (line 27) | class TimeTracker(object):
    method __init__ (line 40) | def __init__(self, list_to_append, num_steps=1):
    method __enter__ (line 44) | def __enter__(self):
    method __exit__ (line 48) | def __exit__(self, unuesd_type, unuesd_value, unuesd_traceback):
  function run_training_loop (line 54) | def run_training_loop(config,
  function two_dim_mesh_grid (line 113) | def two_dim_mesh_grid(num_points, x_min, y_min, x_max, y_max):
  function sample_uniformly (line 158) | def sample_uniformly(num_points, lower_bounds, upper_bounds):
  function get_hypercube_interpolation_fn (line 189) | def get_hypercube_interpolation_fn(coefficients):
  function get_linear_lattice_interpolation_fn (line 224) | def get_linear_lattice_interpolation_fn(lattice_sizes, monotonicities,

FILE: tensorflow_lattice/python/utils.py
  function canonicalize_convexity (line 25) | def canonicalize_convexity(convexity):
  function canonicalize_input_bounds (line 55) | def canonicalize_input_bounds(input_bounds):
  function canonicalize_monotonicity (line 85) | def canonicalize_monotonicity(monotonicity, allow_decreasing=True):
  function canonicalize_monotonicities (line 128) | def canonicalize_monotonicities(monotonicities, allow_decreasing=True):
  function canonicalize_trust (line 157) | def canonicalize_trust(trusts):
  function canonicalize_unimodalities (line 196) | def canonicalize_unimodalities(unimodalities):
  function count_non_zeros (line 232) | def count_non_zeros(*iterables):

FILE: tensorflow_lattice/python/utils_test.py
  class UtilsTest (line 25) | class UtilsTest(parameterized.TestCase, tf.test.TestCase):
    method testCanonicalizeConvexity (line 29) | def testCanonicalizeConvexity(self, convexity,
    method testInvalidConvexity (line 36) | def testInvalidConvexity(self, invalid_convexity):
    method testCanonicalizeInputBounds (line 56) | def testCanonicalizeInputBounds(self, input_bounds,
    method testInvalidInputBounds (line 64) | def testInvalidInputBounds(self, invalid_input_bounds):
    method testCanonicalizeMonotonicity (line 73) | def testCanonicalizeMonotonicity(self, monotonicity,
    method testInvalidMonotonicity (line 81) | def testInvalidMonotonicity(self, invalid_monotonicity):
    method testInvalidDecreasingMonotonicity (line 89) | def testInvalidDecreasingMonotonicity(self, invalid_monotonicity):
    method testCanonicalizeMonotonicities (line 103) | def testCanonicalizeMonotonicities(self, monotonicities,
    method testCanonicalizeTrust (line 115) | def testCanonicalizeTrust(self, trusts, expected_canonicalized_trusts):
    method testInvalidTrustDirection (line 125) | def testInvalidTrustDirection(self, invalid_trusts):
    method testInvalidTrustLength (line 138) | def testInvalidTrustLength(self, invalid_trusts):
    method testCountNonZeros (line 149) | def testCountNonZeros(self, monotonicities, unimodalities,
    method testCanonicalizeUnimodalities (line 157) | def testCanonicalizeUnimodalities(self, unimodalities,
    method testInvalidUnimoadlities (line 168) | def testInvalidUnimoadlities(self, invalid_unimodalities):