1 instantiation of RotaryEmbeddingOutput
Microsoft.ML.GenAI.Core (1)
Module\RotaryEmbedding.cs (1)
123return new(cos.to_type(x.dtype), sin.to_type(x.dtype));
4 references to RotaryEmbeddingOutput
Microsoft.ML.GenAI.Core (4)
Module\Attention.cs (2)
24RotaryEmbeddingOutput positionalEmbeddings, // cos, sin 42public RotaryEmbeddingOutput PositionalEmbeddings { get; set; }
Module\RotaryEmbedding.cs (2)
72RotaryEmbeddingOutput> 104public override RotaryEmbeddingOutput forward(RotaryEmbeddingInput input)