1 #import <ATen/native/metal/MetalConvParams.h> 2 #import <ATen/native/metal/MetalPrepackOpContext.h> 3 #include <c10/util/ArrayRef.h> 4 5 namespace at::native::metal { 6 7 Tensor conv2d( 8 const Tensor& input, 9 const Tensor& weight, 10 const std::optional<at::Tensor>& bias, 11 IntArrayRef stride, 12 IntArrayRef padding, 13 IntArrayRef dilation, 14 int64_t groups); 15 16 namespace prepack { 17 Tensor conv2d(const Tensor& input, Conv2dOpContext& context); 18 } 19 20 } // namespace at::native::metal 21