mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
71ca600af9
Summary: Since caffe2 and torch have been consolidated, CAFFE2_API should be merged with TORCH_API. Addresses a TODO. Manually edited some references of the removed `CAFFE2_API`: * `CONTRIBUTING.md` * `caffe2/proto/CMakeLists.txt` * `cmake/ProtoBuf.cmake` * `c10/macros/Export.h` * `torch/csrc/WindowsTorchApiMacro.h` Pull Request resolved: https://github.com/pytorch/pytorch/pull/49496 Reviewed By: malfet, samestep Differential Revision: D25600726 Pulled By: janeyx99 fbshipit-source-id: 7e068d959e397ac183c097d7e9a9afeca5ddd782
32 lines
785 B
C++
32 lines
785 B
C++
#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 <typename TIndex, typename TData, class Context>
|
|
TORCH_API void Transpose(
|
|
int ndim,
|
|
const TIndex* dims,
|
|
const int* axes,
|
|
const TData* X,
|
|
TData* Y,
|
|
Context* context);
|
|
|
|
template <typename T, class Context>
|
|
TORCH_API void
|
|
NCHW2NHWC(int N, int C, int HxW, const T* X, T* Y, Context* context);
|
|
|
|
template <typename T, class Context>
|
|
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_
|