1 write to FinalLayerNorm
Microsoft.ML.TorchSharp (1)
NasBert\Modules\Layers\FeedForwardLayer.cs (1)
60
FinalLayerNorm
= torch.nn.LayerNorm(new long[] { embeddingDim });
3 references to FinalLayerNorm
Microsoft.ML.TorchSharp (3)
NasBert\Modules\Layers\FeedForwardLayer.cs (3)
70
return
FinalLayerNorm
.forward(layerOutput);
73
public override void CloseLayerNormTraining() =>
FinalLayerNorm
.eval();
85
FinalLayerNorm
.Dispose();