Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
caffe2::MulFunctor< 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_data, const TIn *A_data, const TIn *B_data, const TOut *C_data, TGrad *dA_data, TGrad *dB_data, Context *context) const
 
template<>
bool Backward (const std::vector< int > &A_dims, const std::vector< int > &B_dims, const TGrad *dC, const TIn *A, const TIn *B, const TOut *, TGrad *dA, TGrad *dB, CPUContext *context) const
 

Detailed Description

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

Definition at line 12 of file elementwise_mul_op.h.


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