1 | #ifndef DEF_NODE |
2 | #error The macro DEF_NODE was not declared. |
3 | #endif |
4 | DEF_NODE(Storage, Storage) |
5 | DEF_NODE(Constant, Constant) |
6 | DEF_NODE(Placeholder, Placeholder) |
7 | DEF_NODE(SaveNode, Save) |
8 | DEF_NODE(PadNode, Pad) |
9 | DEF_NODE(ConvolutionNode, Convolution) |
10 | DEF_NODE(ConvolutionGradNode, ConvolutionGrad) |
11 | DEF_NODE(ChannelwiseQuantizedConvolutionNode, ChannelwiseQuantizedConvolution) |
12 | DEF_NODE(ConvTransposeNode, ConvTranspose) |
13 | DEF_NODE(Convolution3DNode, Convolution3D) |
14 | DEF_NODE(Convolution3DGradNode, Convolution3DGrad) |
15 | DEF_NODE(MaxPoolNode, MaxPool) |
16 | DEF_NODE(MaxPoolGradNode, MaxPoolGrad) |
17 | DEF_NODE(ArgMaxNode, ArgMax) |
18 | DEF_NODE(ArgMinNode, ArgMin) |
19 | DEF_NODE(AvgPoolNode, AvgPool) |
20 | DEF_NODE(AvgPoolGradNode, AvgPoolGrad) |
21 | DEF_NODE(AdaptiveAvgPoolNode, AdaptiveAvgPool) |
22 | DEF_NODE(AdaptiveAvgPoolGradNode, AdaptiveAvgPoolGrad) |
23 | DEF_NODE(GemmNode, Gemm) |
24 | DEF_NODE(FullyConnectedNode, FullyConnected) |
25 | DEF_NODE(FullyConnectedGradNode, FullyConnectedGrad) |
26 | DEF_NODE(RowwiseQuantizedFullyConnectedNode, RowwiseQuantizedFullyConnected) |
27 | DEF_NODE(DynamicQuantizedFullyConnectedNode, DynamicQuantizedFullyConnected) |
28 | DEF_NODE(DynamicRowwiseQuantizedFullyConnectedNode, DynamicRowwiseQuantizedFullyConnected) |
29 | DEF_NODE(BatchNormalizationNode, BatchNormalization) |
30 | DEF_NODE(BatchNormalizationGradNode, BatchNormalizationGrad) |
31 | DEF_NODE(InstanceNormalizationNode, InstanceNormalization) |
32 | DEF_NODE(MeanVarNormalizationNode, MeanVarNormalization) |
33 | DEF_NODE(LocalResponseNormalizationNode, LocalResponseNormalization) |
34 | DEF_NODE(LocalResponseNormalizationGradNode, LocalResponseNormalizationGrad) |
35 | DEF_NODE(LayerNormalizationNode, LayerNormalization) |
36 | DEF_NODE(BatchBoxCoxNode, BatchBoxCox) |
37 | DEF_NODE(VectorNormNode, VectorNorm) |
38 | DEF_NODE(BucketizeNode, Bucketize) |
39 | DEF_NODE(SoftMaxNode, SoftMax) |
40 | DEF_NODE(SoftMaxGradNode, SoftMaxGrad) |
41 | DEF_NODE(LogSoftMaxNode, LogSoftMax) |
42 | DEF_NODE(LogSoftMaxGradNode, LogSoftMaxGrad) |
43 | DEF_NODE(CrossEntropyLossNode, CrossEntropyLoss) |
44 | DEF_NODE(CrossEntropyLossGradNode, CrossEntropyLossGrad) |
45 | DEF_NODE(RegressionNode, Regression) |
46 | DEF_NODE(RegressionGradNode, RegressionGrad) |
47 | DEF_NODE(SigmoidCrossEntropyWithLogitsNode, SigmoidCrossEntropyWithLogits) |
48 | DEF_NODE(AddNode, Add) |
49 | DEF_NODE(AddGradNode, AddGrad) |
50 | DEF_NODE(MulNode, Mul) |
51 | DEF_NODE(MulGradNode, MulGrad) |
52 | DEF_NODE(SubNode, Sub) |
53 | DEF_NODE(SubGradNode, SubGrad) |
54 | DEF_NODE(DivNode, Div) |
55 | DEF_NODE(DivGradNode, DivGrad) |
56 | DEF_NODE(FloorDivNode, FloorDiv) |
57 | DEF_NODE(FmodNode, Fmod) |
58 | DEF_NODE(MaxNode, Max) |
59 | DEF_NODE(MinNode, Min) |
60 | DEF_NODE(CmpEQNode, CmpEQ) |
61 | DEF_NODE(CmpNEQNode, CmpNEQ) |
62 | DEF_NODE(CmpLTNode, CmpLT) |
63 | DEF_NODE(CmpLTENode, CmpLTE) |
64 | DEF_NODE(PowNode, Pow) |
65 | DEF_NODE(AndNode, And) |
66 | DEF_NODE(BitwiseAndNode, BitwiseAnd) |
67 | DEF_NODE(OrNode, Or) |
68 | DEF_NODE(BitwiseOrNode, BitwiseOr) |
69 | DEF_NODE(XorNode, Xor) |
70 | DEF_NODE(BitwiseXorNode, BitwiseXor) |
71 | DEF_NODE(NotNode, Not) |
72 | DEF_NODE(BitwiseNotNode, BitwiseNot) |
73 | DEF_NODE(NegNode, Neg) |
74 | DEF_NODE(AbsNode, Abs) |
75 | DEF_NODE(FloorNode, Floor) |
76 | DEF_NODE(SignNode, Sign) |
77 | DEF_NODE(CeilNode, Ceil) |
78 | DEF_NODE(RoundNode, Round) |
79 | DEF_NODE(TruncateNode, Truncate) |
80 | DEF_NODE(SqrtNode, Sqrt) |
81 | DEF_NODE(RsqrtNode, Rsqrt) |
82 | DEF_NODE(ReciprocalNode, Reciprocal) |
83 | DEF_NODE(SinNode, Sin) |
84 | DEF_NODE(CosNode, Cos) |
85 | DEF_NODE(LogNode, Log) |
86 | DEF_NODE(AcosNode, Acos) |
87 | DEF_NODE(AsinNode, Asin) |
88 | DEF_NODE(AtanNode, Atan) |
89 | DEF_NODE(ErfNode, Erf) |
90 | DEF_NODE(ExpNode, Exp) |
91 | DEF_NODE(LogitNode, Logit) |
92 | DEF_NODE(NonZeroNode, NonZero) |
93 | DEF_NODE(SelectNode, Select) |
94 | DEF_NODE(BatchedAddNode, BatchedAdd) |
95 | DEF_NODE(BatchedMulNode, BatchedMul) |
96 | DEF_NODE(MatMulNode, MatMul) |
97 | DEF_NODE(BatchMatMulNode, BatchMatMul) |
98 | DEF_NODE(BatchedReduceAddNode, BatchedReduceAdd) |
99 | DEF_NODE(BatchedReduceSumSquareNode, BatchedReduceSumSquare) |
100 | DEF_NODE(BatchedReduceMeanNode, BatchedReduceMean) |
101 | DEF_NODE(BatchedReduceMinNode, BatchedReduceMin) |
102 | DEF_NODE(BatchedReduceMaxNode, BatchedReduceMax) |
103 | DEF_NODE(BatchedReduceProdNode, BatchedReduceProd) |
104 | DEF_NODE(ChannelShuffleNode, ChannelShuffle) |
105 | DEF_NODE(CumSumNode, CumSum) |
106 | DEF_NODE(LengthsSumNode, LengthsSum) |
107 | DEF_NODE(SparseLengthsSumNode, SparseLengthsSum) |
108 | DEF_NODE(SparseLengthsSumGradNode, SparseLengthsSumGrad) |
109 | DEF_NODE(SparseLengthsWeightedSumNode, SparseLengthsWeightedSum) |
110 | DEF_NODE(SparseLengthsWeightedSumGradNode, SparseLengthsWeightedSumGrad) |
111 | DEF_NODE(EmbeddingNode, Embedding) |
112 | DEF_NODE(EmbeddingBagNode, EmbeddingBag) |
113 | DEF_NODE(EmbeddingBagByteRowwiseOffsetsNode, EmbeddingBagByteRowwiseOffsets) |
114 | DEF_NODE(RowwiseQuantizedSparseLengthsWeightedSumNode, RowwiseQuantizedSparseLengthsWeightedSum) |
115 | DEF_NODE(FusedRowwiseQuantizedSparseLengthsWeightedSumNode, FusedRowwiseQuantizedSparseLengthsWeightedSum) |
116 | DEF_NODE(FusedRowwiseQuantizedSparseLengthsSumNode, FusedRowwiseQuantizedSparseLengthsSum) |
117 | DEF_NODE(LengthsToRangesNode, LengthsToRanges) |
118 | DEF_NODE(LengthsRangeFillNode, LengthsRangeFill) |
119 | DEF_NODE(BatchSparseToDenseNode, BatchSparseToDense) |
120 | DEF_NODE(FillExamplesWithIndicatorNode, FillExamplesWithIndicator) |
121 | DEF_NODE(SparseToDenseMaskNode, SparseToDenseMask) |
122 | DEF_NODE(IsNaNNode, IsNaN) |
123 | DEF_NODE(ReplaceNaNNode, ReplaceNaN) |
124 | DEF_NODE(ModuloNode, Modulo) |
125 | DEF_NODE(BatchedPairwiseDotProductNode, BatchedPairwiseDotProduct) |
126 | DEF_NODE(BatchedPairwiseDotProductGradNode, BatchedPairwiseDotProductGrad) |
127 | DEF_NODE(BatchedUnaryEmbeddingsBagsNode, BatchedUnaryEmbeddingsBags) |
128 | DEF_NODE(IntNBitSplitEmbeddingBagsNode, IntNBitSplitEmbeddingBags) |
129 | DEF_NODE(IntNBitSplitEmbeddingWeightedBagsNode, IntNBitSplitEmbeddingWeightedBags) |
130 | DEF_NODE(GaussianFillNode, GaussianFill) |
131 | DEF_NODE(ReluNode, Relu) |
132 | DEF_NODE(ReluGradNode, ReluGrad) |
133 | DEF_NODE(HardSwishNode, HardSwish) |
134 | DEF_NODE(GeluNode, Gelu) |
135 | DEF_NODE(ClipNode, Clip) |
136 | DEF_NODE(PReluNode, PRelu) |
137 | DEF_NODE(SigmoidNode, Sigmoid) |
138 | DEF_NODE(SigmoidGradNode, SigmoidGrad) |
139 | DEF_NODE(SwishNode, Swish) |
140 | DEF_NODE(TanhNode, Tanh) |
141 | DEF_NODE(TanhGradNode, TanhGrad) |
142 | DEF_NODE(LeakyReluNode, LeakyRelu) |
143 | DEF_NODE(SoftPlusNode, SoftPlus) |
144 | DEF_NODE(ReshapeNode, Reshape) |
145 | DEF_NODE(TransposeNode, Transpose) |
146 | DEF_NODE(ConcatNode, Concat) |
147 | DEF_NODE(SliceNode, Slice) |
148 | DEF_NODE(InsertTensorNode, InsertTensor) |
149 | DEF_NODE(GatherNode, Gather) |
150 | DEF_NODE(GatherNDNode, GatherND) |
151 | DEF_NODE(GatherElementsNode, GatherElements) |
152 | DEF_NODE(GatherRangesNode, GatherRanges) |
153 | DEF_NODE(ScatterDataNode, ScatterData) |
154 | DEF_NODE(TileNode, Tile) |
155 | DEF_NODE(BatchOneHotNode, BatchOneHot) |
156 | DEF_NODE(SpaceToDepthNode, SpaceToDepth) |
157 | DEF_NODE(ResizeNearestNode, ResizeNearest) |
158 | DEF_NODE(ResizeBilinearNode, ResizeBilinear) |
159 | DEF_NODE(BroadcastNode, Broadcast) |
160 | DEF_NODE(SparseLabelSplitNode, SparseLabelSplit) |
161 | DEF_NODE(FlipNode, Flip) |
162 | DEF_NODE(SplatNode, Splat) |
163 | DEF_NODE(TouchNode, Touch) |
164 | DEF_NODE(SGDNode, SGD) |
165 | DEF_NODE(TraceEventNode, TraceEvent) |
166 | DEF_NODE(QuantizationProfileNode, QuantizationProfile) |
167 | DEF_NODE(IntLookupTableNode, IntLookupTable) |
168 | DEF_NODE(QuantizeNode, Quantize) |
169 | DEF_NODE(DequantizeNode, Dequantize) |
170 | DEF_NODE(RescaleQuantizedNode, RescaleQuantized) |
171 | DEF_NODE(TopKNode, TopK) |
172 | DEF_NODE(LSTMUnitNode, LSTMUnit) |
173 | DEF_NODE(ConvertToNode, ConvertTo) |
174 | DEF_NODE(ExternalFunctionCallNode, ExternalFunctionCall) |
175 | DEF_NODE(AudioSpectrogramNode, AudioSpectrogram) |
176 | DEF_NODE(MFCCNode, MFCC) |
177 | DEF_NODE(NonMaxSuppressionNode, NonMaxSuppression) |
178 | DEF_NODE(TFLiteDetectionPostProcessNode, TFLiteDetectionPostProcess) |
179 | DEF_NODE(ROIAlignNode, ROIAlign) |
180 | DEF_NODE(BBoxTransformNode, BBoxTransform) |
181 | DEF_NODE(CollectRpnProposalsNode, CollectRpnProposals) |
182 | DEF_NODE(LookupTableNode, LookupTable) |
183 | DEF_NODE(CPUMaxSplatNode, CPUMaxSplat) |
184 | DEF_NODE(CPUConvDKKC8Node, CPUConvDKKC8) |
185 | #undef DEF_NODE |