@@ -523,14 +523,13 @@ Tensor mean(
523
523
}
524
524
525
525
TORCH_LIBRARY_IMPL (aten, Vulkan, m) {
526
- m.impl (" slice.Tensor" , TORCH_FN (slice));
527
- m.impl (" reshape" , TORCH_FN (reshape));
528
- m.impl (" select.int" , TORCH_FN (select ));
529
- m.impl (" transpose.int" , TORCH_FN (transpose));
530
- m.impl (" _cat" , TORCH_FN (cat));
531
- m.impl_UNBOXED (" transpose_" , transpose_);
532
- m.impl (" view" , TORCH_FN (view));
533
- m.impl (" unsqueeze" , TORCH_FN (unsqueeze));
526
+ m.impl (" slice.Tensor" , TORCH_FN (at::native::vulkan::aten::slice));
527
+ m.impl (" reshape" , TORCH_FN (at::native::vulkan::aten::reshape));
528
+ m.impl (" select.int" , TORCH_FN (at::native::vulkan::aten::select ));
529
+ m.impl (" transpose.int" , TORCH_FN (at::native::vulkan::aten::transpose));
530
+ m.impl_UNBOXED (" transpose_" , at::native::vulkan::aten::transpose_);
531
+ m.impl (" view" , TORCH_FN (at::native::vulkan::aten::view));
532
+ m.impl (" unsqueeze" , TORCH_FN (at::native::vulkan::aten::unsqueeze));
534
533
m.impl_UNBOXED (" empty.memory_format" , at::native::vulkan::aten::empty);
535
534
m.impl (" empty_strided" , TORCH_FN (at::native::vulkan::aten::empty_strided));
536
535
m.impl (" add.Tensor" , TORCH_FN (at::native::vulkan::aten::add));
@@ -544,12 +543,11 @@ TORCH_LIBRARY_IMPL(aten, Vulkan, m) {
544
543
m.impl (
545
544
" _adaptive_avg_pool2d" ,
546
545
TORCH_FN (at::native::vulkan::aten::adaptive_avg_pool2d));
547
- m.impl (" avg_pool2d" , TORCH_FN (avg_pool2d));
546
+ m.impl (" avg_pool2d" , TORCH_FN (at::native::vulkan::aten:: avg_pool2d));
548
547
m.impl (" max_pool2d" , TORCH_FN (at::native::vulkan::aten::max_pool2d));
549
- m.impl (" reshape" , TORCH_FN (at::native::vulkan::aten::reshape));
550
548
m.impl (" _cat" , TORCH_FN (at::native::vulkan::aten::cat));
551
- m.impl (" mul.Scalar" , TORCH_FN (mul_scalar));
552
- m.impl (" add.Scalar" , TORCH_FN (add_scalar));
549
+ m.impl (" mul.Scalar" , TORCH_FN (at::native::vulkan::aten:: mul_scalar));
550
+ m.impl (" add.Scalar" , TORCH_FN (at::native::vulkan::aten:: add_scalar));
553
551
m.impl_UNBOXED (
554
552
" convolution_overrideable" , at::native::vulkan::aten::convolution);
555
553
m.impl_UNBOXED (" hardtanh_" , at::native::vulkan::aten::hardtanh_);
0 commit comments