diff options
Diffstat (limited to 'compiler/luci/import/include/luci/Import/Nodes/CircleUnique.h')
| -rw-r--r-- | compiler/luci/import/include/luci/Import/Nodes/CircleUnique.h | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/compiler/luci/import/include/luci/Import/Nodes/CircleUnique.h b/compiler/luci/import/include/luci/Import/Nodes/CircleUnique.h index ed5b5035d..40e75ec73 100644 --- a/compiler/luci/import/include/luci/Import/Nodes/CircleUnique.h +++ b/compiler/luci/import/include/luci/Import/Nodes/CircleUnique.h @@ -17,17 +17,19 @@ #ifndef __LUCI_IMPORT_OP_CIRCLE_UNIQUE_H__ #define __LUCI_IMPORT_OP_CIRCLE_UNIQUE_H__ -#include "luci/Import/GraphBuilderBase.h" +#include "luci/Import/GraphBuilderMultiOutput.h" namespace luci { -class CircleUniqueGraphBuilder : public GraphBuilderBase +class CircleUniqueGraphBuilder : public GraphBuilderMultiOutput { public: bool validate(const ValidateArgs &args) const final; - void build(const circle::OperatorT &op, GraphBuilderContext *context) const final; +private: + CircleNode *build_node(const BuildNodeArgs &) const final; + CircleNode *build_out(const BuildOutArgs &) const final; }; } // namespace luci |