b84549f
1
2
3
4
5
6
7
8
9
10
# Copyright (c) Microsoft Corporation. # Licensed under the MIT license. weighted_modules = [ 'Conv1d', 'Conv2d', 'Conv3d', 'ConvTranspose1d', 'ConvTranspose2d', 'ConvTranspose3d', 'Linear', 'Bilinear', 'PReLU', 'Embedding', 'EmbeddingBag', ]