caffe2::MulFunctor< 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_data, const TIn *A_data, const TIn *B_data, const TOut *C_data, TGrad *dA_data, TGrad *dB_data, Context *context) const
template<>
bool	Backward (const std::vector< int > &A_dims, const std::vector< int > &B_dims, const TGrad *dC, const TIn *A, const TIn *B, const TOut *, TGrad *dA, TGrad *dB, CPUContext *context) const

Detailed Description

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

Definition at line 12 of file elementwise_mul_op.h.

---

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

• caffe2/operators/elementwise_mul_op.h