JSX_TTS / torch /include /ATen /native /ConvolutionMM3d.h
UMMJ's picture
Upload 5875 files
9dd3461
raw
history blame contribute delete
345 Bytes
#include <ATen/ATen.h>
namespace at {
namespace 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