utils/third_party/nni/compression/pytorch/default_layers.py

# Copyright (c) Microsoft Corporation.
# Licensed under the MIT license.

weighted_modules = [
    'Conv1d', 'Conv2d', 'Conv3d', 'ConvTranspose1d', 'ConvTranspose2d', 'ConvTranspose3d',
    'Linear', 'Bilinear',
    'PReLU',
    'Embedding', 'EmbeddingBag',
]