[moco-tf] Register Conv2DBackpropInput canonicalization (#7649)
author박천교/On-Device Lab(SR)/Engineer/삼성전자 <ch.bahk@samsung.com>
Fri, 20 Sep 2019 07:27:16 +0000 (16:27 +0900)
committer오형석/On-Device Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Fri, 20 Sep 2019 07:27:16 +0000 (16:27 +0900)
This commit registers Conv2DBackpropInput canonicalization.

Signed-off-by: Cheongyo Bahk <ch.bahk@samsung.com>
compiler/moco-tf/src/Canonicalizer.cpp

index 6f39ca3..27c0145 100644 (file)
@@ -27,6 +27,7 @@
 #include "Canonicalization/BiasAddCanonicalizer.h"
 #include "Canonicalization/ConcatV2Canonicalizer.h"
 #include "Canonicalization/ConstCanonicalizer.h"
+#include "Canonicalization/Conv2DBackpropInputCanonicalizer.h"
 #include "Canonicalization/Conv2DCanonicalizer.h"
 #include "Canonicalization/DepthwiseConv2dNativeCanonicalizer.h"
 #include "Canonicalization/IdentityCanonicalizer.h"
@@ -97,6 +98,7 @@ void Canonicalizer::canonicalize(loco::Graph *g) const
   phase.emplace_back(stdex::make_unique<ConcatV2Canonicalizer>());
   if (moco::tf::get<moco::tf::Knob::CanonicalizeConst>())
     phase.emplace_back(stdex::make_unique<ConstCanonicalizer>());
+  phase.emplace_back(stdex::make_unique<Conv2DBackpropInputCanonicalizer>());
   if (moco::tf::get<moco::tf::Knob::CanonicalizeConv2D>())
     phase.emplace_back(stdex::make_unique<Conv2DCanonicalizer>());
   phase.emplace_back(stdex::make_unique<DepthwiseConv2dNativeCanonicalizer>());