tflite: List also shows input/output of subgraph
BUG=b:331900692
TEST=bazel run --config=host_clang common:decompose_model -- list \
--input=<input>
Change-Id: I9b1e6a6d3a31d1f9e139f513211f710387b8adf8
Reviewed-on: https://chromium-review.googlesource.com/c/chromiumos/platform/tflite/+/5724909
Commit-Queue: Ching-Kang Yen <chingkang@chromium.org>
Reviewed-by: Shik Chen <shik@chromium.org>
Tested-by: Ching-Kang Yen <chingkang@chromium.org>
diff --git a/common/decompose_model.cc b/common/decompose_model.cc
index cde2413..5db47aa 100644
--- a/common/decompose_model.cc
+++ b/common/decompose_model.cc
@@ -240,6 +240,12 @@
std::string output_dump = DumpTensorsInfo(op->outputs, subgraph->tensors);
table.push_back({std::to_string(i), op_name, input_dump, output_dump});
}
+ std::cout << "Subgraph inputs: "
+ << DumpTensorsInfo(subgraph->inputs, subgraph->tensors)
+ << std::endl;
+ std::cout << "Subgraph outputs: "
+ << DumpTensorsInfo(subgraph->outputs, subgraph->tensors)
+ << std::endl;
std::cout << TableStringify(table) << std::endl;
return true;
}