Module: tfmot.sparsity.keras.pruning_wrapper.prune_registry

View source on GitHub

Registry responsible for built-in keras classes.

Modules

prunable_layer module: Abstract Base Class for making a custom keras layer prunable.

Classes

class PruneRegistry: Registry responsible for built-in keras layers.

class TensorFlowOpLayer: Wraps a TensorFlow Operation in a Layer.