Sign in
android
/
platform
/
external
/
pytorch
/
f1ce15b50c
/
.
/
caffe2
/
sgd
/
lars_op_gpu.cu
blob: 6008da9313005d547176580bdb63d0d30ae43d9f [
file
] [
log
] [
blame
]
#include
"caffe2/core/context_gpu.h"
#include
"caffe2/operators/operator_fallback_gpu.h"
#include
"caffe2/sgd/lars_op.h"
namespace
caffe2
{
REGISTER_CUDA_OPERATOR
(
Lars
,
GPUFallbackOp
<
LarsOp
<
float
,
CPUContext
>>);
}