Public Member Functions | |
template<typename T > | |
bool | Forward (const std::vector< int > &X_dims, const std::vector< int > &Y_dims, const T *X_data, T *Y_data, Context *context) const |
template<typename T > | |
bool | Backward (const std::vector< int > &dY_dims, const std::vector< int > &dX_dims, const T *dY_data, const T *X_data, const T *Y_data, T *dX_data, Context *context) const |
template<> | |
bool | Backward (const std::vector< int > &dY_dims, const std::vector< int > &dX_dims, const T *dY_data, const T *X_data, const T *Y_data, T *dX_data, CPUContext *) const |
Definition at line 140 of file reduce_ops.h.