caffe2::SubFunctor< Context > Struct Template Reference

Public Member Functions
template<typename TIn , typename TOut >
bool	Forward (const std::vector< int > &A_dims, const std::vector< int > &B_dims, const TIn *A, const TIn *B, TOut *C, Context *context) const
template<typename TGrad , typename TIn , typename TOut >
bool	Backward (const std::vector< int > &A_dims, const std::vector< int > &B_dims, const TGrad *dC, const TIn *, const TIn *, const TOut *, TGrad *dA, TGrad *dB, Context *context) const

Detailed Description

template<class Context>
struct caffe2::SubFunctor< Context >

Definition at line 15 of file elementwise_sub_op.h.

---

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

• caffe2/operators/elementwise_sub_op.h