decomposition.getValue());
    rewriter.replaceOp(op, composite.getResults());
    return success();
  }
};

struct LegalizeStablehloCustomCallToCompositePass
    : public impl::LegalizeStablehloCustomCallToCompositePassBase<
          LegalizeStablehloCustomCallToCompositePass> {
  using LegalizeStablehloCustomCallToCompositePassBase::
      LegalizeStablehloCustomCallToCompositePassBase;

  let summary = "Legalize supported StableHLO CompositeOps to TFL CustomOp";
  let dependentDialects = ["TFL::TensorFlowLiteDialect"];
}
def LegalizeStablehloCustomCallToCompositePass : Pass<"stablehlo-custom-call-legalize-composite", "ModuleOp"> {
  let summary = "Legalize StableHLO custom call ops where the call target is 'stablehlo.composite' to composite ops.";