Base:
1 override of forward
Microsoft.ML.TorchSharp (1)
NasBert\Modules\HiddenTransfer.cs (1)
45public override torch.Tensor forward(torch.Tensor x, int hiddenSize, bool inputTransfer)
2 references to forward
Microsoft.ML.TorchSharp (2)
NasBert\Models\NasBertEncoder.cs (2)
234x = HiddenTransferList[blockIndex].forward(x, HiddenSizePerBlock[blockIndex], true); 241x = HiddenTransferList[blockIndex].forward(x, HiddenSizePerBlock[blockIndex], false);