Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
caffe2::EigenPowFunctor Struct Reference

Public Member Functions

template<int b_is_scalar, typename T1 , typename T2 , typename R >
void Run (size_t n, const T1 *a, const T2 *b, T2 e, R *out, CPUContext *)
 
template<typename T1 , typename T2 , typename R >
void RunWithBroadcast (const T1 *a, const T2 *b, R *out, size_t pre, size_t n, CPUContext *)
 
template<typename T1 , typename T2 , typename R >
void RunWithBroadcast2 (const T1 *a, const T2 *b, R *out, size_t pre, size_t n, size_t post, CPUContext *)
 

Detailed Description

Definition at line 12 of file pow_op.cc.


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