Lines Matching full:linear

21         self.linear = nn.Linear(16, 16, bias=bias)
23             nn.Linear(16, 16, bias=bias), nn.Linear(16, 16, bias=bias)
27         self.linear.weight = nn.Parameter(torch.zeros_like(self.linear.weight) + 1.0)
31             self.linear = nn.Parameter(torch.zeros_like(self.linear.bias) + 10.0)
36         x = self.linear(x)
43         model = nn.Linear(16, 16, bias=False)
58         assert not hasattr(model.linear, "parametrizations")
63             model.linear, "weight", utils.FakeSparsity(mask)
74         assert hasattr(model.linear, "parametrizations")
75         assert parametrize.is_parametrized(model.linear, "weight")
77         assert parametrize.is_parametrized(model.linear, "weight")
79         assert parametrize.is_parametrized(model.linear, "weight")
86             model_save.linear, "weight", utils.FakeSparsity(mask)
99         mask = torch.zeros(model_load.linear.weight.shape)
101             model_load.linear, "weight", utils.FakeSparsity(mask)
115         assert hasattr(model_load.linear, "parametrizations")
116         assert parametrize.is_parametrized(model_load.linear, "weight")
118         assert parametrize.is_parametrized(model_load.linear, "weight")
120         assert parametrize.is_parametrized(model_load.linear, "weight")
124             model_save.linear.parametrizations["weight"].original,
125             model_load.linear.parametrizations["weight"].original,
140             model_save.linear.parametrizations["weight"][0].mask,
141             model_load.linear.parametrizations["weight"][0].mask,
157             model.linear, "weight", utils.FakeSparsity(mask)