diff options
Diffstat (limited to 'caffe2/operators/math_ops.cc')
| -rw-r--r-- | caffe2/operators/math_ops.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/caffe2/operators/math_ops.cc b/caffe2/operators/math_ops.cc index 0e0a07a483..e64b0ba453 100644 --- a/caffe2/operators/math_ops.cc +++ b/caffe2/operators/math_ops.cc @@ -11,6 +11,7 @@ struct SqrCPUFunctor { } }; +namespace { REGISTER_CPU_OPERATOR( Sqr, UnaryElementwiseOp<TensorTypes<float>, CPUContext, SqrCPUFunctor>); @@ -99,4 +100,5 @@ class GetPowGradient : public GradientMakerBase { REGISTER_GRADIENT(Pow, GetPowGradient); +} // namespace } // namespace caffe2 |