Spaces:
Running
Running
File size: 354 Bytes
c61ccee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
#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
|