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