Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_flex_decoding.py405 test_input_strides = [ variable in TestFlexDecoding
414 @common_utils.parametrize("k_s", test_input_strides)
415 @common_utils.parametrize("v_s", test_input_strides)