pytorch  1.8.2
About: PyTorch provides Tensor computation (like NumPy) with strong GPU acceleration and Deep Neural Networks (in Python) built on a tape-based autograd system. LTS (Long Term Support) release.
  Fossies Dox: pytorch-1.8.2.tar.gz  ("unofficial" and yet experimental doxygen-generated source code documentation)  
Public Member Functions | Static Public Member Functions | Protected Member Functions | Static Protected Member Functions | Protected Attributes | Static Private Member Functions | List of all members
caffe2::GradientMakerBase Class Reference

#include <operator_gradient.h>

Inheritance diagram for caffe2::GradientMakerBase:
[legend]
Collaboration diagram for caffe2::GradientMakerBase:
[legend]

Public Member Functions

 GradientMakerBase (const OperatorDef &def, const vector< GradientWrapper > &g_output)
 
virtual ~GradientMakerBase ()
 
virtual bool CopyDeviceOption () const
 
virtual bool CopyEngine () const
 
virtual bool CopyArguments () const
 
virtual void VerifyOp () const
 
virtual GradientOpsMeta Get ()
 Returns the gradient ops meta. More...
 
const OperatorDef & Def () const
 

Static Public Member Functions

static CaffeMap< string, string > MatchGradsToParams (const OperatorDef &op)
 Returns map that returns the parameters that the gradients are for. More...
 

Protected Member Functions

virtual vector< OperatorDef > GetGradientDefs ()
 
string I (const int i)
 
string O (const int i)
 
string GI (const int i)
 
string GI_I (const int i)
 
string GI_V (const int i)
 
string GO (const int i)
 
string GO_I (const int i)
 
string GO_V (const int i)
 
const GradientWrapper & GradOut (int i)
 
void SetDense (const int i, const string &name)
 
void SetSparse (const int i, const string &indices, const string &values)
 

Static Protected Member Functions

template<class... Args>
static vector< OperatorDef > SingleGradientDef (const Args &... args)
 a helper function to allow one to create one single operator def, which is usually the case for many simple operators. More...
 

Protected Attributes

const OperatorDef & def_
 
const vector< GradientWrapper > & g_output_
 
vector< GradientWrapper > g_input_
 

Static Private Member Functions

static string GradientName (const string &name)
 
static bool IsGradientBlob (const string &name)
 
static string GradientNameToParam (const string &name)
 
static string GradientSliceIndices (const string &name)
 
static string GradientSliceValues (const string &name)
 

Detailed Description

Definition at line 47 of file operator_gradient.h.

Constructor & Destructor Documentation

◆ GradientMakerBase()

caffe2::GradientMakerBase::GradientMakerBase ( const OperatorDef &  def,
const vector< GradientWrapper > &  g_output 
)
inline

Definition at line 49 of file operator_gradient.h.

Referenced by caffe2::Arg(), caffe2::argmaxes(), caffe2::GetCopyGradient::GetGradientDefs(), caffe2::GetSquaredL2DistanceGradient::GetGradientDefs(), caffe2::GetMatMulGradient::GetGradientDefs(), caffe2::GetNegateGradientGradient::GetGradientDefs(), caffe2::GetNormalizeGradient::GetGradientDefs(), caffe2::GetPadImageGradient::GetGradientDefs(), caffe2::GetPrependDimGradient::GetGradientDefs(), caffe2::gradient(), caffe2::IdenticalTypeAndShapeOfInput(), caffe2::Input(), caffe2::MaskWithFunctor(), caffe2::NumOutputs(), caffe2::OPERATOR_SCHEMA(), caffe2::Output(), caffe2::REGISTER_GRADIENT(), and caffe2::SetDoc().

◆ ~GradientMakerBase()

virtual caffe2::GradientMakerBase::~GradientMakerBase ( )
inlinevirtual

Definition at line 53 of file operator_gradient.h.

Member Function Documentation

◆ CopyArguments()

virtual bool caffe2::GradientMakerBase::CopyArguments ( ) const
inlinevirtual

Reimplemented in caffe2::GetCastGradient, caffe2::GetMatMulGradient, caffe2::GetPrependDimGradient, and caffe2::GetTransposeGradient.

Definition at line 60 of file operator_gradient.h.

◆ CopyDeviceOption()

