/usr/local/lib64/python3.6/site-packages/torch/include/ATen/native/cpu/
/usr/local/lib64/python3.6/site-packages/torch/include/ATen/native/cpu
NameSizeModeActions
AtomicAddFloat.h7500644editdlrm
avx_mathfun.h174480644editdlrm
CatKernel.h2520644editdlrm
DepthwiseConvKernel.h4470644editdlrm
DistributionTemplates.h140510644editdlrm
GridSamplerKernel.h5920644editdlrm
Intrinsics.h12120644editdlrm
IsContiguous.h24140644editdlrm
Loops.h144110644editdlrm
MaxUnpoolKernel.h4830644editdlrm
moments_utils.h42450644editdlrm
Reduce.h109550644editdlrm
SoftmaxKernel.h6360644editdlrm
StackKernel.h3150644editdlrm
utils.h10930644editdlrm
zmath.h64800644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/ATen/native/cpu/MaxUnpoolKernel.h (483B)
#include #include #include #pragma once namespace at { namespace native { using max_unpooling_fn = void(*)(Tensor&, const Tensor&, const Tensor&); DECLARE_DISPATCH(max_unpooling_fn, max_unpool2d_kernel); DECLARE_DISPATCH(max_unpooling_fn, max_unpool2d_backward_kernel); DECLARE_DISPATCH(max_unpooling_fn, max_unpool3d_kernel); DECLARE_DISPATCH(max_unpooling_fn, max_unpool3d_backward_kernel); }} // at::native