2 references to AttentionStateKey
Microsoft.ML.TorchSharp (2)
NasBert\Modules\MultiHeadAttention.cs (2)
316
return GetIncrementalState(this, incrementalState,
AttentionStateKey
) ?? new Dictionary<string, torch.Tensor>();
323
SetIncrementalState(this, incrementalState,
AttentionStateKey
, buffer);