o->remove();
          block.push_back(o);
        }
      }
    }
  });
  UnmarkCompilationOps(getOperation());
}

}  // namespace

std::unique_ptr<OperationPass<func::FuncOp>> CreateMoveTpuCompileToFrontPass() {
  return std::make_unique<MoveTpuCompileToFrontPass>();
}

}  // namespace TF

std::unique_ptr<OperationPass<ModuleOp>> CreatePrintPass(
    raw_ostream* os = nullptr);

// Moves TPUCompileMlir ops as far to the front as possible.
std::unique_ptr<OperationPass<func::FuncOp>> CreateMoveTpuCompileToFrontPass();

// Decomposes OptionalFromValue, OptionalGetValue, OptionalNone,
// and OptionalHasValue
std::unique_ptr<OperationPass<ModuleOp>> CreateDecomposeOptionalsPass();

    a data or control dependency prevening the reordering, it will leave the ops
    where they are.
  }];
  let constructor = "TF::CreateMoveTpuCompileToFrontPass()";
}

def DecomposeOptionalsPass : Pass<"tf-decompose-optionals", "ModuleOp"> {
  let summary = "Decomposes ops of the 'Optional' family";
  let description = [{