26 lines
567 B
C++
26 lines
567 B
C++
#pragma once
|
|
|
|
#include <xmmintrin.h>
|
|
|
|
#include "./ActivationFunction.h"
|
|
|
|
namespace NeuralNetwork {
|
|
namespace ActivationFunction {
|
|
|
|
/**
|
|
* @author Tomas Cernik (Tom.Cernik@gmail.com)
|
|
* @brief Abstract class of activation function with support of SSE
|
|
*/
|
|
class StreamingActivationFunction : public ActivationFunction {
|
|
public:
|
|
|
|
virtual float operator()(const float &x) const=0;
|
|
|
|
/**
|
|
* @brief Returns value of four outputs
|
|
* @param x is float[4], in every array value can be stored
|
|
*/
|
|
virtual __m128 operator()(const __m128 &x) const=0;
|
|
};
|
|
}
|
|
} |