3 references to DecoderLayerOutput
Microsoft.ML.GenAI.LLaMA (3)
Module\LlamaDecoderLayer.cs (2)
65
internal class LlamaDecoderLayer : nn.Module<DecoderLayerInput,
DecoderLayerOutput
>, IDynamicLoadModule
114
public override
DecoderLayerOutput
forward(DecoderLayerInput input)
Module\LlamaModel.cs (1)
138
var
layerOutput = layer.forward(decoderInput);