//   %0 = tfl.custom(<args>) {
//     custom_code = "odml.some_op",
//     custom_option = #tfl<const_bytes : "flexbuffer_serialized_attrs">
//   }
struct LegalizeCompositeToCustomOpPass
    : public impl::LegalizeCompositeToCustomOpPassBase<
          LegalizeCompositeToCustomOpPass> {
  using LegalizeCompositeToCustomOpPassBase::
      LegalizeCompositeToCustomOpPassBase;

  void runOnOperation() override {

def LegalizeVhloToStablehloPass : Pass<"vhlo-legalize-stablehlo", "ModuleOp"> {
  let summary = "Legalize VHLO to StableHLO for ODML.";
  let dependentDialects = ["mlir::stablehlo::StablehloDialect"];
}

def LegalizeCompositeToCustomOpPass : Pass<"stablehlo-composite-legalize-tfl-custom", "func::FuncOp"> {
  let summary = "Legalize supported StableHLO CompositeOps to TFL CustomOp";
  let dependentDialects = ["TFL::TensorFlowLiteDialect"];
}