Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/test/
H A Dvulkan_api_test.cpp189 const auto out_vulkan = at::slice(in_vulkan, dim, start, end, step); in slice_test() local
222 const auto out_vulkan = at::clone(in_vulkan, optional_memory_format); in clone_test() local
369 auto out_vulkan = in_vulkan.cpu(); in test_copy_to_texture_bool() local
421 const auto out_vulkan = at::adaptive_avg_pool2d(in_cpu.vulkan(), {3, 3}); in TEST_F() local
439 const auto out_vulkan = at::add(in_vulkan, other_vulkan, alpha); in test_add() local
507 const auto out_vulkan = at::add(in_vulkan, other_cpu, alpha); in test_add_other_cpu_int() local
716 const auto out_vulkan = at::addmm(bias_cpu, m1_vulkan, m2_cpu, beta, alpha); in TEST_F() local
736 const auto out_vulkan = at::addmm(bias_cpu, m1_vulkan, m2_cpu, beta, alpha); in TEST_F() local
756 const auto out_vulkan = at::addmm(bias_cpu, m1_vulkan, m2_cpu, beta, alpha); in TEST_F() local
781 const auto out_vulkan = at::avg_pool2d(in_cpu.vulkan(), {5, 3}, {1, 2}, {2, 0}, true); in TEST_F() local
[all …]
H A Dvulkan_quantized_api_test.cpp711 const auto out_vulkan = at::native::vulkan::ops::quantize_per_tensor( in TEST_F() local
888 const auto out_vulkan = at::native::vulkan::ops::quantize_per_tensor( in TEST_F() local
1174 const auto out_vulkan = at::native::vulkan::ops::quantize_per_tensor( in TEST_F() local
1220 const auto out_vulkan = at::native::vulkan::ops::quantize_per_tensor( in TEST_F() local
1272 const auto out_vulkan = at::native::vulkan::ops::quantize_per_tensor( in TEST_F() local
1324 const auto out_vulkan = at::native::vulkan::ops::quantize_per_tensor( in TEST_F() local
1376 const auto out_vulkan = at::native::vulkan::ops::quantize_per_tensor( in TEST_F() local
1425 const auto out_vulkan = at::native::vulkan::ops::quantize_per_tensor( in TEST_F() local
1527 const auto out_vulkan = at::native::vulkan::ops::quantize_per_tensor( in test_conv2d() local
1635 const auto out_vulkan = at::native::vulkan::ops::quantize_per_tensor( in TEST_F() local
[all …]