FLImaging 6.5.16.1
ComputationalGraphSoftsign.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 CComputationalGraphSoftsign : public CComputationalGraph<T>
13 {
14 private:
15 CComputationalGraphSoftsign();
16
17 protected:
18 CComputationalGraphSoftsign(const CComputationalGraphSoftsign<T>& cg);
19
20 virtual const CResult Softsign(CTensor<T>* pTsrOperand, CTensor<T>* pTsrResult);
21
22 public:
23 CComputationalGraphSoftsign(const CComputationalBase<T>& cbOperand);
24 virtual ~CComputationalGraphSoftsign();
25
26 virtual CTensor<T>& Forward() override;
27 virtual CTensor<T>* Backward() override;
28 virtual CComputationalBase<T>* Clone() const override;
29 virtual const CResult PrintNodeParamInfo() const override;
30
31 virtual const CResult GetBinaryData(Base::CFLData& fldBinary, bool bSuperClass = false, int32_t i32Version = -1, bool bDumpMode = false) const override;
32 virtual const CResult GetBinaryData(Base::CFLData* pFldBinary, bool bSuperClass = false, int32_t i32Version = -1, bool bDumpMode = false) const override;
33
34 virtual const CResult SetBinaryData(const Base::CFLData& fldBinary, int64_t* pI64Offset = nullptr) override;
35 virtual const CResult SetBinaryData(const Base::CFLData* pFldBinary, int64_t* pI64Offset = nullptr) override;
36
37 virtual const std::vector<int64_t>& GetEstimatedShape(bool bRecursive = true) const override;
38
39
40 DeclareGetClassType();
41 SupportToDuplicateObjectWithoutCreateNewObject(CComputationalGraphSoftsign, *this);
42
43 public:
44 DeclareGetSignletonObject(CComputationalGraphSoftsign);
45 };
46
47 #define CCGFSoftsign(...) (*(new CComputationalGraphSoftsign<float>(__VA_ARGS__)))
48 #define CCGDSoftsign(...) (*(new CComputationalGraphSoftsign<double>(__VA_ARGS__)))
49
50 #define CCGTSoftsign(T, ...) (*(new CComputationalGraphSoftsign<T>(__VA_ARGS__)))
51 }
52}
53
54
55#endif