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

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

weighted_modules = [
    'Conv1D', 'Conv2D', 'Conv3D', 'Conv1DTranspose', 'Conv2DTranspose', 'Conv3DTranspose',
    'Dense',
    'PReLU',
    'Embedding',
]