assert(_ctx.at(output_index).shape() == _ctx.at(input_index).shape());
}
-void OperationValidator::visit(const model::operation::Conv2DNode &)
-{
- // DO NOTHING
-}
-
-void OperationValidator::visit(const model::operation::DepthwiseConv2DNode &)
-{
- // DO NOTHING
-}
-
-void OperationValidator::visit(const model::operation::MaxPool2DNode &)
-{
- // DO NOTHING
-}
-
-void OperationValidator::visit(const model::operation::AvgPool2DNode &)
-{
- // DO NOTHING
-}
-
-void OperationValidator::visit(const model::operation::ConcatNode &)
-{
- // DO NOTHING
-}
-
-void OperationValidator::visit(const model::operation::FullyConnectedNode &)
-{
- // DO NOTHING
-}
-
-void OperationValidator::visit(const model::operation::MulNode &)
-{
- // DO NOTHING
-}
-
-void OperationValidator::visit(const model::operation::ReshapeNode &)
-{
- // DO NOTHING
-}
-
void OperationValidator::visit(const model::operation::SoftmaxNode &node)
{
VERBOSE(Softmax) << "Configure SOFTMAX operation" << std::endl;
assert(_ctx.at(scale_index).shape().rank() == 0);
}
-void OperationValidator::visit(const model::operation::StridedSliceNode &)
-{
- // DO NOTHING
-}
-
-void OperationValidator::visit(const model::operation::TanhNode &)
-{
- // DO NOTHING
-}
-
-void OperationValidator::visit(const model::operation::LogisticNode &)
-{
- // DO NOTHING
-}
-
void OperationValidator::visit(const model::operation::PermuteNode &node)
{
VERBOSE(Permute) << "Configure Permute operation" << std::endl;
assert(_ctx.at(output_index).shape().rank() == _ctx.at(input_index).shape().rank());
}
-void OperationValidator::visit(const model::operation::AddNode &)
-{
- // DO NOTHING
-}
-
-void OperationValidator::visit(const model::operation::SubNode &)
-{
- // DO NOTHING
-}
-
} // namespace compiler
} // namespace neurun
OperationValidator(const neurun::model::operand::Set &ctx) : _ctx{ctx} {}
public:
-#define OP(InternalName, IsNnApi, NnApiName) \
- virtual void visit(const model::operation::InternalName &) override;
-#include "model/operation/Op.lst"
-#undef OP
+ virtual void visit(const model::operation::CastNode &node) override;
+ virtual void visit(const model::operation::SoftmaxNode &node) override;
+ virtual void visit(const model::operation::PermuteNode &node) override;
private:
const neurun::model::operand::Set &_ctx;