#ifndef CONDITIONAL_OP_H #define CONDITIONAL_OP_H #include "caffe2/core/context.h" #include "caffe2/core/operator.h" #include "caffe2/core/tensor.h" namespace caffe2 { template class ConditionalOp final : public Operator { public: USE_OPERATOR_CONTEXT_FUNCTIONS; template explicit ConditionalOp(Args&&... args) : Operator(std::forward(args)...) {} bool RunOnDevice() override; }; } // caffe2 #endif