#ifndef CAFFE2_OPERATORS_FEED_BLOB_OP_H_ #define CAFFE2_OPERATORS_FEED_BLOB_OP_H_ #include "caffe2/core/context.h" #include "caffe2/core/operator.h" namespace caffe2 { template class FeedBlobOp : public Operator { public: template explicit FeedBlobOp(Args&&... args) : Operator(std::forward(args)...) { CAFFE_ENFORCE( this->template HasSingleArgumentOfType("value"), "value argument must exist and be passed as a string"); value_ = this->template GetSingleArgument("value", ""); } USE_OPERATOR_CONTEXT_FUNCTIONS; bool RunOnDevice() override { *this->template Output(0) = value_; return true; } private: std::string value_; }; } // namespace caffe2 #endif