1 write to key
Microsoft.ML.TorchSharp (1)
Roberta\Modules\AttentionSelf.cs (1)
38key = torch.nn.Linear(hiddenSize, hiddenSize, true);
2 references to key
Microsoft.ML.TorchSharp (2)
Roberta\Modules\AttentionSelf.cs (2)
49var mixedKeyLayer = key.forward(hiddenStates); 93key.Dispose();