Caffe2 - C++ API
A deep learning, cross platform ML framework
logit_op.h
1 #include "caffe2/core/context.h"
2 #include "caffe2/core/operator.h"
3 #include "caffe2/utils/math.h"
4 
5 namespace caffe2 {
6 
7 template <typename T, class Context>
8 class LogitGradientOp final : public Operator<Context> {
9  public:
10  USE_OPERATOR_CONTEXT_FUNCTIONS;
11  LogitGradientOp(const OperatorDef& operator_def, Workspace* ws)
12  : Operator<Context>(operator_def, ws),
13  eps_(OperatorBase::GetSingleArgument<float>("eps", 1e-6f)) {}
14  ~LogitGradientOp() {}
15 
16  bool RunOnDevice() override;
17 
18  protected:
19  float eps_;
20 };
21 
22 } // namespace caffe2
Workspace is a class that holds all the related objects created during runtime: (1) all blobs...
Definition: workspace.h:63
Copyright (c) 2016-present, Facebook, Inc.