2 implementations of SetIncrementalState
Microsoft.ML.TorchSharp (2)
NasBert\Modules\IncrementalState.cs (1)
74
public void
SetIncrementalState
(
NasBert\Modules\MultiHeadAttention.cs (1)
377
public void
SetIncrementalState
(torch.nn.Module module, Dictionary<string, Dictionary<string, torch.Tensor>> incrementalState, string key, Dictionary<string, torch.Tensor> value)
1 reference to SetIncrementalState
Microsoft.ML.TorchSharp (1)
NasBert\Modules\MultiHeadAttention.cs (1)
379
_incrementalState.
SetIncrementalState
(module, incrementalState, key, value);