ConvolutionMM3d.h 340 B

1234567891011121314
  1. #include <ATen/core/Tensor.h>
  2. namespace at::native {
  3. std::tuple<Tensor, Tensor, Tensor> slow_conv3d_backward_cpu(
  4. const Tensor& grad_output,
  5. const Tensor& self,
  6. const Tensor& weight,
  7. IntArrayRef kernel_size,
  8. IntArrayRef stride,
  9. IntArrayRef padding,
  10. std::array<bool, 3> output_mask);
  11. } // namespace at::native