1 write to _vDim
Microsoft.ML.TorchSharp (1)
NasBert\Modules\MultiHeadAttention.cs (1)
68_vDim = vDim ?? embeddingDim;
2 references to _vDim
Microsoft.ML.TorchSharp (2)
NasBert\Modules\MultiHeadAttention.cs (2)
69_qkvSameDim = (_kDim == _embeddingDim) && (_vDim == _embeddingDim); 93VProjection = torch.nn.Linear(_vDim, _embeddingDim, _addBiasProj);