Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now


Class PrunableLayer

Abstract Base Class for making your own keras layer prunable.

Defined in python/core/sparsity/keras/

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




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.