include/ATen/native/ConvolutionMM3d.h · duality-group / torch v2.1.2+cpu - python package | Gemfury

Gemfury

Why Gemfury? Push, build, and install RubyGems npm packages Python packages Maven artifacts PHP packages Go Modules Debian packages RPM packages NuGet packages

duality-group / torch python

Repository URL to install this package:

Details

torch / include / ATen / native / ConvolutionMM3d.h

#include <ATen/core/Tensor.h>

namespace at::native {

std::tuple<Tensor, Tensor, Tensor> slow_conv3d_backward_cpu(
    const Tensor& grad_output,
    const Tensor& self,
    const Tensor& weight,
    IntArrayRef kernel_size,
    IntArrayRef stride,
    IntArrayRef padding,
    std::array<bool, 3> output_mask);

} // namespace at::native