virtual bool caffe2::GradientMakerBase::CopyDeviceOption ( ) const
inlinevirtual

Definition at line 54 of file operator_gradient.h.

◆ CopyEngine()

virtual bool caffe2::GradientMakerBase::CopyEngine ( ) const
inlinevirtual

Definition at line 57 of file operator_gradient.h.

◆ Def()

const OperatorDef& caffe2::GradientMakerBase::Def ( ) const
inline

Definition at line 94 of file operator_gradient.h.

References def_.

Referenced by caffe2::GetMatMulGradient::GetGradientDefs(), caffe2::SegmentOpGetGradient< ForwardOp, ReducerDef, ReducerGradient, Sorted, SparseFused >::GetGradientDefs(), caffe2::LengthsOpGetGradient< ForwardOp, ReducerDef, ReducerGradient, SparseFused, GradientNeedIndices >::GetGradientDefs(), caffe2::GetAddPaddingGradient::GetGradientDefs(), caffe2::GetRemovePaddingGradient::GetGradientDefs(), caffe2::python::GetPythonGradient::GetGradientDefs(), and caffe2::REGISTER_GRADIENT().

◆ Get()

virtual GradientOpsMeta caffe2::GradientMakerBase::Get ( )
inlinevirtual

Returns the gradient ops meta.

If your gradient op generator only use standard input and output manipulations, you can simply implement GetGradientDefs() that returns vector<OperatorDef>. In that, you can call GI, GI_V and GI_I that will automatically create the gradient registration for you.

If you need to do custom gradient name registration, overload this function directly.

Reimplemented in caffe2::ThrowInTheTowelIfGradientIsCalled, and caffe2::GradientNotImplementedYet.

Definition at line 85 of file operator_gradient.h.

◆ GetGradientDefs()

virtual vector<OperatorDef> caffe2::GradientMakerBase::GetGradientDefs ( )
inlineprotectedvirtual

Reimplemented in caffe2::NoGradient, caffe2::GetFooGradient, caffe2::GetFCDecompGradient, caffe2::GetBatchGatherGradient, caffe2::GetCastGradient, caffe2::GetConvGradient, caffe2::GetConvTransposeGradient, caffe2::GetCopyGradient, caffe2::GetGPUToCPUGradient, caffe2::GetCPUToGPUGradient, caffe2::GetLabelCrossEntropyGradient, caffe2::GetMakeTwoClassGradient, caffe2::GetSigmoidCrossEntropyWithLogitsGradient, caffe2::GetWeightedSigmoidCrossEntropyWithLogitsGradient, caffe2::GetCrossEntropyGradient, caffe2::GetSquaredL2DistanceGradient, caffe2::GetL1DistanceGradient, caffe2::GetDotProductGradient, caffe2::GetCosineSimilarityGradient, caffe2::GetDotProductWithPaddingGradient, caffe2::GetDropoutGradient, caffe2::GetElementwiseLinearGradient, caffe2::FeedBlob::GetExpGradient, caffe2::GetSqueezeGradient, caffe2::GetExpandDimsGradient, caffe2::TensorInferenceFunction::GetFCGradient, caffe2::GetGRUUnitGradient, caffe2::GetFloatToHalfGradient, caffe2::GetHalfToFloatGradient, caffe2::GetIm2ColGradient, caffe2::GetCol2ImGradient, caffe2::GetIntegralImageGradient, caffe2::GetBernoulliJSDGradient, caffe2::GetLeakyReluGradient, caffe2::GetTTSparseLengthsGradient, caffe2::GetLRNGradient, caffe2::FeedBlob::GetLogGradient, caffe2::GetAveragedLossGradient, caffe2::GetLSTMUnitGradient, caffe2::GetMatMulGradient, caffe2::GetMeanGradient, caffe2::GetNegateGradientGradient, caffe2::GetNormalizeGradient, caffe2::GetPackSegmentsGradient, caffe2::GetUnpackSegmentsGradient, caffe2::GetPadImageGradient, caffe2::GetPrependDimGradient, caffe2::GetReduceFrontMaxGradient, caffe2::GetReduceBackMaxGradient, caffe2::GetReduceFrontMeanGradient, caffe2::GetReduceFrontSumGradient, caffe2::GetReduceBackSumGradient, caffe2::GetSumElementsGradient, caffe2::GetResizeNearest3DGradient, caffe2::GetResizeNearestGradient, caffe2::GetReversePackedSegsGradient, caffe2::GetRecurrentNetworkGradient, caffe2::GetRecurrentGradient, caffe2::GetScaleGradient, caffe2::AbstractSortedSegmentRangeDef< T, SIndex, Context, ReducerDef >::GetGradient, caffe2::AbstractReduceFrontDef< T, Context, ReducerDef >::GetGradient, caffe2::AbstractReduceBackDef< T, Context, ReducerDef >::GetGradient, caffe2::SegmentOpGetGradient< ForwardOp, ReducerDef, ReducerGradient, Sorted, SparseFused >, caffe2::LengthsOpGetGradient< ForwardOp, ReducerDef, ReducerGradient, SparseFused, GradientNeedIndices >, caffe2::GetSeluGradient, caffe2::GetAddPaddingGradient, caffe2::GetRemovePaddingGradient, caffe2::GetSoftplusGradient, caffe2::GetSpaceToBatchGradient, caffe2::GetBatchToSpaceGradient, caffe2::FeedBlob::GetSqrtGradient, caffe2::GetSquareRootDivideGradient, caffe2::GetTopKGradient, caffe2::GetTransposeGradient, caffe2::GetUpsampleBilinearGradient, caffe2::GetNanCheckGradient, caffe2::GetZeroGradientOpGradient, caffe2::python::GetPythonGradient, caffe2::GetGroupSpatialSoftmaxGradient, caffe2::GetSampleAsGradient, caffe2::GetSelectSmoothL1LossGradient, caffe2::GetSigmoidCrossEntropyLossGradient, caffe2::GetSigmoidFocalLossGradient, caffe2::GetSmoothL1LossGradient, caffe2::GetSoftmaxFocalLossGradient, caffe2::SpatialNarrowAsGradient, and caffe2::GetUpsampleNearestGradient.

