Caffe2 - C++ API
A deep learning, cross platform ML framework
conditional_op.h
1 
18 #ifndef CONDITIONAL_OP_H
19 #define CONDITIONAL_OP_H
20 
21 #include "caffe2/core/context.h"
22 #include "caffe2/core/operator.h"
23 #include "caffe2/core/tensor.h"
24 
25 namespace caffe2 {
26 
27 template <class Context>
28 class ConditionalOp final : public Operator<Context> {
29  public:
30  USE_OPERATOR_CONTEXT_FUNCTIONS;
31  ConditionalOp(const OperatorDef& operator_def, Workspace* ws)
32  : Operator<Context>(operator_def, ws) {}
33 
34  bool RunOnDevice() override;
35 };
36 
37 } // caffe2
38 
39 #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.