ConvolutionMM3d.h 594 B

12345678910111213141516171819
  1. #if !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
  2. #include <ATen/core/Tensor.h>
  3. namespace at::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
  13. #else
  14. #error "This file should not be included when either TORCH_STABLE_ONLY or TORCH_TARGET_VERSION is defined."
  15. #endif // !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)