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 | 
Definition at line 321 of file reduce_ops.h.
 1.8.11