Searched defs:output_vulkan (Results 1 – 2 of 2) sorted by relevance
926 …const auto output_vulkan = at::batch_norm(input_vulkan, weight_vulkan, bias_vulkan, running_mean_v… in TEST_F() local955 …const auto output_vulkan = at::batch_norm(input_vulkan, weight_vulkan, bias_vulkan, running_mean_v… in TEST_F() local985 …const auto output_vulkan = at::batch_norm(input_vulkan, weight_vulkan, bias_vulkan, running_mean_v… in TEST_F() local1650 const auto output_vulkan = at::conv2d( in TEST_F() local1737 const auto output_vulkan = at::conv2d( in TEST_F() local1802 const auto output_vulkan = at::conv2d( in TEST_F() local1870 const auto output_vulkan = at::conv2d( in TEST_F() local1960 const auto output_vulkan = at::conv2d( in TEST_F() local2760 auto output_vulkan = vulkan_output[0].toTensor(); in test_packed_layer_norm() local2859 const auto output_vulkan = at::layer_norm( in test_layer_norm() local[all …]
1532 const auto output_vulkan = at::native::vulkan::ops::quantized_conv2d( in test_conv2d() local1640 const auto output_vulkan = at::native::vulkan::ops::quantized_conv2d( in TEST_F() local1743 const auto output_vulkan = at::native::vulkan::ops::quantized_conv2d( in TEST_F() local