From b131b30fb59aa41bc826588b53571f6f98dcabc7 Mon Sep 17 00:00:00 2001
From: Eugene Zhulenev <ezhulenev@google.com>
Date: Fri, 15 Jan 2021 08:08:50 -0800
Subject: [PATCH] PR #46097: [INTEL MKL] Change order for remapper
Imported from GitHub PR https://github.com/tensorflow/tensorflow/pull/46097
This PR is to test if there will be any regression if the order of remapper in grappler meta_optimizer is moved before arithmetic_optimizer.
Copybara import of the project:
--
613041d6e6b28f331aecd01aa6f69c8f0953fdac by mdfaijul <md.faijul.amin@intel.com>:
Change order for remapper.
PiperOrigin-RevId: 352011080
Change-Id: I5041e87bfeeb41120517ce454d1baa247c32d4fb
---
tensorflow/core/grappler/optimizers/meta_optimizer.cc | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/tensorflow/core/grappler/optimizers/meta_optimizer.cc b/tensorflow/core/grappler/optimizers/meta_optimizer.cc
index cc375124047..40094932814 100644
--- a/tensorflow/core/grappler/optimizers/meta_optimizer.cc
+++ b/tensorflow/core/grappler/optimizers/meta_optimizer.cc
@@ -269,9 +269,6 @@ Status MetaOptimizer::InitializeOptimizers(
if (cfg_.pin_to_host_optimization() == RewriterConfig::ON) {
optimizers->push_back(MakeUnique<PinToHostOptimizer>());
}
- if (cfg_.remapping() != RewriterConfig::OFF) {
- optimizers->push_back(MakeUnique<Remapper>(cfg_.remapping()));
- }
if (cfg_.arithmetic_optimization() != RewriterConfig::OFF) {
optimizers->push_back(
MakeUnique<ArithmeticOptimizer>(cfg_.arithmetic_optimization()));
@@ -281,6 +278,9 @@ Status MetaOptimizer::InitializeOptimizers(
/*optimization level*/ cfg_.layout_optimizer(),
/*CPU layout conversion*/ cfg_.cpu_layout_conversion()));
}
+ if (cfg_.remapping() != RewriterConfig::OFF) {
+ optimizers->push_back(MakeUnique<Remapper>(cfg_.remapping()));
+ }
if (cfg_.loop_optimization() != RewriterConfig::OFF) {
optimizers->push_back(
MakeUnique<LoopOptimizer>(cfg_.loop_optimization(), cpu_device_));