Definition at line 99 of file operator_gradient.h.

References CAFFE_NOT_IMPLEMENTED.

◆ GI()

string caffe2::GradientMakerBase::GI ( const int  i)
inlineprotected

Definition at line 117 of file operator_gradient.h.

References caffe2::CAFFE_ENFORCE(), and def_.

Referenced by caffe2::GetMatMulGradient::CopyArguments(), caffe2::GetFooGradient::GetGradientDefs(), caffe2::GetFCDecompGradient::GetGradientDefs(), caffe2::GetConvTransposeGradient::GetGradientDefs(), caffe2::GetGPUToCPUGradient::GetGradientDefs(), caffe2::GetElementwiseLinearGradient::GetGradientDefs(), caffe2::GetGRUUnitGradient::GetGradientDefs(), caffe2::GetIm2ColGradient::GetGradientDefs(), caffe2::GetCol2ImGradient::GetGradientDefs(), caffe2::GetIntegralImageGradient::GetGradientDefs(), caffe2::GetBernoulliJSDGradient::GetGradientDefs(), caffe2::GetAveragedLossGradient::GetGradientDefs(), caffe2::GetLSTMUnitGradient::GetGradientDefs(), caffe2::GetMatMulGradient::GetGradientDefs(), caffe2::GetNegateGradientGradient::GetGradientDefs(), caffe2::GetNormalizeGradient::GetGradientDefs(), caffe2::GetPackSegmentsGradient::GetGradientDefs(), caffe2::GetUnpackSegmentsGradient::GetGradientDefs(), caffe2::GetReduceFrontMaxGradient::GetGradientDefs(), caffe2::GetReduceBackMaxGradient::GetGradientDefs(), caffe2::GetReduceFrontMeanGradient::GetGradientDefs(), caffe2::GetReduceFrontSumGradient::GetGradientDefs(), caffe2::GetReduceBackSumGradient::GetGradientDefs(), caffe2::GetSumElementsGradient::GetGradientDefs(), caffe2::GetResizeNearestGradient::GetGradientDefs(), caffe2::GetReversePackedSegsGradient::GetGradientDefs(), caffe2::GetRecurrentGradient::GetGradientDefs(), caffe2::GetScaleGradient::GetGradientDefs(), caffe2::AbstractSortedSegmentRangeDef< T, SIndex, Context, ReducerDef >::GetGradient::GetGradientDefs(), caffe2::AbstractReduceFrontDef< T, Context, ReducerDef >::GetGradient::GetGradientDefs(), caffe2::AbstractReduceBackDef< T, Context, ReducerDef >::GetGradient::GetGradientDefs(), caffe2::SegmentOpGetGradient< ForwardOp, ReducerDef, ReducerGradient, Sorted, SparseFused >::GetGradientDefs(), caffe2::LengthsOpGetGradient< ForwardOp, ReducerDef, ReducerGradient, SparseFused, GradientNeedIndices >::GetGradientDefs(), caffe2::GetAddPaddingGradient::GetGradientDefs(), caffe2::GetRemovePaddingGradient::GetGradientDefs(), caffe2::GetSoftplusGradient::GetGradientDefs(), caffe2::GetSpaceToBatchGradient::GetGradientDefs(), caffe2::GetBatchToSpaceGradient::GetGradientDefs(), caffe2::GetTopKGradient::GetGradientDefs(), caffe2::GetUpsampleBilinearGradient::GetGradientDefs(), caffe2::GetNanCheckGradient::GetGradientDefs(), caffe2::GetZeroGradientOpGradient::GetGradientDefs(), caffe2::python::GetPythonGradient::GetGradientDefs(), caffe2::GetGroupSpatialSoftmaxGradient::GetGradientDefs(), caffe2::GetSampleAsGradient::GetGradientDefs(), caffe2::GetSelectSmoothL1LossGradient::GetGradientDefs(), caffe2::GetSigmoidCrossEntropyLossGradient::GetGradientDefs(), caffe2::GetSigmoidFocalLossGradient::GetGradientDefs(), caffe2::GetSmoothL1LossGradient::GetGradientDefs(), caffe2::GetSoftmaxFocalLossGradient::GetGradientDefs(), caffe2::SpatialNarrowAsGradient::GetGradientDefs(), caffe2::GetUpsampleNearestGradient::GetGradientDefs(), and caffe2::REGISTER_GRADIENT().

