Searched defs:d_head (Results 1 – 2 of 2) sorted by relevance
105 def _split_heads_ref(X, dims, nheads, d_head): argument113 def _combine_heads_ref(X, dims, nheads, d_head): argument
3475 int d_head) { in _split_heads_ref()3485 int d_head) { in _combine_heads_ref()3518 const auto d_head = d_3_10(generator); in _multihead_attn_test_helper() local