1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
|
#include "caffe2/operators/scale_blobs_op.h"
namespace caffe2 {
REGISTER_CPU_OPERATOR(ScaleBlobs, ScaleBlobsOp<CPUContext>);
OPERATOR_SCHEMA(ScaleBlobs)
.NumInputs(1, INT_MAX)
.NumOutputs(1, INT_MAX)
.AllowInplace([](int, int) { return true; })
.IdenticalTypeAndShape()
.SetDoc(R"DOC(
ScaleBlobs takes one or more input data (Tensor) and produces one
or more output data (Tensor) whose value is the input data tensor
scaled element-wise.
)DOC")
.Arg("scale", "(float, default 1.0) the scale to apply.");
} // namespace caffe2
|
