:py:mod:`neural_compressor.adaptor.torch_utils.pattern_detector` ================================================================ .. py:module:: neural_compressor.adaptor.torch_utils.pattern_detector .. autoapi-nested-parse:: Block detector for Transformer-based model. Module Contents --------------- Classes ~~~~~~~ .. autoapisummary:: neural_compressor.adaptor.torch_utils.pattern_detector.TransformerBasedModelBlockPatternDetector .. py:class:: TransformerBasedModelBlockPatternDetector(model: torch, pattern_lst: List[List[Union[str, int]]] = BLOCK_PATTERNS) Detect the attention block and FFN block in transformer-based model.