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