From 180804e23d0cf8f64d906a43f96bf240b2d2d389 Mon Sep 17 00:00:00 2001
From: =?utf8?q?=EB=B0=95=EC=B2=9C=EA=B5=90/On-Device=20Lab=28SR=29/Enginee?=
 =?utf8?q?r/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= <ch.bahk@samsung.com>
Date: Wed, 6 Nov 2019 16:55:05 +0900
Subject: [PATCH] [exo] Register TensorReduce conversion (#8809)

This commit registers conversion for TensorReduce to exo conversion
stage.

Signed-off-by: Cheongyo Bahk <ch.bahk@samsung.com>
---
 compiler/exo/src/Conversions.h | 1 +
 compiler/exo/src/Convert.cpp   | 1 +
 2 files changed, 2 insertions(+)

diff --git a/compiler/exo/src/Conversions.h b/compiler/exo/src/Conversions.h
index bc540d6..a952ebf 100644
--- a/compiler/exo/src/Conversions.h
+++ b/compiler/exo/src/Conversions.h
@@ -36,6 +36,7 @@
 #include "Conversion/TensorConcatConverter.h"
 // TODO loco::TensorBiasAdd
 #include "Conversion/TensorBroadcastConverter.h"
+#include "Conversion/TensorReduceConverter.h"
 // TODO loco::TensorSoftmax
 #include "Conversion/TensorTransposeConverter.h"
 #include "Conversion/TransposedConv2DConverter.h"
diff --git a/compiler/exo/src/Convert.cpp b/compiler/exo/src/Convert.cpp
index 850c552..8cb0e41 100644
--- a/compiler/exo/src/Convert.cpp
+++ b/compiler/exo/src/Convert.cpp
@@ -72,6 +72,7 @@ void convert_to_TFLNodes(loco::Graph *graph)
     phase.emplace_back(stdex::make_unique<TensorConcatConverter>());
     // TODO loco::TensorBiasAdd
     phase.emplace_back(stdex::make_unique<TensorBroadcastConverter>());
+    phase.emplace_back(stdex::make_unique<TensorReduceConverter>());
     // TODO loco::TensorSoftmax
     phase.emplace_back(stdex::make_unique<TensorTransposeConverter>());
     phase.emplace_back(stdex::make_unique<TransposedConv2DConverter>());
-- 
2.7.4