BackendReLU.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "BackendBase.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class FL_EXPORT CTensor;
 
        template <typename T>
        class FL_EXPORT CBackendReLU : public CBackendBase<T>
        {
        public:
            CBackendReLU();
            CBackendReLU(const CBackendReLU<T>& bf);
            virtual ~CBackendReLU();
 
            virtual const CResult Forward(const CTensor<T>* pTsrOperand, CTensor<T>* pTsrResult, bool bInplace, bool bRetained, int8_t** ppReserveBuffer, size_t stReserveSize, int32_t i32DeviceIndex);
            virtual const CResult Backward(const CTensor<T>* pTsrOperand, CTensor<T>* pTsrResult, CTensor<T>* pTsrX, bool bAddGradient, bool bInplace, bool bRetained, int8_t* pReserveBuffer);
 
            DeclareGetClassType();
            SupportToDuplicateObjectWithoutCreateNewObject(CBackendReLU<T>, *this);
 
        private:
            int64_t m_i64KernelX;
            int64_t m_i64KernelY;
            int64_t m_i64StrideX;
            int64_t m_i64StrideY;
            int64_t m_i64PaddingX;
            int64_t m_i64PaddingY;
 
        };
    }
}
 
#endif