1 write to _qkvSameDim
Microsoft.ML.TorchSharp (1)
NasBert\Modules\MultiHeadAttention.cs (1)
69
_qkvSameDim
= (_kDim == _embeddingDim) && (_vDim == _embeddingDim);
2 references to _qkvSameDim
Microsoft.ML.TorchSharp (2)
NasBert\Modules\MultiHeadAttention.cs (2)
82
if (_selfAttention && !
_qkvSameDim
)
110
if (
_qkvSameDim
)