ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: tfl.layers

'layers' namespace for TFL layers.


class Aggregation: Layer which represents an aggregation function.

class CDF: Cumulative Distribution Function (CDF) layer.

class CategoricalCalibration: Categorical calibration layer with monotonicity and bound constraints.

class KroneckerFactoredLattice: Kronecker-Factored Lattice layer.

class Lattice: Lattice layer.

class Linear: Layer which represents linear function.

class PWLCalibration: Piecewise linear calibration layer.

class ParallelCombination: Wraps several parallel calibration layers under single one.

class RTL: Layer which includes a random ensemble of lattices.