2 writes to shiftSize
Microsoft.ML.TorchSharp (2)
AutoFormerV2\AutoFormerV2Block.cs (2)
46
this.
shiftSize
= shiftSize;
50
this.
shiftSize
= 0;
6 references to shiftSize
Microsoft.ML.TorchSharp (6)
AutoFormerV2\AutoFormerV2Block.cs (6)
95
if (this.useShiftWindow && this.
shiftSize
> 0)
97
shiftedX = torch.roll(x, shifts: new long[] { -this.
shiftSize
, -this.
shiftSize
}, dims: new long[] { 1, 2 });
113
if (this.useShiftWindow && this.
shiftSize
> 0)
115
x = torch.roll(shiftedX, shifts: new long[] { this.
shiftSize
, this.
shiftSize
}, dims: new long[] { 1, 2 });