Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
caffe2::SumReducer< Context > Class Template Reference

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 *, const T *, T *dX_data, Context *context) const
 

Detailed Description

template<class Context>
class caffe2::SumReducer< Context >

Definition at line 202 of file reduce_ops.h.


The documentation for this class was generated from the following file: