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