BackendEvaluatorForDenoisingDiffusion.h

#pragma once
 
#if _MSC_VER >= 1900 && defined(_M_X64)
 
#include "DefinitionsAI.h"
#include "BackendEvaluatorBase.h"
 
namespace FLImaging
{
    namespace AI
    {
        template <typename T>
        class CTensor;
 
        template <typename T>
        class CComputationalGraphPlaceholder;
 
        template <typename T>
        class CEvaluator;
 
        template <typename T>
        class CComputationalGraphUtilities;
 
        template <typename T>
        class FL_EXPORT CBackendEvaluatorForDenoisingDiffusion : public CBackendEvaluatorBase<T>
        {
        public:
            CBackendEvaluatorForDenoisingDiffusion();
            CBackendEvaluatorForDenoisingDiffusion(CEvaluator<T>* pEvaluator);
            CBackendEvaluatorForDenoisingDiffusion(CBackendEvaluatorForDenoisingDiffusion<T>& evaluator);
            CBackendEvaluatorForDenoisingDiffusion(CBackendEvaluatorForDenoisingDiffusion<T>* pEvaluator);
            virtual ~CBackendEvaluatorForDenoisingDiffusion();
 
            virtual const CResult Clear() override;
 
            virtual const CResult Assign(CBackendEvaluatorBase<T>* pEvaluator) override;
 
            virtual const CResult Assign(CBackendEvaluatorForDenoisingDiffusion<T>& evaluator);
            virtual const CResult Assign(CBackendEvaluatorForDenoisingDiffusion<T>* pEvaluator);
 
            virtual const CResult Evaluate(int32_t* pI32IterationFeedback = nullptr, bool* pBStopImmediatedly = nullptr, std::vector<bool>* pVctEvaluatedIndex = nullptr);
 
            virtual const CResult SetTotalSamplingStep(int32_t i32TotalSamplingStep = 1000);
            virtual int32_t GetTotalSamplingStep() const;
 
            virtual int32_t GetCurrentSamplingStep() const;
 
 
            DeclareGetClassType();
            SupportToDuplicateObject(CBackendEvaluatorForDenoisingDiffusion<T>);
        protected:
            int32_t m_i32TotalSamplingStep;
            int32_t m_i32CurrentSampllingStep;
 
        protected:
            template <typename T>
            friend class CComputationalGraphUtilities;
 
            template <typename T>
            friend class CEvaluator;
        };
    }
}
 
#endif