diff options
Diffstat (limited to 'caffe2/operators/sigmoid_op.cc')
| -rw-r--r-- | caffe2/operators/sigmoid_op.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/caffe2/operators/sigmoid_op.cc b/caffe2/operators/sigmoid_op.cc index cd8b2681a9..2734f003f9 100644 --- a/caffe2/operators/sigmoid_op.cc +++ b/caffe2/operators/sigmoid_op.cc @@ -21,6 +21,7 @@ struct SigmoidGradientCPUFunctor { } }; +namespace { REGISTER_CPU_OPERATOR( Sigmoid, UnaryElementwiseOp< TensorTypes<float>, CPUContext, SigmoidCPUFunctor>); @@ -64,4 +65,5 @@ class GetSigmoidGradient : public GradientMakerBase { } }; REGISTER_GRADIENT(Sigmoid, GetSigmoidGradient); +} // namespace } // namespace caffe2 |