FLImaging 6.5.16.1
ComputationalGraphExp.h
1#pragma once
2
3#if _MSC_VER >= 1900 && defined(_M_X64)
4
5#include "ComputationalGraph.h"
6
7namespace FLImaging
8{
9 namespace AI
10 {
11 template <typename T>
12 class FL_EXPORT CComputationalGraphExp : public CComputationalGraph<T>
13 {
14 private:
15 CComputationalGraphExp();
16
17 protected:
18 CComputationalGraphExp(const CComputationalGraphExp<T>& cg);
19
20 public:
21 CComputationalGraphExp(const CComputationalBase<T>& cbOperand);
22 virtual ~CComputationalGraphExp();
23
24 virtual CTensor<T>& Forward() override;
25 virtual CTensor<T>* Backward() override;
26 virtual CComputationalBase<T>* Clone() const override;
27 virtual const CResult PrintNodeParamInfo() const override;
28
29 virtual const CResult GetBinaryData(Base::CFLData& fldBinary, bool bSuperClass = false, int32_t i32Version = -1, bool bDumpMode = false) const override;
30 virtual const CResult GetBinaryData(Base::CFLData* pFldBinary, bool bSuperClass = false, int32_t i32Version = -1, bool bDumpMode = false) const override;
31
32 virtual const CResult SetBinaryData(const Base::CFLData& fldBinary, int64_t* pI64Offset = nullptr) override;
33 virtual const CResult SetBinaryData(const Base::CFLData* pFldBinary, int64_t* pI64Offset = nullptr) override;
34
35 virtual const std::vector<int64_t>& GetEstimatedShape(bool bRecursive = true) const override;
36
37
38 DeclareGetClassType();
39 SupportToDuplicateObjectWithoutCreateNewObject(CComputationalGraphExp, *this);
40
41 public:
42 DeclareGetSignletonObject(CComputationalGraphExp);
43 };
44
45 #define CCGFExp(...) (*(new CComputationalGraphExp<float>(__VA_ARGS__)))
46 #define CCGDExp(...) (*(new CComputationalGraphExp<double>(__VA_ARGS__)))
47
48 #define CCGTExp(T, ...) (*(new CComputationalGraphExp<T>(__VA_ARGS__)))
49 }
50}
51
52#endif