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

Move canonicalizer pass to after TPU cluster formation to prevent canonicalization from losing _tpu_replicate attributes

Browse Source 
PiperOrigin-RevId: 279394331
Change-Id: Ia1eb850d8bb09620d5e3c333cd77ac9a64960c39
This commit is contained in:
Yanan Cao 2019-11-08 14:42:53 -08:00 committed by TensorFlower Gardener
parent a3904f96a0
commit 514cf2d96e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc
View File

@ -26,8 +26,8 @@ namespace TFTPU {
void createTPUBridge(OpPassManager &pm) {
OpPassManager &func_pm = pm.nest<FuncOp>();
func_pm.addPass(tf_executor::CreateTFExecutorIslandCoarseningPass());
func_pm.addPass(createCanonicalizerPass());
func_pm.addPass(CreateTPUClusterFormationPass());
func_pm.addPass(createCanonicalizerPass());
func_pm.addPass(tf_executor::CreateTFExecutorConstantSinkingPass());
func_pm.addPass(TFDevice::CreateResourceOpLiftingPass());