[mir_onnx] Register Transpose operator converter (#6947)
authorСергей Баранников/AI Tools Lab /SRR/Engineer/삼성전자 <s.barannikov@samsung.com>
Wed, 28 Aug 2019 07:54:25 +0000 (16:54 +0900)
committerAlexander Efimov/AI Tools Lab/./Samsung Electronics <a.efimov@samsung.com>
Wed, 28 Aug 2019 07:54:25 +0000 (10:54 +0300)
It was lost not long ago after improper resolving of a merge conflict.

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

index 2d458a0..c3eb51f 100644 (file)
@@ -79,6 +79,7 @@ inline void registerSupportedOps()
   registry.registerConverter("Sigmoid", stdex::make_unique<SigmoidNodeConverter>());
   registry.registerConverter("Softmax", stdex::make_unique<SoftmaxNodeConverter>());
   registry.registerConverter("Sum", stdex::make_unique<SumNodeConverter>());
+  registry.registerConverter("Transpose", stdex::make_unique<TransposeNodeConverter>());
   registry.registerConverter("Unsqueeze", stdex::make_unique<UnsqueezeNodeConverter>());
   registry.registerConverter("Upsample", stdex::make_unique<UpsampleNodeConverter>());
 }