Searched defs:bias_k (Results 1 – 4 of 4) sorted by relevance
45 Tensor bias_k, in MultiheadAttentionForwardFuncOptions()
858 Tensor bias_k; variable
3593 torch::Tensor bias_k; in _multihead_attn_test_helper() local
14915 bias_k, # type: Optional[Tensor] argument