activations.hpp

#pragma once
 
#include "../core/common.hpp"
 
#ifdef __CUDACC__
#include <cuda_runtime.h>
#endif
 
#define ACTIVATION(name, body)                      \
    template <typename T>                           \
    class name                                      \
    {                                               \
    public:                                         \
        cudaFn T operator()(const T x) const body   \
    };
 // End of custom_activations group
 
namespace CellularAutomata
{
    namespace Activations
    {
        template<typename T>
        class normal
        {
        public:
            cudaFn T operator()(const T x) const;
        };
 
        template<typename T>
        class life
        {
        public:
            cudaFn T operator()(const T x) const;
        };
 
        template<typename T>
        class sigmoid
        {
        public:
            cudaFn T operator()(const T x) const;
        };
 
        template<typename T>
        class tanh
        {
        public:
            cudaFn T operator()(const T x) const;
        };
 
        template<typename T>
        class sin
        {
        public:
            cudaFn T operator()(const T x) const;
        };
 
        template<typename T>
        class cos
        {
        public:
            cudaFn T operator()(const T x) const;
        };
 
        template<typename T>
        class tan
        {
        public:
            cudaFn T operator()(const T x) const;
        };
    }
}
 
#include "activations.inl"