-
Notifications
You must be signed in to change notification settings - Fork 11
Expand file tree
/
Copy pathNativeParam.cs
More file actions
102 lines (99 loc) · 2.4 KB
/
Copy pathNativeParam.cs
File metadata and controls
102 lines (99 loc) · 2.4 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
using Tensornet.Exceptions;
using Tensornet.Common;
namespace Tensornet.Native.Param{
internal struct MatmulParam{
}
internal struct DotParam{
}
internal struct InterelemParam{
internal InterElemOperationType operationType;
}
internal struct PermuteParam{
internal IntPtr dims;
}
internal struct TransposeParam{
internal int dimA;
internal int dimB;
}
internal struct ArgmxxParam{
internal int axis;
internal bool isMax;
}
internal struct RepeatParam{
internal int repeats;
internal int axis;
}
internal struct FlipParam{
internal IntPtr dims;
}
internal struct RotateParam{
internal int k;
internal int dimA;
internal int dimB;
internal RotateParam(int k, int dimA, int dimB){
this.k = k;
if(dimA > dimB){
(this.dimA, this.dimB) = (dimB, dimA);
}
else{
(this.dimA, this.dimB) = (dimA, dimB);
}
}
}
internal struct PadParam{
internal PadMode mode;
internal int size;
internal IntPtr widths;
internal IntPtr constants;
}
internal struct SortParam{
internal int axis;
internal SortOrder order;
}
internal struct OnehotParam{
internal int maxValue;
}
internal struct SumParam{
internal IntPtr dims;
}
internal struct MeanParam{
internal IntPtr dims;
}
internal struct MaxParam{
internal IntPtr dims;
}
internal struct MinParam{
internal IntPtr dims;
}
internal struct TypeConvertParam{
internal DType targetType;
}
internal struct NormalParam{
internal double mean;
internal double std;
}
internal struct UniformParam{
internal double minValue;
internal double maxValue;
}
internal struct EyeParam{
internal int k;
}
internal struct FillParam{
internal double value;
}
internal struct ArangeParam{
internal double start;
internal double stop;
internal double step;
}
internal struct LinspaceParam{
internal double start;
internal double stop;
internal int num;
internal bool isEndpoint;
}
internal struct ConcatParam{
internal int axis;
}
}