[mir_onnx] Inherit tensor names (#6855)
authorСергей Баранников/AI Tools Lab /SRR/Engineer/삼성전자 <s.barannikov@samsung.com>
Thu, 22 Aug 2019 12:38:49 +0000 (21:38 +0900)
committerAlexander Efimov/AI Tools Lab/./Samsung Electronics <a.efimov@samsung.com>
Thu, 22 Aug 2019 12:38:49 +0000 (15:38 +0300)
Set `Operation::Output`s names based on model tensor names.

Signed-off-by: Sergei Barannikov <s.barannikov@samsung.com>
compiler/mir-onnx-importer/ONNXNodeConverterRegistry.h

index e0f999c..8efeefd 100644 (file)
@@ -51,6 +51,7 @@ public:
 
   void setOutput(const std::string &name, mir::Operation::Output *output)
   {
+    output->setName(name);
     auto result = _tensorNameToOutput.emplace(name, output);
     if (!result.second)
       throw std::runtime_error("Name duplication: " + output->getNode()->getName());