FLImaging
7.8.25.3
Toggle main menu visibility
FLImagingAI
FLImagingAI
ComputationalGraphExp.h
1
#pragma once
2
3
#if _MSC_VER >= 1900 && defined(_M_X64)
4
5
#include "ComputationalGraph.h"
6
7
namespace
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
FLImaging::AI
Definition
AlgorithmAIBase.h:18
FLImaging
Generated on
for FLImaging by
1.17.0