Caffe2 - C++ API
A deep learning, cross platform ML framework
channel_backprop_stats_op.h
1 
17 #ifndef CHANNEL_BACKPROP_STATS_OP_H
18 #define CHANNEL_BACKPROP_STATS_OP_H
19 
20 #include "caffe2/core/context.h"
21 #include "caffe2/core/operator.h"
22 #include "caffe2/utils/math.h"
23 
24 namespace caffe2 {
25 
26 template <class Context>
27 class ChannelBackpropStatsOp : public Operator<Context> {
28  public:
29  USE_OPERATOR_CONTEXT_FUNCTIONS;
30  ChannelBackpropStatsOp(const OperatorDef& operator_def, Workspace* ws)
31  : Operator<Context>(operator_def, ws) {}
33 
34  bool RunOnDevice() override {
35  return true;
36  }
37 
38  protected:
39  INPUT_TAGS(INPUT, SAVED_MEAN, SAVED_INV_STDDEV, OUTPUT_GRAD);
40  OUTPUT_TAGS(SCALE_GRAD, BIAS_GRAD);
41 };
42 
43 } // namespace caffe2
44 
45 #endif
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.