*/
class IVisitor {
public:
- virtual void visit(INode *node, ops::ConcatOp &op) = 0;
- virtual void visit(INode *node, ops::Conv2DOp &op) = 0;
- virtual void visit(INode *node, ops::DepthwiseConv2DOp &op) = 0;
- virtual void visit(INode *node, ops::SoftmaxOp &op) = 0;
- virtual void visit(INode *node, ops::PoolOp &op) = 0;
- virtual void visit(INode *node, ops::FullyConnectedOp &op) = 0;
- virtual void visit(INode *node, ops::CappedReluOp &op) = 0;
- virtual void visit(INode *node, ops::BiasAddOp &op) = 0;
- virtual void visit(INode *node, ops::VariableOp &op) = 0;
- virtual void visit(INode *node, ops::ReluOp &op) = 0;
- virtual void visit(INode *node, ops::ReshapeOp &op) = 0;
- virtual void visit(INode *node, ops::ScaleOp &op) = 0;
- virtual void visit(INode *node, ops::BatchNormOp &op) = 0;
- virtual void visit(INode *node, ops::DropoutOp &op) = 0;
- virtual void visit(INode *node, ops::TanhOp &op) = 0;
- virtual void visit(INode *node, ops::ElementwiseOp &op) = 0;
- virtual void visit(INode *node, ops::DeConv2DOp &op) = 0;
- virtual void visit(INode *node, ops::EluOp &op) = 0;
+ virtual void visit(INode* node, ops::ConcatOp& op) = 0;
+ virtual void visit(INode* node, ops::Conv2DOp& op) = 0;
+ virtual void visit(INode* node, ops::DepthwiseConv2DOp& op) = 0;
+ virtual void visit(INode* node, ops::SoftmaxOp& op) = 0;
+ virtual void visit(INode* node, ops::PoolOp& op) = 0;
+ virtual void visit(INode* node, ops::FullyConnectedOp& op) = 0;
+ virtual void visit(INode* node, ops::CappedReluOp& op) = 0;
+ virtual void visit(INode* node, ops::BiasAddOp& op) = 0;
+ virtual void visit(INode* node, ops::VariableOp& op) = 0;
+ virtual void visit(INode* node, ops::ReluOp& op) = 0;
+ virtual void visit(INode* node, ops::ReshapeOp& op) = 0;
+ virtual void visit(INode* node, ops::ScaleOp& op) = 0;
+ virtual void visit(INode* node, ops::BatchNormOp& op) = 0;
+ virtual void visit(INode* node, ops::DropoutOp& op) = 0;
+ virtual void visit(INode* node, ops::TanhOp& op) = 0;
+ virtual void visit(INode* node, ops::ElementwiseOp& op) = 0;
+ virtual void visit(INode* node, ops::DeConv2DOp& op) = 0;
+ virtual void visit(INode* node, ops::EluOp& op) = 0;
virtual ~IVisitor() = default;
};
*/
class Visitor: public IVisitor{
public:
- void visit(INode *node, ops::ConcatOp &op) override;
- void visit(INode *node, ops::Conv2DOp &op) override;
- void visit(INode *node, ops::DepthwiseConv2DOp &op) override;
- void visit(INode *node, ops::SoftmaxOp &op) override;
- void visit(INode *node, ops::PoolOp &op) override;
- void visit(INode *node, ops::FullyConnectedOp &op) override;
- void visit(INode *node, ops::CappedReluOp &op) override;
- void visit(INode *node, ops::BiasAddOp &op) override;
- void visit(INode *node, ops::VariableOp &op) override;
- void visit(INode *node, ops::ReluOp &op) override;
- void visit(INode *node, ops::ReshapeOp &op) override;
- void visit(INode *node, ops::ScaleOp &op) override;
- void visit(INode *node, ops::BatchNormOp &op) override;
- void visit(INode *node, ops::DropoutOp &op) override;
- void visit(INode *node, ops::TanhOp &op) override;
- void visit(INode *node, ops::ElementwiseOp &op) override;
- void visit(INode *node, ops::DeConv2DOp &op) override;
- void visit(INode *node, ops::EluOp &op) override;
+ void visit(INode* node, ops::ConcatOp& op) override;
+ void visit(INode* node, ops::Conv2DOp& op) override;
+ void visit(INode* node, ops::DepthwiseConv2DOp& op) override;
+ void visit(INode* node, ops::SoftmaxOp& op) override;
+ void visit(INode* node, ops::PoolOp& op) override;
+ void visit(INode* node, ops::FullyConnectedOp& op) override;
+ void visit(INode* node, ops::CappedReluOp& op) override;
+ void visit(INode* node, ops::BiasAddOp& op) override;
+ void visit(INode* node, ops::VariableOp& op) override;
+ void visit(INode* node, ops::ReluOp& op) override;
+ void visit(INode* node, ops::ReshapeOp& op) override;
+ void visit(INode* node, ops::ScaleOp& op) override;
+ void visit(INode* node, ops::BatchNormOp& op) override;
+ void visit(INode* node, ops::DropoutOp& op) override;
+ void visit(INode* node, ops::TanhOp& op) override;
+ void visit(INode* node, ops::ElementwiseOp& op) override;
+ void visit(INode* node, ops::DeConv2DOp& op) override;
+ void visit(INode* node, ops::EluOp& op) override;
~Visitor() override = default;