namespace mlir {
namespace TF {

namespace {

#define GEN_PASS_DEF_MOVETPUCOMPILETOFRONTPASS
#include "tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.h.inc"

struct MoveTpuCompileToFrontPass
    : public impl::MoveTpuCompileToFrontPassBase<MoveTpuCompileToFrontPass> {
  void runOnOperation() override;
};

void MarkCompilationOps(Operation* func) {
  func->walk([&](Operation* op) {
    if (llvm::isa<TF::_TPUCompileMlirOp>(op)) {

    pipeline where you want to dump, and are able to easily modify the pipeline
    source code.
  }];
  let constructor = "TF::CreatePrintPass()";
}

def MoveTpuCompileToFrontPass : Pass<"tf-move-tpu-compile-to-front", "mlir::func::FuncOp"> {
  let summary = "Moves compile ops to the front.";
  let description = [{
    Moves all _TPUCompileMlir ops as far to the front as possible. For example,