From 84202f860fa45b0508a9a23aae4698fc044e0bb0 Mon Sep 17 00:00:00 2001
From: Andy Ly <lyandy@google.com>
Date: Thu, 8 Oct 2020 13:23:27 -0700
Subject: [PATCH] Remove TPUMergeVariablesWithExecutePass from TPU bridge
pipeline.
TPUMergeVariablesWithExecutePass should already handle merging ReadVariableOp and AssignVariableOp across ParallelExecute.
PiperOrigin-RevId: 336152089
Change-Id: I64a210da38cae3d8b8b2e05a865f61a4fdd82a46
---
tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc | 1 -
1 file changed, 1 deletion(-)
diff --git a/tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc b/tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc
index 2e1e63eb143..eccbe5feaec 100644
--- a/tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc
+++ b/tensorflow/compiler/mlir/tensorflow/transforms/bridge.cc
@@ -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());