◆ GI_I()

string caffe2::GradientMakerBase::GI_I ( const int  i)
inlineprotected

Definition at line 126 of file operator_gradient.h.

References caffe2::CAFFE_ENFORCE(), and def_.

◆ GI_V()

string caffe2::GradientMakerBase::GI_V ( const int  i)
inlineprotected

Definition at line 135 of file operator_gradient.h.

References caffe2::CAFFE_ENFORCE(), and def_.

Referenced by caffe2::SegmentOpGetGradient< ForwardOp, ReducerDef, ReducerGradient, Sorted, SparseFused >::GetGradientDefs(), and caffe2::LengthsOpGetGradient< ForwardOp, ReducerDef, ReducerGradient, SparseFused, GradientNeedIndices >::GetGradientDefs().

◆ GO()

string caffe2::GradientMakerBase::GO ( const int  i)
inlineprotected

Definition at line 144 of file operator_gradient.h.

References caffe2::CAFFE_ENFORCE(), and def_.

Referenced by caffe2::GetPrependDimGradient::CopyArguments(), caffe2::GetFooGradient::GetGradientDefs(), caffe2::GetFCDecompGradient::GetGradientDefs(), caffe2::GetConvTransposeGradient::GetGradientDefs(), caffe2::GetSquaredL2DistanceGradient::GetGradientDefs(), caffe2::GetCosineSimilarityGradient::GetGradientDefs(), caffe2::GetElementwiseLinearGradient::GetGradientDefs(), caffe2::GetGRUUnitGradient::GetGradientDefs(), caffe2::GetIm2ColGradient::GetGradientDefs(), caffe2::GetCol2ImGradient::GetGradientDefs(), caffe2::GetIntegralImageGradient::GetGradientDefs(), caffe2::GetBernoulliJSDGradient::GetGradientDefs(), caffe2::GetAveragedLossGradient::GetGradientDefs(), caffe2::GetLSTMUnitGradient::GetGradientDefs(), caffe2::GetMatMulGradient::GetGradientDefs(), caffe2::GetNegateGradientGradient::GetGradientDefs(), caffe2::GetNormalizeGradient::GetGradientDefs(), caffe2::GetPackSegmentsGradient::GetGradientDefs(), caffe2::GetUnpackSegmentsGradient::GetGradientDefs(), caffe2::GetReduceFrontMaxGradient::GetGradientDefs(), caffe2::GetReduceBackMaxGradient::GetGradientDefs(), caffe2::GetReduceFrontMeanGradient::GetGradientDefs(), caffe2::GetReduceFrontSumGradient::GetGradientDefs(), caffe2::GetReduceBackSumGradient::GetGradientDefs(), caffe2::GetSumElementsGradient::GetGradientDefs(), caffe2::GetResizeNearestGradient::GetGradientDefs(), caffe2::GetReversePackedSegsGradient::GetGradientDefs(), caffe2::GetRecurrentGradient::GetGradientDefs(), caffe2::GetScaleGradient::GetGradientDefs(), caffe2::AbstractSortedSegmentRangeDef< T, SIndex, Context, ReducerDef >::GetGradient::GetGradientDefs(), caffe2::AbstractReduceFrontDef< T, Context, ReducerDef >::GetGradient::GetGradientDefs(), caffe2::AbstractReduceBackDef< T, Context, ReducerDef >::GetGradient::GetGradientDefs(), caffe2::SegmentOpGetGradient< ForwardOp, ReducerDef, ReducerGradient, Sorted, SparseFused >::GetGradientDefs(), caffe2::LengthsOpGetGradient< ForwardOp, ReducerDef, ReducerGradient, SparseFused, GradientNeedIndices >::GetGradientDefs(), caffe2::GetAddPaddingGradient::GetGradientDefs(), caffe2::GetRemovePaddingGradient::GetGradientDefs(), caffe2::GetSoftplusGradient::GetGradientDefs(), caffe2::GetSpaceToBatchGradient::GetGradientDefs(), caffe2::GetBatchToSpaceGradient::GetGradientDefs(), caffe2::GetTopKGradient::GetGradientDefs(), caffe2::GetUpsampleBilinearGradient::GetGradientDefs(), caffe2::GetNanCheckGradient::GetGradientDefs(), caffe2::python::GetPythonGradient::GetGradientDefs(), caffe2::GetGroupSpatialSoftmaxGradient::GetGradientDefs(), caffe2::GetSampleAsGradient::GetGradientDefs(), caffe2::GetSelectSmoothL1LossGradient::GetGradientDefs(), caffe2::GetSigmoidCrossEntropyLossGradient::GetGradientDefs(), caffe2::GetSigmoidFocalLossGradient::GetGradientDefs(), caffe2::GetSmoothL1LossGradient::GetGradientDefs(), caffe2::GetSoftmaxFocalLossGradient::GetGradientDefs(), caffe2::SpatialNarrowAsGradient::GetGradientDefs(), caffe2::GetUpsampleNearestGradient::GetGradientDefs(), and caffe2::REGISTER_GRADIENT().

