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 |