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 |
Definition at line 15 of file elementwise_add_op.h.