Searched defs:attn_output_weights (Results 1 – 2 of 2) sorted by relevance
884 auto attn_output_weights = torch::bmm(q, k.transpose(1, 2)); variable
535 auto attn_output_weights = std::get<1>(output); in TEST_F() local