namespace middleend
{
-void SubTensorAnalyzer::visit(const graph::operation::Conv2D::Implicit::Node &)
+void SubTensorAnalyzer::visit(const graph::operation::Conv2DNode &)
{
// DO NOTHING
}
-void SubTensorAnalyzer::visit(const graph::operation::MaxPool2D::Implicit::Node &)
+void SubTensorAnalyzer::visit(const graph::operation::MaxPool2DNode &)
{
// DO NOTHING
}
-void SubTensorAnalyzer::visit(const graph::operation::AvgPool2D::Implicit::Node &)
+void SubTensorAnalyzer::visit(const graph::operation::AvgPool2DNode &)
{
// DO NOTHING
}
-void SubTensorAnalyzer::visit(const graph::operation::Concat::Node &node)
+void SubTensorAnalyzer::visit(const graph::operation::ConcatNode &node)
{
// If operator is concat (or other operators related with subsumption), fill subsumption info
// TODO: if one tensor is subset of many parents or model input
}
}
-void SubTensorAnalyzer::visit(const graph::operation::FullyConnected::Node &)
+void SubTensorAnalyzer::visit(const graph::operation::FullyConnectedNode &)
{
// DO NOTHING
}
-void SubTensorAnalyzer::visit(const graph::operation::Reshape::Node &)
+void SubTensorAnalyzer::visit(const graph::operation::ReshapeNode &)
{
// DO NOTHING
}
-void SubTensorAnalyzer::visit(const graph::operation::Softmax::Node &)
+void SubTensorAnalyzer::visit(const graph::operation::SoftmaxNode &)
{
// DO NOTHING
}
-void SubTensorAnalyzer::visit(const graph::operation::NOP::Node &)
+void SubTensorAnalyzer::visit(const graph::operation::NOPNode &)
{
// DO NOTHING
}
-void SubTensorAnalyzer::visit(const graph::operation::Permute::Node &)
+void SubTensorAnalyzer::visit(const graph::operation::PermuteNode &)
{
// DO NOTHING
}
SubTensorAnalyzer(neurun::graph::operand::Set &ctx) : _ctx{ctx} {}
public:
- virtual void visit(const graph::operation::Conv2D::Implicit::Node &) override;
- virtual void visit(const graph::operation::MaxPool2D::Implicit::Node &) override;
- virtual void visit(const graph::operation::AvgPool2D::Implicit::Node &) override;
- virtual void visit(const graph::operation::Concat::Node &) override;
- virtual void visit(const graph::operation::Reshape::Node &) override;
- virtual void visit(const graph::operation::FullyConnected::Node &) override;
- virtual void visit(const graph::operation::Softmax::Node &) override;
- virtual void visit(const graph::operation::NOP::Node &) override;
- virtual void visit(const graph::operation::Permute::Node &) override;
+ virtual void visit(const graph::operation::Conv2DNode &) override;
+ virtual void visit(const graph::operation::MaxPool2DNode &) override;
+ virtual void visit(const graph::operation::AvgPool2DNode &) override;
+ virtual void visit(const graph::operation::ConcatNode &) override;
+ virtual void visit(const graph::operation::ReshapeNode &) override;
+ virtual void visit(const graph::operation::FullyConnectedNode &) override;
+ virtual void visit(const graph::operation::SoftmaxNode &) override;
+ virtual void visit(const graph::operation::NOPNode &) override;
+ virtual void visit(const graph::operation::PermuteNode &) override;
virtual void visit(const graph::operation::AddNode &) override;
private: