Searched defs:multi_head_attention_forward (Results 1 – 2 of 2) sorted by relevance
932 inline std::tuple<Tensor, Tensor> multi_head_attention_forward( in multi_head_attention_forward() function
5868 def multi_head_attention_forward( function