1 instantiation of DecoderLayerOutput
Microsoft.ML.GenAI.Mistral (1)
MistralDecoderLayer.cs (1)
143
return new
DecoderLayerOutput
(
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);