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

Remove TPUMergeVariablesWithExecutePass from TPU bridge pipeline.

Browse Source 
TPUMergeVariablesWithExecutePass should already handle merging ReadVariableOp and AssignVariableOp across ParallelExecute.

PiperOrigin-RevId: 336152089
Change-Id: I64a210da38cae3d8b8b2e05a865f61a4fdd82a46
This commit is contained in:
Andy Ly 2020-10-08 13:23:27 -07:00 committed by TensorFlower Gardener
parent c51c793a53
commit 84202f860f
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -120,7 +120,6 @@ void CreateTPUBridgePipeline(OpPassManager &pm) {
pm.addPass(createSymbolDCEPass());
pm.addNestedPass<FuncOp>(TFDevice::CreateReplicateInvariantOpHoistingPass());
pm.addNestedPass<FuncOp>(CreateTPUDynamicLayoutPass());
pm.addNestedPass<FuncOp>(CreateTPUParallelExecuteSinkResourceWritePass());
pm.addNestedPass<FuncOp>(CreateTPUMergeVariablesWithExecutePass());
pm.addNestedPass<FuncOp>(CreateTPUColocateCompositeResourceOps());
pm.addPass(CreateTPUVariableReformattingPass());