/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/StackKernel.h (315B)
// Copyright 2004-present Facebook. All Rights Reserved. #pragma once #include #include namespace at { namespace native { using stack_serial_fn = void(*)(Tensor &, TensorList, int64_t); DECLARE_DISPATCH(stack_serial_fn, stack_serial_stub); }} // namespace at::native