1 write to blocks
Microsoft.ML.TorchSharp (1)
AutoFormerV2\BasicLayer.cs (1)
48
this.
blocks
= new ModuleList<AutoFormerV2Block>();
4 references to blocks
Microsoft.ML.TorchSharp (4)
AutoFormerV2\BasicLayer.cs (4)
51
this.
blocks
.Add(new AutoFormerV2Block(inChannels: inChannels, numHeads: numHeads, windowSize: windowSize, shiftSize: (i % 2 == 0) ? 0 : (windowSize / 2), mlpRatio: mlpRatio, dropRatio: dropRatio, localConvSize: localConvSize, useShiftWindow: useShiftWindow, useInterpolate: useInterpolate));
103
for (int i = 0; i < this.
blocks
.Count; i++)
105
x = this.
blocks
[i].forward(x, h, w, attnMask);
144
blocks
.Dispose();