4 implementations of LoadToDeviceFunc
Microsoft.ML.GenAI.Core (1)
Module\DynamicLoadingModule.cs (1)
33
public Action<nn.Module>?
LoadToDeviceFunc
{ get; set; }
Microsoft.ML.GenAI.LLaMA (1)
Module\LlamaDecoderLayer.cs (1)
77
public Action<nn.Module>?
LoadToDeviceFunc
{ get; set; }
Microsoft.ML.GenAI.Mistral (1)
MistralDecoderLayer.cs (1)
71
public Action<nn.Module>?
LoadToDeviceFunc
{ get; set; }
Microsoft.ML.GenAI.Phi (1)
Module\Phi3DecoderLayer.cs (1)
100
public Action<nn.Module>?
LoadToDeviceFunc
{ get; set; }
1 write to LoadToDeviceFunc
Microsoft.ML.GenAI.Core (1)
Extension\ModuleExtension.cs (1)
138
dynamicModule.
LoadToDeviceFunc
= (nn.Module module) =>