From 42bf04c581aacd9a6456d1cb625ea694ef808159 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?= Date: Tue, 6 Aug 2019 10:46:21 +0900 Subject: [PATCH] [moco-tf] Squeeze Canonicalizer (#6234) Squeeze canonicalizer is to canonicalize TF-dialect TFSqueeze into canonical FixedReshape. Signed-off-by: Cheongyo Bahk --- .../src/Canonicalization/SqueezeCanonicalizer.cpp | 108 +++++++++++++++++++++ .../src/Canonicalization/SqueezeCanonicalizer.h | 46 +++++++++ 2 files changed, 154 insertions(+) create mode 100644 compiler/moco-tf/src/Canonicalization/SqueezeCanonicalizer.cpp create mode 100644 compiler/moco-tf/src/Canonicalization/SqueezeCanonicalizer.h diff --git a/compiler/moco-tf/src/Canonicalization/SqueezeCanonicalizer.cpp b/compiler/moco-tf/src/Canonicalization/SqueezeCanonicalizer.cpp new file mode 100644 index 0000000..6f4e445 --- /dev/null +++ b/compiler/moco-tf/src/Canonicalization/SqueezeCanonicalizer.cpp @@ -0,0 +1,108 @@ +/* + * Copyright (c) 2019 Samsung Electronics Co., Ltd. All Rights Reserved + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#include "SqueezeCanonicalizer.h" + +#include "Annotations/ShapeInferenceData.h" + +#include "Dialect/TFDialect.h" +#include "Dialect/TFNodes.h" +#include "Dialect/TFNodeVisitor.h" +#include "Dialect/TFNodeImpl.h" + +#include + +namespace +{ + +bool canonicalize_squeeze_to_reshape(loco::Graph *graph, moco::tf::TFSqueeze *node) +{ + LOGGER(l); + + INFO(l) << "TFNodeCanonicalize TFSqueeze begin"; + + using FixedReshape = loco::Reshape; + + /** + * This will replace shape inferred TFSqueeze node into canonical FixedReshape + * + * Before + * In ---- TFSqueeze ---- Out(s) + * + * After + * ------ TFSqueeze + * / + * In ---- FixedReshape ----- Out(s) + */ + + auto squeeze_shape = node->annot(); + // canonicalize into FixedReshape is valid when squeeze has shape info + // TODO Support general Squeeze case + assert(squeeze_shape); + + auto squeeze_tensor_shape = squeeze_shape->tensor_shape(); + + // Create loco node to replace + auto reshape = graph->nodes()->create(); + + // Copy shape + reshape->rank(squeeze_tensor_shape.rank()); + for (uint32_t axis = 0; axis < squeeze_tensor_shape.rank(); ++axis) + { + assert(squeeze_tensor_shape.dim(axis).known()); + reshape->dim(axis) = squeeze_tensor_shape.dim(axis); + } + + // replace + auto in = node->input(); + reshape->input(in); + replace(node).with(reshape); + + INFO(l) << "TFNodeCanonicalize TFSqueeze done"; + + return true; +} + +} // namespace + +namespace moco +{ +namespace tf +{ + +bool SqueezeCanonicalizer::run(loco::Graph *graph) +{ + auto active_nodes = loco::active_nodes(loco::output_nodes(graph)); + bool changed = false; + + for (auto node : active_nodes) + { + if (node->dialect() == TFDialect::get()) + { + auto tf_squeeze = dynamic_cast(node); + if (tf_squeeze != nullptr) + { + if (canonicalize_squeeze_to_reshape(graph, tf_squeeze)) + changed = true; + } + } + } + + return changed; +} + +} // namespace tf +} // namespace moco diff --git a/compiler/moco-tf/src/Canonicalization/SqueezeCanonicalizer.h b/compiler/moco-tf/src/Canonicalization/SqueezeCanonicalizer.h new file mode 100644 index 0000000..dc5b2d7 --- /dev/null +++ b/compiler/moco-tf/src/Canonicalization/SqueezeCanonicalizer.h @@ -0,0 +1,46 @@ +/* + * Copyright (c) 2019 Samsung Electronics Co., Ltd. All Rights Reserved + * + * Licensed under the Apache License, Version 2.0 (the "License"); + * you may not use this file except in compliance with the License. + * You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +#ifndef __MOCO_TF_SQUEEZE_CANONICALIZER_H__ +#define __MOCO_TF_SQUEEZE_CANONICALIZER_H__ + +#include "Transform.h" + +#include + +namespace moco +{ +namespace tf +{ + +/** + * @brief Canonicalize TF-dialect TFSqueeze into canonical FixedReshape node + * + * @note There is no canonical Squeeze node + */ +class SqueezeCanonicalizer : public Transform +{ +public: + const char *name(void) const final { return "SqueezeCanonicalizer"; } + +public: + bool run(loco::Graph *graph) override; +}; + +} // namespace tf +} // namespace moco + +#endif // __MOCO_TF_SQUEEZE_CANONICALIZER_H__ -- 2.7.4