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);
93
VProjection = torch.nn.Linear(
_vDim
, _embeddingDim, _addBiasProj);