◆ GO_I()

string caffe2::GradientMakerBase::GO_I ( const int  i)
inlineprotected

Definition at line 153 of file operator_gradient.h.

References caffe2::CAFFE_ENFORCE(), and def_.

◆ GO_V()

string caffe2::GradientMakerBase::GO_V ( const int  i)
inlineprotected

Definition at line 162 of file operator_gradient.h.

References caffe2::CAFFE_ENFORCE(), and def_.

Referenced by caffe2::REGISTER_GRADIENT().

◆ GradientName()

static string caffe2::GradientMakerBase::GradientName ( const string &  name)
inlinestaticprivate

Definition at line 221 of file operator_gradient.h.

References name.

◆ GradientNameToParam()

static string caffe2::GradientMakerBase::GradientNameToParam ( const string &  name)
inlinestaticprivate

Definition at line 229 of file operator_gradient.h.

References CHECK, and name.

◆ GradientSliceIndices()

static string caffe2::GradientMakerBase::GradientSliceIndices ( const string &  name)
inlinestaticprivate

Definition at line 234 of file operator_gradient.h.

References name.

◆ GradientSliceValues()

static string caffe2::GradientMakerBase::GradientSliceValues ( const string &  name)
inlinestaticprivate

Definition at line 238 of file operator_gradient.h.

References name.

◆ GradOut()

const GradientWrapper& caffe2::GradientMakerBase::GradOut ( int  i)
inlineprotected

Definition at line 171 of file operator_gradient.h.

◆ I()

