2 types derived from TransformerCell
Microsoft.ML.TorchSharp (2)
NasBert\Modules\TransformerCell.cs (2)
30
internal sealed class TransformerCellNonDiscrete :
TransformerCell
100
internal sealed class TransformerCellDiscrete :
TransformerCell
5 references to TransformerCell
Microsoft.ML.TorchSharp (5)
NasBert\Models\NasBertEncoder.cs (4)
43
private readonly ModuleList<
TransformerCell
> Layers;
117
Layers = new ModuleList<
TransformerCell
>(layers);
231
var
layer = Layers[i];
282
foreach (
var
layer in Layers)
NasBert\Modules\TransformerCell.cs (1)
117
: base(nameof(
TransformerCell
))