Home
last modified time | relevance | path

Searched refs:_multihead_attn_test_helper (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/test/nn/
H A Dtest_multihead_attention.py139 def _multihead_attn_test_helper( function
405 _multihead_attn_test_helper(add_bias_kv=True)
408 _multihead_attn_test_helper(add_zero_attn=True)
411 _multihead_attn_test_helper()
414 _multihead_attn_test_helper(add_key_padding_mask=True)
417 _multihead_attn_test_helper(saved_kv=True)
420 _multihead_attn_test_helper(
425 _multihead_attn_test_helper(
430 _multihead_attn_test_helper(
438 _multihead_attn_test_helper(
/aosp_15_r20/external/pytorch/test/cpp/api/
H A Dmodules.cpp3502 void _multihead_attn_test_helper( in _multihead_attn_test_helper() function
3782 _multihead_attn_test_helper( in TEST_F()
3791 _multihead_attn_test_helper( in TEST_F()
3800 _multihead_attn_test_helper(); in TEST_F()
3803 _multihead_attn_test_helper( in TEST_F()
3812 _multihead_attn_test_helper( in TEST_F()
3821 _multihead_attn_test_helper( in TEST_F()
3830 _multihead_attn_test_helper( in TEST_F()
3840 _multihead_attn_test_helper( in TEST_F()
3850 _multihead_attn_test_helper( in TEST_F()