1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
#pragma once
#include <torch/csrc/autograd/custom_function.h>
namespace torch::lazy {
struct MaxPool3dAutogradFunctionTS
: public torch::autograd::Function<MaxPool3dAutogradFunctionTS> {
static at::Tensor forward(
torch::autograd::AutogradContext* ctx,
const 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 torch::lazy
|
