diff options
Diffstat (limited to 'compiler/locomotiv/src/Node.lst')
-rw-r--r-- | compiler/locomotiv/src/Node.lst | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/compiler/locomotiv/src/Node.lst b/compiler/locomotiv/src/Node.lst new file mode 100644 index 000000000..be3b10520 --- /dev/null +++ b/compiler/locomotiv/src/Node.lst @@ -0,0 +1,40 @@ +#ifndef NODE +#error Define NODE first +#endif // NODE + +// NODE(Name) : alphabetic order please + +NODE(AvgPool2D) +NODE(BiasAdd<loco::Domain::Feature>) +NODE(BiasAdd<loco::Domain::Tensor>) +NODE(BiasEncode) +NODE(ConstGen) +NODE(Conv2D) +NODE(DepthwiseConv2D) +NODE(DepthwiseFilterEncode) +NODE(EltwiseAdd) +NODE(EltwiseDiv) +NODE(EltwiseMax) +NODE(EltwiseMul) +NODE(EltwiseSqrt) +NODE(EltwiseSub) +NODE(FeatureDecode) +NODE(FeatureEncode) +NODE(FilterEncode) +NODE(Forward) +NODE(MatrixDecode) +NODE(MatrixEncode) +NODE(MatMul) +NODE(MaxPool2D) +NODE(Pull) +NODE(Push) +NODE(ReLU) +NODE(ReLU6) +NODE(Reshape<loco::ReshapeType::Fixed>) +NODE(Tanh) +NODE(TensorBroadcast) +NODE(TensorConcat) +NODE(TensorConstantPad) +NODE(TensorReduce) +NODE(TensorSoftmax) +NODE(TransposedConv2D) |