Public Member Functions | |
template<typename T > | |
bool | Forward (const std::vector< int > &dY_dims, const std::vector< int > &Y_dims, const T *dY, const T *Y, T *dX, Context *context) const |
template<> | |
bool | Forward (const std::vector< int > &dY_dims, const std::vector< int > &, const T *dY, const T *Y, T *dX, CPUContext *) const |
Definition at line 21 of file rsqrt_op.h.