1 #pragma once 2 3 #include <torch/csrc/autograd/custom_function.h> 4 5 namespace torch { 6 namespace lazy { 7 8 struct MaxPool3dAutogradFunctionTS 9 : public torch::autograd::Function<MaxPool3dAutogradFunctionTS> { 10 static at::Tensor forward( 11 torch::autograd::AutogradContext* ctx, 12 at::Tensor self, 13 at::IntArrayRef kernel_size, 14 at::IntArrayRef stride, 15 at::IntArrayRef padding, 16 at::IntArrayRef dilation, 17 bool ceil_mode); 18 static torch::autograd::variable_list backward( 19 torch::autograd::AutogradContext* ctx, 20 torch::autograd::variable_list grad_output); 21 }; 22 23 } // namespace lazy 24 } // namespace torch 25