}
    return success();
  }
};

class LiftTfliteFlexOpsPass
    : public impl::LiftTfliteFlexOpsPassBase<LiftTfliteFlexOpsPass> {
  void getDependentDialects(DialectRegistry& registry) const override {
    registry.insert<TF::TensorFlowDialect>();
  }

 public:
  MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(LiftTfliteFlexOpsPass)

  void runOnOperation() override {

  let summary = "Legalize TensorFlow variables to TensorFlow Lite dialect.";
  let constructor = "CreateLegalizeVariablesPass()";
  let dependentDialects = ["TFL::TensorFlowLiteDialect"];
}

def LiftTfliteFlexOpsPass : Pass<"tfl-lift-tflite-flex-ops", "mlir::func::FuncOp"> {
  let summary = "Lifts TFLite Custom ops into TF dialect operations.";
  let constructor = "CreateLiftTfliteFlexOpsPass()";