::llvm::cl::desc(
          "Skip tf.StatefulPartitionedCall and tf.PartitionedCall")};
};

void PopulateLegalizeTFToStablehloPipeline(
    OpPassManager &pm, const TFToStablehloOptions &options) {
  // TODO(burmako): Migrate this pass from producing MHLO to producing StableHLO
  // by aligning with the TF/XLA bridge on the corresponding functionality and
  // reusing their work, perhaps through `LowerToMlProgramAndHlo`.