FLImaging 6.5.16.1
ComputationalGraphXavierRandom.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 CComputationalGraphXavierRandom : public CComputationalGraph<T>
13 {
14 private:
15 CComputationalGraphXavierRandom();
16
17 protected:
18 CComputationalGraphXavierRandom(const CComputationalGraphXavierRandom<T>& cg);
19
20 public:
21
22 CComputationalGraphXavierRandom(const CComputationalBase<T>& cbOperand1);
23 CComputationalGraphXavierRandom(const CTensor<T>& tsrOperand1);
24 virtual ~CComputationalGraphXavierRandom();
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(CComputationalGraphXavierRandom, *this);
42
43 protected:
44 public:
45 DeclareGetSignletonObject(CComputationalGraphXavierRandom);
46 };
47
48 #define CCGFXavierRandom(...) (*(new CComputationalGraphXavierRandom<float>(__VA_ARGS__)))
49 #define CCGDXavierRandom(...) (*(new CComputationalGraphXavierRandom<double>(__VA_ARGS__)))
50
51 #define CCGTXavierRandom(T, ...) (*(new CComputationalGraphXavierRandom<T>(__VA_ARGS__)))
52 }
53}
54
55#endif