Searched refs:attn_output_weights_expected (Results 1 – 1 of 1) sorted by relevance
536 auto attn_output_weights_expected = torch::tensor( in TEST_F() local546 attn_output_weights, attn_output_weights_expected, 1e-05, 2e-04)); in TEST_F()