Home
last modified time | relevance | path

Searched refs:MaxPool3dAutogradFunctionTS (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/lazy/ts_backend/
H A Dts_autograd_functions.h8 struct MaxPool3dAutogradFunctionTS struct
9 : public torch::autograd::Function<MaxPool3dAutogradFunctionTS> {
H A Dts_autograd_functions.cpp9 at::Tensor MaxPool3dAutogradFunctionTS::forward( in forward()
29 torch::autograd::variable_list MaxPool3dAutogradFunctionTS::backward( in backward()
H A Dts_native_functions.cpp333 return torch::lazy::MaxPool3dAutogradFunctionTS::apply( in max_pool3d()