FLImaging
7.8.25.3
Toggle main menu visibility
FLImagingAI
FLImagingAI
ComputationalGraphDegrees.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 CComputationalGraphDegrees :
public
CComputationalGraph<T>
13
{
14
private
:
15
CComputationalGraphDegrees();
16
17
protected
:
18
CComputationalGraphDegrees(
const
CComputationalGraphDegrees<T>& cg);
19
20
public
:
21
CComputationalGraphDegrees(
const
CComputationalBase<T>& cbOperand);
22
virtual
~CComputationalGraphDegrees();
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
DeclareGetClassType();
38
SupportToDuplicateObjectWithoutCreateNewObject(CComputationalGraphDegrees, *
this
);
39
40
public
:
41
DeclareGetSignletonObject(CComputationalGraphDegrees);
42
};
43
44
#define CCGFDegrees(...) (*(new CComputationalGraphDegrees<float>(__VA_ARGS__)))
45
#define CCGDDegrees(...) (*(new CComputationalGraphDegrees<double>(__VA_ARGS__)))
46
47
#define CCGTDegrees(T, ...) (*(new CComputationalGraphDegrees<T>(__VA_ARGS__)))
48
}
49
}
50
51
#endif
FLImaging::AI
Definition
AlgorithmAIBase.h:18
FLImaging
Generated on
for FLImaging by
1.17.0