3 references to DecoderLayerOutput
Microsoft.ML.GenAI.Mistral (3)
MistralDecoderLayer.cs (2)
59
internal class MistralDecoderLayer : nn.Module<DecoderLayerInput,
DecoderLayerOutput
>, IDynamicLoadModule
108
public override
DecoderLayerOutput
forward(DecoderLayerInput input)
MistralModel.cs (1)
132
var
layerOutput = layer.forward(decoderInput);