xref: /aosp_15_r20/external/pytorch/aten/src/ATen/native/ConvolutionMM3d.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 #include <ATen/core/Tensor.h>
2 
3 namespace at::native {
4 
5 std::tuple<Tensor, Tensor, Tensor> slow_conv3d_backward_cpu(
6     const Tensor& grad_output,
7     const Tensor& self,
8     const Tensor& weight,
9     IntArrayRef kernel_size,
10     IntArrayRef stride,
11     IntArrayRef padding,
12     std::array<bool, 3> output_mask);
13 
14 } // namespace at::native
15