pytorch/caffe2/operators/lpnorm_op.cu

11 lines
275 B
Plaintext
Raw Normal View History

2023-11-08 09:01:59 -07:00
#include "caffe2/core/context_gpu.h"
#include "caffe2/operators/lpnorm_op.h"
#include "caffe2/operators/operator_fallback_gpu.h"
namespace caffe2 {
REGISTER_CUDA_OPERATOR(LpNorm, GPUFallbackOp);
REGISTER_CUDA_OPERATOR(LpNormGradient, GPUFallbackOp);
} // namespace caffe2