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

struct TPUVariableRuntimeReformattingPass
    : public impl::TPUVariableRuntimeReformattingPassBase<
          TPUVariableRuntimeReformattingPass> {
  void runOnOperation() final;
};

// Returns the earlier value of which `v` is an identity. If `skipped` is

  ];

  let dependentDialects = [
    "mlir::mhlo::MhloDialect",
    "mlir::tf_device::TensorFlowDeviceDialect"
  ];
  let constructor = "mlir::TFTPU::CreateTPURewritePass()";
}

def TPUVariableRuntimeReformattingPass : Pass<"tf-tpu-variable-runtime-reformatting", "ModuleOp"> {
  let summary = "Adds device variable formatting op to allow compilation-guided "
           "variable formatting.";