ConvolutionMM3d.h 352 B

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