experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

PR #46097: [INTEL MKL] Change order for remapper

Browse Source 
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:

--
613041d6e6 by mdfaijul <md.faijul.amin@intel.com>:

Change order for remapper.

PiperOrigin-RevId: 352011080
Change-Id: I5041e87bfeeb41120517ce454d1baa247c32d4fb
This commit is contained in:
Eugene Zhulenev 2021-01-15 08:08:50 -08:00 committed by TensorFlower Gardener
parent 28500f00eb
commit b131b30fb5
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/core/grappler/optimizers/meta_optimizer.cc
View File

@ -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_));