1 write to attn
Microsoft.ML.TorchSharp (1)
AutoFormerV2\AutoFormerV2Block.cs (1)
58
this.
attn
= new Attention(inChannels, headChannels, numHeads, attnRatio: 1, windowResolution: windowResolution);
2 references to attn
Microsoft.ML.TorchSharp (2)
AutoFormerV2\AutoFormerV2Block.cs (2)
108
var attnWindows = this.
attn
.forward(xWindows, mask: attnMask);
193
attn
.Dispose();