1#pragma once
2
3#include <torch/csrc/autograd/custom_function.h>
4
5namespace torch {
6namespace lazy {
7
8struct 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