// This transformation pass takes an operation with unknown op properties and
// wrap it by a TFL::CustomTfOp.
struct RaiseCustomOpsPass
    : public impl::RaiseCustomOpsPassBase<RaiseCustomOpsPass> {
 public:
  MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(RaiseCustomOpsPass)

  explicit RaiseCustomOpsPass() {}
  explicit RaiseCustomOpsPass(const std::vector<std::string> &target_ops) {
    this->target_ops_ = target_ops;
  }

  let summary = "Quantize variables";
  let constructor = "CreatePrepareQuantizeVariablesPass()";
  let dependentDialects = ["TFL::TensorFlowLiteDialect"];
}

def RaiseCustomOpsPass : Pass<"tfl-raise-custom-ops", "mlir::func::FuncOp"> {
  let summary = "Raise custom ops into tflite dialect.";
  let constructor = "CreateRaiseCustomOpsPass()";
  let dependentDialects = ["TFL::TensorFlowLiteDialect"];