From 0e10395d2d225a3506480eaf322181a7e1edfd27 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=EB=B0=95=EC=84=B8=ED=9D=AC/=EB=8F=99=EC=9E=91=EC=A0=9C?= =?utf8?q?=EC=96=B4Lab=28SR=29/Principal=20Engineer/=EC=82=BC=EC=84=B1?= =?utf8?q?=EC=A0=84=EC=9E=90?= Date: Mon, 5 Mar 2018 11:31:04 +0900 Subject: [PATCH] [tf] Add operator dump in toco (#30) * [tf] Add operator dump in toco This will dump operators in toco tool * simplify for --- externals/tensorflow/tensorflow/contrib/lite/toco/toco_tooling.cc | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/externals/tensorflow/tensorflow/contrib/lite/toco/toco_tooling.cc b/externals/tensorflow/tensorflow/contrib/lite/toco/toco_tooling.cc index 1b836fb..db98587 100644 --- a/externals/tensorflow/tensorflow/contrib/lite/toco/toco_tooling.cc +++ b/externals/tensorflow/tensorflow/contrib/lite/toco/toco_tooling.cc @@ -302,6 +302,13 @@ void Transform(const TocoFlags& toco_flags, Model* model) { LOG(INFO) << "Estimated count of arithmetic ops: " << 1e-9 * ops_count << " billion (note that a multiply-add is counted as 2 ops)."; } + + std::cout << "=== Begin Operators in the model ===" << std::endl; + for (auto& op : model->operators) { + std::string name(OperatorTypeName(op->type)); + std::cout << name << std::endl; + } + std::cout << "=== End Operators in the model ===" << std::endl; } void Export(const TocoFlags& toco_flags, const Model& model, -- 2.7.4