diff options
Diffstat (limited to 'compiler/luci/import/include/luci/Import/Nodes/CircleCustom.h')
-rw-r--r-- | compiler/luci/import/include/luci/Import/Nodes/CircleCustom.h | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/compiler/luci/import/include/luci/Import/Nodes/CircleCustom.h b/compiler/luci/import/include/luci/Import/Nodes/CircleCustom.h index 65745be4b..f0d7e303d 100644 --- a/compiler/luci/import/include/luci/Import/Nodes/CircleCustom.h +++ b/compiler/luci/import/include/luci/Import/Nodes/CircleCustom.h @@ -17,17 +17,19 @@ #ifndef __LUCI_IMPORT_OP_CIRCLE_CUSTOM_H__ #define __LUCI_IMPORT_OP_CIRCLE_CUSTOM_H__ -#include "luci/Import/GraphBuilder.h" +#include "luci/Import/GraphBuilderMultiOutput.h" namespace luci { -class CircleCustomGraphBuilder : public GraphBuilderBase +class CircleCustomGraphBuilder : 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 |