string caffe2::GradientMakerBase::I ( const int  i)
inlineprotected

Definition at line 109 of file operator_gradient.h.

References caffe2::CAFFE_ENFORCE(), and def_.

Referenced by caffe2::GetFCDecompGradient::GetGradientDefs(), caffe2::GetConvTransposeGradient::GetGradientDefs(), caffe2::GetSquaredL2DistanceGradient::GetGradientDefs(), caffe2::GetCosineSimilarityGradient::GetGradientDefs(), caffe2::GetElementwiseLinearGradient::GetGradientDefs(), caffe2::GetGRUUnitGradient::GetGradientDefs(), caffe2::GetIm2ColGradient::GetGradientDefs(), caffe2::GetIntegralImageGradient::GetGradientDefs(), caffe2::GetBernoulliJSDGradient::GetGradientDefs(), caffe2::GetAveragedLossGradient::GetGradientDefs(), caffe2::GetLSTMUnitGradient::GetGradientDefs(), caffe2::GetMatMulGradient::GetGradientDefs(), caffe2::GetNormalizeGradient::GetGradientDefs(), caffe2::GetPackSegmentsGradient::GetGradientDefs(), caffe2::GetUnpackSegmentsGradient::GetGradientDefs(), caffe2::GetReduceFrontMaxGradient::GetGradientDefs(), caffe2::GetReduceBackMaxGradient::GetGradientDefs(), caffe2::GetReduceFrontMeanGradient::GetGradientDefs(), caffe2::GetReduceFrontSumGradient::GetGradientDefs(), caffe2::GetReduceBackSumGradient::GetGradientDefs(), caffe2::GetSumElementsGradient::GetGradientDefs(), caffe2::GetResizeNearestGradient::GetGradientDefs(), caffe2::GetReversePackedSegsGradient::GetGradientDefs(), caffe2::GetRecurrentGradient::GetGradientDefs(), caffe2::AbstractSortedSegmentRangeDef< T, SIndex, Context, ReducerDef >::GetGradient::GetGradientDefs(), caffe2::AbstractReduceFrontDef< T, Context, ReducerDef >::GetGradient::GetGradientDefs(), caffe2::AbstractReduceBackDef< T, Context, ReducerDef >::GetGradient::GetGradientDefs(), caffe2::SegmentOpGetGradient< ForwardOp, ReducerDef, ReducerGradient, Sorted, SparseFused >::GetGradientDefs(), caffe2::LengthsOpGetGradient< ForwardOp, ReducerDef, ReducerGradient, SparseFused, GradientNeedIndices >::GetGradientDefs(), caffe2::GetTopKGradient::GetGradientDefs(), caffe2::GetUpsampleBilinearGradient::GetGradientDefs(), caffe2::GetZeroGradientOpGradient::GetGradientDefs(), caffe2::python::GetPythonGradient::GetGradientDefs(), caffe2::GetSampleAsGradient::GetGradientDefs(), caffe2::GetSelectSmoothL1LossGradient::GetGradientDefs(), caffe2::GetSigmoidCrossEntropyLossGradient::GetGradientDefs(), caffe2::GetSigmoidFocalLossGradient::GetGradientDefs(), caffe2::GetSmoothL1LossGradient::GetGradientDefs(), caffe2::GetSoftmaxFocalLossGradient::GetGradientDefs(), caffe2::SpatialNarrowAsGradient::GetGradientDefs(), caffe2::GetUpsampleNearestGradient::GetGradientDefs(), and caffe2::REGISTER_GRADIENT().

◆ IsGradientBlob()

static bool caffe2::GradientMakerBase::IsGradientBlob ( const string &  name)
inlinestaticprivate

Definition at line 225 of file operator_gradient.h.

References name.

◆ MatchGradsToParams()

static CaffeMap<string, string> caffe2::GradientMakerBase::MatchGradsToParams ( const OperatorDef &  op)
inlinestatic

Returns map that returns the parameters that the gradients are for.

Definition at line 207 of file operator_gradient.h.

References matmul_dlmc_bench::m, and caffe2::out.

Referenced by caffe2::InferBlobShapesAndTypes().

◆ O()

string caffe2::GradientMakerBase::O ( const int  i)
inlineprotected

Definition at line 113 of file operator_gradient.h.

