From 8e903bde5fcedd6f83e146ddb9f5656376369c68 Mon Sep 17 00:00:00 2001 From: =?utf8?q?=D0=A1=D0=B5=D1=80=D0=B3=D0=B5=D0=B9=20=D0=91=D0=B0=D1=80?= =?utf8?q?=D0=B0=D0=BD=D0=BD=D0=B8=D0=BA=D0=BE=D0=B2/AI=20Tools=20Lab=20/S?= =?utf8?q?RR/Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Wed, 28 Aug 2019 21:43:14 +0900 Subject: [PATCH] [mir_onnx] Do not set names of operations (#6956) Remove useless setting of operation names. Signed-off-by: Sergei Barannikov --- compiler/mir-onnx-importer/ONNXHelpers.h | 3 +-- compiler/mir-onnx-importer/ONNXImporterImpl.cpp | 8 +++----- compiler/mir-onnx-importer/ONNXNodeConverterRegistry.h | 3 +-- compiler/mir-onnx-importer/Op/Constant.cpp | 3 +-- 4 files changed, 6 insertions(+), 11 deletions(-) diff --git a/compiler/mir-onnx-importer/ONNXHelpers.h b/compiler/mir-onnx-importer/ONNXHelpers.h index 73c3a2d..334fe17 100644 --- a/compiler/mir-onnx-importer/ONNXHelpers.h +++ b/compiler/mir-onnx-importer/ONNXHelpers.h @@ -32,8 +32,7 @@ mir::TensorVariant createTensor(const onnx::TensorProto *tensor); template mir::Operation *createOp(mir::Graph *graph, Types &&... args) { - // TODO: how to name operations? - return graph->create("", std::forward(args)...); + return graph->create(std::forward(args)...); } inline mir::Operation::Output *convertONNXToMIR(mir::Graph *graph, mir::Operation::Output *arg) diff --git a/compiler/mir-onnx-importer/ONNXImporterImpl.cpp b/compiler/mir-onnx-importer/ONNXImporterImpl.cpp index d1efd07..29d5b70 100644 --- a/compiler/mir-onnx-importer/ONNXImporterImpl.cpp +++ b/compiler/mir-onnx-importer/ONNXImporterImpl.cpp @@ -108,9 +108,8 @@ void ONNXImporterImpl::createGraphInputs() // Create all initializer Tensors for (const auto &tensor : initializer) { - assert(tensor.has_name()); const auto mir_tensor = createTensor(&tensor); - auto *op = _graph->create(tensor.name(), mir_tensor); + auto *op = _graph->create(mir_tensor); _context->setOutput(tensor.name(), op->getOutput(0)); } @@ -128,7 +127,7 @@ void ONNXImporterImpl::createGraphInputs() shape.dim(i) = static_cast(onnx_input_shape.dim(i).dim_value()); } - auto *op = _graph->create(input.name(), shape); + auto *op = _graph->create(shape); _context->setOutput(input.name(), op->getOutput(0)); } } @@ -172,8 +171,7 @@ std::unique_ptr ONNXImporterImpl::createIR() if (mir_output == nullptr) throw std::runtime_error("Bad output name!"); - _graph->create(output.name(), mir_output); - mir_output->getNode()->setName(""); + _graph->create(mir_output); } return std::move(_graph); diff --git a/compiler/mir-onnx-importer/ONNXNodeConverterRegistry.h b/compiler/mir-onnx-importer/ONNXNodeConverterRegistry.h index 8efeefd..da3a34c 100644 --- a/compiler/mir-onnx-importer/ONNXNodeConverterRegistry.h +++ b/compiler/mir-onnx-importer/ONNXNodeConverterRegistry.h @@ -54,7 +54,7 @@ public: output->setName(name); auto result = _tensorNameToOutput.emplace(name, output); if (!result.second) - throw std::runtime_error("Name duplication: " + output->getNode()->getName()); + throw std::runtime_error("Name duplication: " + name); } mir::Operation::Output *getOutput(const std::string &name) const @@ -89,7 +89,6 @@ public: assert(!outputs.empty()); for (std::size_t i = 0; i < outputs.size(); ++i) { - outputs[i]->getNode()->setName(onnx_node.output(i)); setOutput(onnx_node.output(i), outputs[i]); } } diff --git a/compiler/mir-onnx-importer/Op/Constant.cpp b/compiler/mir-onnx-importer/Op/Constant.cpp index 3a1a9e9..8cf3f4d 100644 --- a/compiler/mir-onnx-importer/Op/Constant.cpp +++ b/compiler/mir-onnx-importer/Op/Constant.cpp @@ -46,10 +46,9 @@ void ConstantNodeConverter::convertV1(const onnx::NodeProto &onnx_node, mir::Graph *graph = context->getGraph(); const auto onnx_tensor = getAttributeValue(onnx_node, "value"); - const auto &name = onnx_node.output(0); auto mir_tensor = createTensor(&onnx_tensor); - auto result = graph->create(name, mir_tensor)->getOutput(0); + auto result = graph->create(mir_tensor)->getOutput(0); context->setNodeOutputs(onnx_node, {result}); } -- 2.7.4