FunctionTable.h

Go to the documentation of this file.

#ifndef included_AMP_FunctionTable
#define included_AMP_FunctionTable
 
 
#include "AMP/utils/ArraySize.h"
 
#include <functional>
 
 
namespace AMP {
 
 
template<class TYPE>
class FunctionTable final
{
public:
    typedef TYPE value_type;
    template<class TYPE2>
    using cloneTo = FunctionTable<TYPE2>;
 
public:
    static void rand( size_t N, TYPE *x );
 
    template<typename LAMBDA>
    static TYPE reduce( LAMBDA op, size_t N, const TYPE *A, TYPE initialValue );
 
    template<typename LAMBDA>
    static TYPE reduce( LAMBDA op, size_t N, const TYPE *A, const TYPE *B, TYPE initialValue );
 
    template<typename LAMBDA>
    static void transform( LAMBDA fun, size_t N, const TYPE *x, TYPE *y );
 
    template<typename LAMBDA>
    static void transform( LAMBDA fun, size_t N, const TYPE *x, const TYPE *y, TYPE *z );
 
    static TYPE min( size_t N, const TYPE *x );
 
    static TYPE max( size_t N, const TYPE *x );
 
    static TYPE sum( size_t N, const TYPE *x );
 
    static void multiply( const ArraySize &sa,
                          const TYPE *a,
                          const ArraySize &sb,
                          const TYPE *b,
                          const ArraySize &sc,
                          TYPE *c );
 
    static void scale( size_t N, TYPE x, TYPE *y );
 
    static void px( size_t N, TYPE x, TYPE *y );
 
    static void px( size_t N, const TYPE *x, TYPE *y );
 
    static void mx( size_t N, TYPE x, TYPE *y );
 
    static void mx( size_t N, const TYPE *x, TYPE *y );
 
    static void axpy( TYPE alpha, size_t N, const TYPE *x, TYPE *y );
 
    static void axpby( TYPE alpha, size_t N, const TYPE *x, TYPE beta, TYPE *y );
 
    static void gemm( TYPE alpha,
                      const ArraySize &sa,
                      const TYPE *A,
                      const ArraySize &sb,
                      const TYPE *B,
                      TYPE beta,
                      const ArraySize &sc,
                      TYPE *C );
 
 
    static bool equals( size_t N, const TYPE *A, const TYPE *B, TYPE tol );
 
 
    /* Specialized Functions */
 
    static void transformReLU( size_t N, const TYPE *A, TYPE *B );
 
    static void transformAbs( size_t N, const TYPE *A, TYPE *B );
 
    static void transformTanh( size_t N, const TYPE *A, TYPE *B );
 
    static void transformHardTanh( size_t N, const TYPE *A, TYPE *B );
 
    static void transformSigmoid( size_t N, const TYPE *A, TYPE *B );
 
    static void transformSoftPlus( size_t N, const TYPE *A, TYPE *B );
 
private:
    FunctionTable() = delete;
};
 
 
} // namespace AMP
 
#endif