1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19
|
#include "caffe2/operators/stop_gradient.h"
namespace caffe2 {
REGISTER_CPU_OPERATOR(StopGradient, StopGradientOp<CPUContext>);
// TODO(jiayq): Add example to the doc string.
OPERATOR_SCHEMA(StopGradient)
.NumInputs(1, 1)
.NumOutputs(1, 1)
.AllowInplace({{0, 0}})
.IdenticalTypeAndShape()
.SetDoc(R"DOC(
StopGradient is a helper operator that does no actual numerical computation,
and in the gradient computation phase stops the gradient from being computed
through it.
)DOC");
NO_GRADIENT(StopGradient);
} // namespace caffe2
|