References caffe2::CAFFE_ENFORCE(), and def_.

Referenced by caffe2::GetGRUUnitGradient::GetGradientDefs(), caffe2::GetLSTMUnitGradient::GetGradientDefs(), caffe2::GetReduceFrontMaxGradient::GetGradientDefs(), caffe2::GetReduceBackMaxGradient::GetGradientDefs(), caffe2::GetRecurrentGradient::GetGradientDefs(), caffe2::AbstractSortedSegmentRangeDef< T, SIndex, Context, ReducerDef >::GetGradient::GetGradientDefs(), caffe2::AbstractReduceFrontDef< T, Context, ReducerDef >::GetGradient::GetGradientDefs(), caffe2::AbstractReduceBackDef< T, Context, ReducerDef >::GetGradient::GetGradientDefs(), caffe2::LengthsOpGetGradient< ForwardOp, ReducerDef, ReducerGradient, SparseFused, GradientNeedIndices >::GetGradientDefs(), caffe2::GetAddPaddingGradient::GetGradientDefs(), caffe2::GetRemovePaddingGradient::GetGradientDefs(), caffe2::GetSoftplusGradient::GetGradientDefs(), caffe2::GetTopKGradient::GetGradientDefs(), caffe2::python::GetPythonGradient::GetGradientDefs(), caffe2::GetGroupSpatialSoftmaxGradient::GetGradientDefs(), caffe2::GetSoftmaxFocalLossGradient::GetGradientDefs(), and caffe2::REGISTER_GRADIENT().

◆ SetDense()

void caffe2::GradientMakerBase::SetDense ( const int  i,
const string &  name 
)
inlineprotected

Definition at line 176 of file operator_gradient.h.

References caffe2::CAFFE_ENFORCE(), def_, and name.

◆ SetSparse()

void caffe2::GradientMakerBase::SetSparse ( const int  i,
const string &  indices,
const string &  values 
)
inlineprotected

Definition at line 184 of file operator_gradient.h.

References caffe2::CAFFE_ENFORCE(), def_, caffe2::indices, and caffe2::values.

Referenced by caffe2::SegmentOpGetGradient< ForwardOp, ReducerDef, ReducerGradient, Sorted, SparseFused >::GetGradientDefs(), and caffe2::LengthsOpGetGradient< ForwardOp, ReducerDef, ReducerGradient, SparseFused, GradientNeedIndices >::GetGradientDefs().

◆ SingleGradientDef()

template<class... Args>
static vector<OperatorDef> caffe2::GradientMakerBase::SingleGradientDef ( const Args &...  args)
inlinestaticprotected

a helper function to allow one to create one single operator def, which is usually the case for many simple operators.

Definition at line 199 of file operator_gradient.h.

References caffe2::args, and caffe2::CreateOperatorDef().

