[enco/frontend] Add AvgPool2DGraphBuilder into registry (#2253)
author윤현식/동작제어Lab(SR)/Principal Engineer/삼성전자 <hyunsik.yoon@samsung.com>
Tue, 13 Nov 2018 10:43:11 +0000 (19:43 +0900)
committer박종현/동작제어Lab(SR)/Staff Engineer/삼성전자 <jh1302.park@samsung.com>
Tue, 13 Nov 2018 10:43:11 +0000 (19:43 +0900)
With this commit, AvgPool2DGraphBuilder is added into registry.

Signed-off-by: Hyun Sik Yoon <hyunsik.yoon@samsung.com>
contrib/enco/frontend/tflite/src/Frontend.cpp

index 0f74f14..966ec37 100644 (file)
@@ -665,6 +665,7 @@ private:
   {
     // add GraphBuilder for each tflite operation.
     _builder_map[tflite::BuiltinOperator_CONV_2D] = make_unique<Conv2DGraphBuilder>();
+    _builder_map[tflite::BuiltinOperator_AVERAGE_POOL_2D] = make_unique<AvgPool2DGraphBuilder>();
   }
 
 private: