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(TFDevice::CreateReplicateInvariantOpHoistingPass()); pm.addNestedPass(CreateTPUDynamicLayoutPass()); - pm.addNestedPass(CreateTPUParallelExecuteSinkResourceWritePass()); pm.addNestedPass(CreateTPUMergeVariablesWithExecutePass()); pm.addNestedPass(CreateTPUColocateCompositeResourceOps()); pm.addPass(CreateTPUVariableReformattingPass());