mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
19 lines
508 B
C++
19 lines
508 B
C++
#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
|