Home
last modified time | relevance | path

Searched defs:vulkan_output (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dvulkan_api_test.cpp1498 const auto vulkan_output = callOpByName( in test_conv2d_context() local
1539 const auto vulkan_output = callOpByName( in test_backwards_compatible_conv2d_context() local
1581 const auto vulkan_output = callOpByName( in test_transposed_conv2d_context() local
2755 auto vulkan_output = callOpByName( in test_packed_layer_norm() local
7460 auto vulkan_output = std::get<0>(out_vulkan); in TEST_F() local
7525 auto vulkan_output = std::get<0>(out_vulkan); in TEST_F() local
7691 auto vulkan_output = out_vulkan[0].toTensor(); in TEST_F() local
7856 auto vulkan_output = callOpByName( in test_linear() local
7971 auto vulkan_output = std::get<0>(out_vulkan); in TEST_F() local
8047 auto vulkan_output = std::get<0>(out_vulkan); in TEST_F() local
[all …]
H A Dvulkan_quantized_api_test.cpp1860 auto vulkan_output = callOpByName( in test_quantized_conv_transpose2d() local
2666 const auto vulkan_output = callOpByName( in test_quantized_conv2d() local