Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Data Fields
caffe2::BinaryFunctorWithDefaultCtor< Functor > Struct Template Reference

Public Member Functions

 BinaryFunctorWithDefaultCtor (OperatorBase &)
 
template<typename TIn , typename TOut , class Context >
bool Forward (const std::vector< int > &A_dims, const std::vector< int > &B_dims, const TIn *A_data, const TIn *B_data, TOut *C_data, Context *context) const
 
template<typename TGrad , typename TIn , typename TOut , class Context >
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
 

Data Fields

Functor functor {}
 

Detailed Description

template<class Functor>
struct caffe2::BinaryFunctorWithDefaultCtor< Functor >

Definition at line 326 of file elementwise_ops.h.


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