%identity = "tf.Identity"(%arg0) : (tensor<i32>) -> tensor<i32>
      return %identity : tensor<i32>
    }
    ```
  }];

  let constructor = "TFTPU::CreateTPUResourceReadForWritePass()";
}

def HoistReplicateInvariantResourceWritesPass : Pass<"tf-hoist-replicate-invariant-resource-writes", "mlir::func::FuncOp"> {
  let summary = "Hoists writes to replicate invariant resource variables.";