BackendAdaptiveAvgPool2D.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "BackendBase.h"
#include "Parameters.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class FL_EXPORT CTensor;
 
        template <typename T>
        class FL_EXPORT CBackendAdaptiveAvgPool2D : public CBackendBase<T>
        {
        public:
            CBackendAdaptiveAvgPool2D();
            CBackendAdaptiveAvgPool2D(const CBackendAdaptiveAvgPool2D<T>& ba);
            virtual ~CBackendAdaptiveAvgPool2D();
 
            virtual const CResult SetAdaptivePoolParams(const SAdaptivePoolParameters& adaptivePoolParams);
            virtual SAdaptivePoolParameters GetAdaptivePoolParams();
 
            virtual const CResult Forward(const CTensor<T>* pTsrOperand, CTensor<T>* pTsrResult);
            virtual const CResult Backward(const CTensor<T>* pTsrOperand, CTensor<T>* pTsrResult, bool bAdd = false, CTensor<T>* pTsrDerivativeTemp = nullptr);
 
            DeclareGetClassType();
            SupportToDuplicateObjectWithoutCreateNewObject(CBackendAdaptiveAvgPool2D<T>, *this);
 
        protected:
            SAdaptivePoolParameters m_adaptivePoolParams;
        };
    }
}
 
#endif