BackendAvgPool2D.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 CBackendAvgPool2D : public CBackendBase<T>
        {
        public:
            CBackendAvgPool2D();
            CBackendAvgPool2D(const CBackendAvgPool2D<T>& bf);
            virtual ~CBackendAvgPool2D();
 
            virtual const CResult SetParameters(int64_t _i64KernelX = 3, int64_t _i64KernelY = 3, int64_t _i64StrideX = 1, int64_t _i64StrideY = 1, int64_t _i64PaddingX = 0, int64_t _i64PaddingY = 0);
 
            virtual const CResult Forward(const CTensor<T>* pTsrOperand, CTensor<T>* pTsrResult);
            virtual const CResult Backward(const CTensor<T>* pTsrOperand, CTensor<T>* pTsrResult, std::vector<int64_t> vctOutputShapeAsc, bool bAddGradient, CTensor<T>* pTsrAddGradientTemp);
 
            DeclareGetClassType();
            SupportToDuplicateObjectWithoutCreateNewObject(CBackendAvgPool2D<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