TensorFlow 2.0 Beta is available Learn more

tfmot.sparsity.keras.PrunableLayer

Class PrunableLayer

Abstract Base Class for making your own keras layer prunable.

Defined in python/core/sparsity/keras/prunable_layer.py.

Custom keras layers which want to add pruning should implement this class.

Methods

get_prunable_weights

get_prunable_weights()

Returns list of prunable weight tensors.

All the weight tensors which the layer wants to be pruned during training must be returned by this method.

Returns: List of weight tensors/kernels in the keras layer which must be pruned during training.