`neural_compressor.compression.pruner.wanda.wrapper`

Module Contents

Classes

This class wraps a GPT layer for specific operations.

class neural_compressor.compression.pruner.wanda.wrapper.WrappedGPT(layer, layer_id=0, layer_name='none')[source]: This class wraps a GPT layer for specific operations.