FLImaging 6.5.16.1
ComputationalGraphLog10.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 CComputationalGraphLog10 : public CComputationalGraph<T>
13 {
14 private:
15 CComputationalGraphLog10();
16
17 protected:
18 CComputationalGraphLog10(const CComputationalGraphLog10<T>& cg);
19
20 public:
21 CComputationalGraphLog10(const CComputationalBase<T>& cbOperand);
22 virtual ~CComputationalGraphLog10();
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 virtual int64_t GetRequiredDedicatedMemory(bool bTraining = false, bool bRecursively = true, int64_t i64BatchSize = 1) const override;
38
39
40 DeclareGetClassType();
41 SupportToDuplicateObjectWithoutCreateNewObject(CComputationalGraphLog10, *this);
42
43 public:
44 DeclareGetSignletonObject(CComputationalGraphLog10);
45 };
46
47 #define CCGFLog10(...) (*(new CComputationalGraphLog10<float>(__VA_ARGS__)))
48 #define CCGDLog10(...) (*(new CComputationalGraphLog10<double>(__VA_ARGS__)))
49
50 #define CCGTLog10(T, ...) (*(new CComputationalGraphLog10<T>(__VA_ARGS__)))
51 }
52}
53
54#endif