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 = [{