Searched refs:test_quantized_linear (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_quantized_api_test.cpp | 3472 void test_quantized_linear( in test_quantized_linear() function
3492 test_quantized_linear({1, 100}, {1, 100}, {1}); in TEST_F()
3496 test_quantized_linear({2, 3}, {4, 3}, {4}); in TEST_F()
3500 test_quantized_linear({1287, 17}, {23, 17}, {23}); in TEST_F()
3504 test_quantized_linear({1, 1, 37}, {41, 37}, {41}); in TEST_F()
3508 test_quantized_linear({2, 3, 4}, {5, 4}, {5}); in TEST_F()
3512 test_quantized_linear({23, 17, 41}, {15, 41}, {15}); in TEST_F()
3516 test_quantized_linear({1, 1, 1, 37}, {41, 37}, {41}); in TEST_F()
3520 test_quantized_linear({2, 3, 4, 5}, {6, 5}, {6}); in TEST_F()
3524 test_quantized_linear({9, 13, 11, 17}, {23, 17}, {23}); in TEST_F()
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_aot_inductor.py | 991 def test_quantized_linear(self): member in AOTInductorTestsTemplate
|
| /aosp_15_r20/external/pytorch/test/onnx/ |
| H A D | test_pytorch_onnx_onnxruntime.py | 12732 def test_quantized_linear(self): member in TestONNXRuntime
|