2 implementations of GetIncrementalState
Microsoft.ML.TorchSharp (2)
NasBert\Modules\IncrementalState.cs (1)
64
public Dictionary<string, torch.Tensor>
GetIncrementalState
(
NasBert\Modules\MultiHeadAttention.cs (1)
372
public Dictionary<string, torch.Tensor>
GetIncrementalState
(torch.nn.Module module, Dictionary<string, Dictionary<string, torch.Tensor>> incrementalState, string key)
1 reference to GetIncrementalState
Microsoft.ML.TorchSharp (1)
NasBert\Modules\MultiHeadAttention.cs (1)
374
return _incrementalState.
GetIncrementalState
(module, incrementalState, key);