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