Sign in
android
/
platform
/
external
/
pytorch
/
ab6afc2b23
/
.
/
caffe2
/
operators
/
transpose_op.cu
blob: d5e9507dfdc54ff81034be63a102728059fe6a3b [
file
] [
log
] [
blame
]
#include
"caffe2/core/context_gpu.h"
#include
"caffe2/operators/transpose_op.h"
namespace
caffe2
{
REGISTER_CUDA_OPERATOR
(
Transpose
,
TransposeOp
<
CUDAContext
>);
}
// namespace caffe2