diff options
author | Chunseok Lee <chunseok.lee@samsung.com> | 2020-07-30 11:40:16 +0900 |
---|---|---|
committer | Chunseok Lee <chunseok.lee@samsung.com> | 2020-07-30 11:40:16 +0900 |
commit | 9e45ab56bd165609118989c0d1bec309c3754560 (patch) | |
tree | 4979e8674abc7d21a6471770c1355e0e6c0e8a3f /compiler/circledump/src/OpPrinter.cpp | |
parent | 05e0ec30a632339a8533082476f27bda31ccde16 (diff) | |
download | nnfw-9e45ab56bd165609118989c0d1bec309c3754560.tar.gz nnfw-9e45ab56bd165609118989c0d1bec309c3754560.tar.bz2 nnfw-9e45ab56bd165609118989c0d1bec309c3754560.zip |
patch for rebase master on release/1.7.0submit/tizen/20200731.060745submit/tizen/20200730.075407submit/tizen/20200730.023729accepted/tizen/unified/20200803.122223tizen_6_m1_backup
Change-Id: Id38b617d325ef7e854995a47f032bdf482a779b3
Diffstat (limited to 'compiler/circledump/src/OpPrinter.cpp')
-rw-r--r-- | compiler/circledump/src/OpPrinter.cpp | 15 |
1 files changed, 15 insertions, 0 deletions
diff --git a/compiler/circledump/src/OpPrinter.cpp b/compiler/circledump/src/OpPrinter.cpp index 2c0320396..5aa5d51c1 100644 --- a/compiler/circledump/src/OpPrinter.cpp +++ b/compiler/circledump/src/OpPrinter.cpp @@ -593,6 +593,20 @@ public: } }; +class UniquePrinter : public OpPrinter +{ +public: + void options(const circle::Operator *op, std::ostream &os) const override + { + if (auto *params = op->builtin_options_as_UniqueOptions()) + { + os << " "; + os << "idx_out_type(" << EnumNameTensorType(params->idx_out_type()) << ") "; + os << std::endl; + } + } +}; + class WhilePrinter : public OpPrinter { public: @@ -744,6 +758,7 @@ OpPrinterRegistry::OpPrinterRegistry() _op_map[circle::BuiltinOperator_SUM] = make_unique<ReducerPrinter>(); _op_map[circle::BuiltinOperator_TRANSPOSE_CONV] = make_unique<TransposeConvPrinter>(); // There is no Option for TOPK_V2 + _op_map[circle::BuiltinOperator_UNIQUE] = make_unique<UniquePrinter>(); _op_map[circle::BuiltinOperator_WHILE] = make_unique<WhilePrinter>(); _op_map[circle::BuiltinOperator_CUSTOM] = make_unique<CustomOpPrinter>(); |