diff options
Diffstat (limited to 'runtimes/neurun/backend/acl_cl/StageGenerator.h')
| -rw-r--r-- | runtimes/neurun/backend/acl_cl/StageGenerator.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/runtimes/neurun/backend/acl_cl/StageGenerator.h b/runtimes/neurun/backend/acl_cl/StageGenerator.h index 3a05104c9..5dad5c707 100644 --- a/runtimes/neurun/backend/acl_cl/StageGenerator.h +++ b/runtimes/neurun/backend/acl_cl/StageGenerator.h @@ -76,6 +76,9 @@ public: virtual void visit(const model::operation::SQRTNode &) override; virtual void visit(const model::operation::LogicalOrNode &) override; virtual void visit(const model::operation::LogicalNotNode &) override; + virtual void visit(const model::operation::EqualNode &) override; + virtual void visit(const model::operation::SquaredDifferenceNode &) override; + virtual void visit(const model::operation::TopKV2Node &) override; private: const neurun::model::operand::Set &_ctx; |