1 write to d
Microsoft.ML.TorchSharp (1)
AutoFormerV2\Attention.cs (1)
54
this.
d
= attnRatio * keyChannels;
2 references to d
Microsoft.ML.TorchSharp (2)
AutoFormerV2\Attention.cs (2)
55
this.dh = this.
d
* numHeads;
108
var tmp = qkv.split(new long[] { this.keyChannels, this.keyChannels, this.
d
}, dim: 3);