/usr/local/lib64/python3.6/site-packages/torch/include/caffe2/utils/math
NameSizeModeActions
broadcast.h5550644editdlrm
elementwise.h64550644editdlrm
half_utils.h13290644editdlrm
reduce.h26720644editdlrm
transpose.h7850644editdlrm
utils.h46430644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/caffe2/utils/math/broadcast.h (555B)
#ifndef CAFFE2_UTILS_MATH_BROADCAST_H_ #define CAFFE2_UTILS_MATH_BROADCAST_H_ #include "caffe2/core/common.h" #include "caffe2/core/types.h" namespace caffe2 { namespace math { bool can_use_broadcast_fastpath(int ndim, const int* dims); template TORCH_API void AffineChannel( const int N, const int C, const int HxW, const T* X, const T* scale, const T* bias, T* Y, Context* context); } // namespace math } // namespace caffe2 #endif // CAFFE2_UTILS_MATH_BROADCAST_H_