BackendSoftmax.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "BackendBase.h"
#include <vector>
 
namespace FLImaging
{
    namespace AI
    {
        #ifdef CUDNN_MODE
        template <typename T>
        class CCuda_ComputationalGraphSoftmax_Cudnn;
        #endif
 
        template <typename T>
        class FL_EXPORT CTensor;
 
        template <typename T>
        class FL_EXPORT CBackendSoftmax : public CBackendBase<T>
        {
        public:
            CBackendSoftmax();
            CBackendSoftmax(const CBackendSoftmax<T>& bl);
            virtual ~CBackendSoftmax();
 
            virtual const CResult SoftmaxCPU(const CTensor<T>* pTsrInput, CTensor<T>* pTsrResult, int64_t i64Axis);
            virtual const CResult SoftmaxCuda(const CTensor<T>* pTsrInput, CTensor<T>* pTsrResult, int64_t i64Axis);
 
            virtual const CResult SoftmaxBackwardCPU(const CTensor<T>* pTsrDy, const CTensor<T>* pTsrY, CTensor<T>* pTsrDx, int64_t i64Axis, bool bAddGradients = false);
            virtual const CResult SoftmaxBackwardCuda(const CTensor<T>* pTsrDy, const CTensor<T>* pTsrY, CTensor<T>* pTsrDx, int64_t i64Axis, bool bAddGradients = false);
        
            DeclareGetClassType();
            SupportToDuplicateObjectWithoutCreateNewObject(CBackendSoftmax<T>, *this);
 
        protected:
            #ifdef CUDNN_MODE
            CCuda_ComputationalGraphSoftmax_Cudnn<T>* m_cudnn;
            #endif
        
        };
    }
}
 
#endif