Name	Size	Mode	Actions
broadcast.h	555	0644	editdlrm
elementwise.h	6455	0644	editdlrm
half_utils.h	1329	0644	editdlrm
reduce.h	2672	0644	editdlrm
transpose.h	785	0644	editdlrm
utils.h	4643	0644	editdlrm

#ifndef CAFFE2_UTILS_MATH_TRANSPOSE_H_ #define CAFFE2_UTILS_MATH_TRANSPOSE_H_ #include "caffe2/core/common.h" #include "caffe2/core/types.h" namespace caffe2 { namespace math { // Transpose tensor X with dims by axes and write the result to tensor Y. template TORCH_API void Transpose( int ndim, const TIndex* dims, const int* axes, const TData* X, TData* Y, Context* context); template TORCH_API void NCHW2NHWC(int N, int C, int HxW, const T* X, T* Y, Context* context); template TORCH_API void NHWC2NCHW(int N, int C, int HxW, const T* X, T* Y, Context* context); } // namespace math } // namespace caffe2 #endif // CAFFE2_UTILS_MATH_TRANSPOSE_H_