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 *, const TIn *B, const TOut *C, TGrad *dA, TGrad *dB, CPUContext *context) const |
Definition at line 12 of file elementwise_div_op.h.