| #include "caffe2/operators/feed_blob_op.h" | |
| namespace caffe2 { | |
| REGISTER_CPU_OPERATOR(FeedBlob, FeedBlobOp<CPUContext>); | |
| SHOULD_NOT_DO_GRADIENT(FeedBlob); | |
| OPERATOR_SCHEMA(FeedBlob) | |
| .NumInputs(0, 0) | |
| .NumOutputs(1, 1) | |
| .SetDoc(R"DOC( | |
| FeedBlobs the content of the blobs. The input and output blobs should be | |
| one-to-one inplace.)DOC") | |
| .Arg( | |
| "value", | |
| "(string) if provided then we will use this string as the value for the" | |
| "provided output tensor"); | |
| } // namespace caffe2 |