Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_cuda_cpp_wrapper.py220 tests=test_select_algorithm.TestSelectAlgorithm(),
224 tests=test_select_algorithm.TestSelectAlgorithm(),
239 tests=test_select_algorithm.TestSelectAlgorithm(),
243 tests=test_select_algorithm.TestSelectAlgorithm(),
H A Dtest_cpu_select_algorithm.py142 class TestSelectAlgorithm(BaseTestSelectAlgorithm): class
1459 test_linear_dynamic_shapes = TestSelectAlgorithm.test_linear_static_shapes
1461 TestSelectAlgorithm.test_linear_with_pointwise
1464 TestSelectAlgorithm.test_linear_with_transpose
1467 TestSelectAlgorithm.test_linear_with_unary_binary
1469 test_linear_amx_dynamic_shapes = TestSelectAlgorithm.test_linear_amx
1471 TestSelectAlgorithm.test_linear_with_embedding
1474 TestSelectAlgorithm.test_quantized_linear_with_pointwise
1477 TestSelectAlgorithm.test_quantized_linear_with_pointwise_binary
1480 TestSelectAlgorithm.test_quantized_linear_amx
[all …]
H A Dtest_select_algorithm.py49 class TestSelectAlgorithm(TestCase): class