Referenced by caffe2::GetFCDecompGradient::GetGradientDefs(), caffe2::GetConvGradient::GetGradientDefs(), caffe2::GetConvTransposeGradient::GetGradientDefs(), caffe2::GetSquaredL2DistanceGradient::GetGradientDefs(), caffe2::GetElementwiseLinearGradient::GetGradientDefs(), caffe2::GetGRUUnitGradient::GetGradientDefs(), caffe2::GetIm2ColGradient::GetGradientDefs(), caffe2::GetCol2ImGradient::GetGradientDefs(), caffe2::GetIntegralImageGradient::GetGradientDefs(), caffe2::GetBernoulliJSDGradient::GetGradientDefs(), caffe2::GetAveragedLossGradient::GetGradientDefs(), caffe2::GetLSTMUnitGradient::GetGradientDefs(), caffe2::GetNegateGradientGradient::GetGradientDefs(), caffe2::GetNormalizeGradient::GetGradientDefs(), caffe2::GetPackSegmentsGradient::GetGradientDefs(), caffe2::GetUnpackSegmentsGradient::GetGradientDefs(), caffe2::GetPadImageGradient::GetGradientDefs(), caffe2::GetReduceFrontMaxGradient::GetGradientDefs(), caffe2::GetReduceBackMaxGradient::GetGradientDefs(), caffe2::GetReduceFrontMeanGradient::GetGradientDefs(), caffe2::GetReduceFrontSumGradient::GetGradientDefs(), caffe2::GetReduceBackSumGradient::GetGradientDefs(), caffe2::GetSumElementsGradient::GetGradientDefs(), caffe2::GetResizeNearestGradient::GetGradientDefs(), caffe2::GetReversePackedSegsGradient::GetGradientDefs(), caffe2::GetRecurrentGradient::GetGradientDefs(), caffe2::GetScaleGradient::GetGradientDefs(), caffe2::AbstractSortedSegmentRangeDef< T, SIndex, Context, ReducerDef >::GetGradient::GetGradientDefs(), caffe2::GetSeluGradient::GetGradientDefs(), caffe2::GetRemovePaddingGradient::GetGradientDefs(), caffe2::GetSoftplusGradient::GetGradientDefs(), caffe2::GetSpaceToBatchGradient::GetGradientDefs(), caffe2::GetBatchToSpaceGradient::GetGradientDefs(), caffe2::GetTopKGradient::GetGradientDefs(), caffe2::GetUpsampleBilinearGradient::GetGradientDefs(), caffe2::GetZeroGradientOpGradient::GetGradientDefs(), caffe2::python::GetPythonGradient::GetGradientDefs(), caffe2::GetGroupSpatialSoftmaxGradient::GetGradientDefs(), caffe2::GetSampleAsGradient::GetGradientDefs(), caffe2::GetSelectSmoothL1LossGradient::GetGradientDefs(), caffe2::GetSigmoidCrossEntropyLossGradient::GetGradientDefs(), caffe2::GetSigmoidFocalLossGradient::GetGradientDefs(), caffe2::GetSmoothL1LossGradient::GetGradientDefs(), caffe2::GetSoftmaxFocalLossGradient::GetGradientDefs(), caffe2::SpatialNarrowAsGradient::GetGradientDefs(), caffe2::GetUpsampleNearestGradient::GetGradientDefs(), and caffe2::REGISTER_GRADIENT().

◆ VerifyOp()

virtual void caffe2::GradientMakerBase::VerifyOp ( ) const
inlinevirtual

Definition at line 64 of file operator_gradient.h.

References caffe2::CAFFE_ENFORCE(), def_, caffe2::ProtoDebugString(), and caffe2::OpSchemaRegistry::Schema().

Member Data Documentation

◆ def_

const OperatorDef& caffe2::GradientMakerBase::def_
protected

Definition at line 245 of file operator_gradient.h.

Referenced by caffe2::ThrowInTheTowelIfGradientIsCalled::Get(), caffe2::GradientNotImplementedYet::Get(), caffe2::GetFCDecompGradient::GetGradientDefs(), caffe2::GetConvGradient::GetGradientDefs(), caffe2::GetConvTransposeGradient::GetGradientDefs(), caffe2::GetGRUUnitGradient::GetGradientDefs(), caffe2::GetLSTMUnitGradient::GetGradientDefs(), caffe2::GetMatMulGradient::GetGradientDefs(), caffe2::GetNegateGradientGradient::GetGradientDefs(), caffe2::GetNormalizeGradient::GetGradientDefs(), caffe2::GetReduceFrontMaxGradient::GetGradientDefs(), caffe2::GetReduceBackMaxGradient::GetGradientDefs(), caffe2::GetReduceFrontMeanGradient::GetGradientDefs(), caffe2::GetReduceFrontSumGradient::GetGradientDefs(), caffe2::GetReduceBackSumGradient::GetGradientDefs(), caffe2::GetResizeNearestGradient::GetGradientDefs(), caffe2::AbstractReduceFrontDef< T, Context, ReducerDef >::GetGradient::GetGradientDefs(), caffe2::AbstractReduceBackDef< T, Context, ReducerDef >::GetGradient::GetGradientDefs(), caffe2::GetSeluGradient::GetGradientDefs(), caffe2::GetUpsampleBilinearGradient::GetGradientDefs(), caffe2::python::GetPythonGradient::GetGradientDefs(), and caffe2::REGISTER_GRADIENT().

◆ g_input_

vector<GradientWrapper> caffe2::GradientMakerBase::g_input_
protected

Definition at line 247 of file operator_gradient.h.

◆ g_output_

const vector<GradientWrapper>& caffe2::GradientMakerBase::g_output_
protected

Definition at line 246 of file operator_gradient.h.

The documentation for this class was generated from the following file: