#ifndef __NEURUN_MODEL_OP_CODE_H__
#define __NEURUN_MODEL_OP_CODE_H__
-#include <unordered_map>
+#include <functional>
+#include <stdint.h>
namespace neurun
{
COUNT
};
+const char *toString(OpCode opcode);
+
} // namespace model
} // namespace neurun
public:
virtual void accept(OperationVisitor &v) const = 0;
- virtual std::string name() const = 0;
+ virtual std::string name() const { return std::string{toString(opcode())}; }
virtual OpCode opcode() const = 0;
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Abs"; }
OpCode opcode() const final { return OpCode::Abs; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Add"; }
OpCode opcode() const final { return OpCode::Add; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "ArgMax"; }
OpCode opcode() const final { return OpCode::ArgMax; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "AvgPool2D"; }
OpCode opcode() const final { return OpCode::AvgPool2D; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "BatchToSpaceND"; }
OpCode opcode() const final { return OpCode::BatchToSpaceND; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Cast"; }
OpCode opcode() const final { return OpCode::Cast; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Comparison"; }
OpCode opcode() const final { return OpCode::Comparison; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Concat"; }
OpCode opcode() const final { return OpCode::Concat; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Conv2D"; }
OpCode opcode() const final { return OpCode::Conv2D; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "DepthToSpace"; }
OpCode opcode() const final { return OpCode::DepthToSpace; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "DepthwiseConv2D"; }
OpCode opcode() const final { return OpCode::DepthwiseConv2D; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Dequantize"; }
OpCode opcode() const final { return OpCode::Dequantize; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Div"; }
OpCode opcode() const final { return OpCode::Div; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "EmbeddingLookup"; }
OpCode opcode() const final { return OpCode::EmbeddingLookup; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Exp"; }
OpCode opcode() const final { return OpCode::Exp; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Floor"; }
OpCode opcode() const final { return OpCode::Floor; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "FullyConnected"; }
OpCode opcode() const final { return OpCode::FullyConnected; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Gather"; }
OpCode opcode() const final { return OpCode::Gather; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "HashTableLookup"; }
OpCode opcode() const final { return OpCode::HashtableLookup; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "InstanceNorm"; }
OpCode opcode() const final { return OpCode::InstanceNorm; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "L2Normalization"; }
OpCode opcode() const final { return OpCode::L2Normalization; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "L2Pool2D"; }
OpCode opcode() const final { return OpCode::L2Pool2D; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "LSTM"; }
OpCode opcode() const final { return OpCode::LSTM; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "LocalResponseNormalization"; }
OpCode opcode() const final { return OpCode::LocalResponseNormalization; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "LogicalAnd"; }
OpCode opcode() const final { return OpCode::LogicalAnd; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "LogicalNot"; }
OpCode opcode() const final { return OpCode::LogicalNot; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "LogicalOr"; }
OpCode opcode() const final { return OpCode::LogicalOr; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Logistic"; }
OpCode opcode() const final { return OpCode::Logistic; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "MaxPool2D"; }
OpCode opcode() const final { return OpCode::MaxPool2D; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Mean"; }
OpCode opcode() const final { return OpCode::Mean; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Mul"; }
OpCode opcode() const final { return OpCode::Mul; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Neg"; }
OpCode opcode() const final { return OpCode::Neg; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "PReLU"; }
OpCode opcode() const final { return OpCode::PReLU; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Pack"; }
OpCode opcode() const final { return OpCode::Pack; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Pad"; }
OpCode opcode() const final { return OpCode::Pad; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Permute"; }
OpCode opcode() const final { return OpCode::Permute; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "RNN"; }
OpCode opcode() const final { return OpCode::RNN; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "RSQRT"; }
OpCode opcode() const final { return OpCode::RSQRT; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "ReLU"; }
OpCode opcode() const final { return OpCode::ReLU; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "ReLU1"; }
OpCode opcode() const final { return OpCode::ReLU1; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "ReLU6"; }
OpCode opcode() const final { return OpCode::ReLU6; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "ReduceMax"; }
OpCode opcode() const final { return OpCode::ReduceMax; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "ReduceMin"; }
OpCode opcode() const final { return OpCode::ReduceMin; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "ReduceSum"; }
OpCode opcode() const final { return OpCode::ReduceSum; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Reshape"; }
OpCode opcode() const final { return OpCode::Reshape; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "ResizeBilinear"; }
OpCode opcode() const final { return OpCode::ResizeBilinear; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "SQRT"; }
OpCode opcode() const final { return OpCode::SQRT; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "SoftMax"; }
OpCode opcode() const final { return OpCode::Softmax; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "SpaceToBatchND"; }
OpCode opcode() const final { return OpCode::SpaceToBatchND; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "SpaceToDepth"; }
OpCode opcode() const final { return OpCode::SpaceToDepth; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Split"; }
OpCode opcode() const final { return OpCode::Split; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "SquaredDifference"; }
OpCode opcode() const final { return OpCode::SquaredDifference; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Squeeze"; }
OpCode opcode() const final { return OpCode::Squeeze; }
const Param ¶m() const { return _param; }
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "StridedSlice"; }
OpCode opcode() const final { return OpCode::StridedSlice; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Sub"; }
OpCode opcode() const final { return OpCode::Sub; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Tanh"; }
OpCode opcode() const final { return OpCode::Tanh; }
};
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "TopKV2"; }
OpCode opcode() const final { return OpCode::TopKV2; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Transpose"; }
OpCode opcode() const final { return OpCode::Transpose; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "TransposeConv"; }
OpCode opcode() const final { return OpCode::TransposeConv; }
public:
public:
void accept(OperationVisitor &v) const override;
- std::string name() const override { return "Unpack"; }
OpCode opcode() const final { return OpCode::Unpack; }
public:
--- /dev/null
+/*
+ * Copyright (c) 2019 Samsung Electronics Co., Ltd. All Rights Reserved
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include "model/OpCode.h"
+
+#include <unordered_map>
+
+namespace neurun
+{
+namespace model
+{
+
+const char *toString(OpCode opcode)
+{
+ static const std::unordered_map<OpCode, const char *> map{{OpCode::Invalid, "Invalid"},
+#define OP(Name) {OpCode::Name, #Name},
+#include "model/Operations.lst"
+#undef OP
+ {OpCode::COUNT, "COUNT"}};
+ return map.at(opcode);
+}
+
+} // namespace model
+} // namespace neurun
public:
void accept(neurun::model::OperationVisitor &) const override {}
- std::string name() const override { return "SimpleMock"; }
neurun::model::OpCode opcode() const final { return neurun::model::OpCode::Invalid; }
};
projects / platform / core / ml / nnfw.git / commitdiff