#define GEN_PASS_DEF_TPUMERGEVARIABLESWITHEXECUTEPASS
#include "tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/runtime_passes.h.inc"

class TPUMergeVariablesWithExecutePass
    : public impl::TPUMergeVariablesWithExecutePassBase<
          TPUMergeVariablesWithExecutePass> {
  void runOnOperation() override;
};

// Information for a pair of input/output of the TPUExecute op and the
// surrounding read/assign ops.

      }
    ```
  }];
  let dependentDialects = [
    "mhlo::MhloDialect",
    "tf_device::TensorFlowDeviceDialect"
  ];
}

def TPUMergeVariablesWithExecutePass : Pass<"tf-tpu-merge-variables-with-execute", "ModuleOp"> {
  let summary = "Merges device variable reads and updates into TPU execute ops";

  let description = [{