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

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, const TIn *, const TIn *, const TOut *, TGrad *dA, TGrad *dB, Context *context) const
 

Detailed Description

template<class Context>
struct caffe2::SubFunctor< Context >

Definition at line 15 of file elementwise_sub_op.h.


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