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