namespace {
#define GEN_PASS_DEF_REDUCEWHILEOPERANDSPASS
#include "tensorflow/compiler/mlir/lite/transforms/passes.h.inc"

struct ReduceWhileOperandsPass
    : public impl::ReduceWhileOperandsPassBase<ReduceWhileOperandsPass> {
 public:
  MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(ReduceWhileOperandsPass)

  void getDependentDialects(DialectRegistry &registry) const override {

  let options = [
      ListOption<"target_ops_", "test-raise-tf-targets", "std::string",
             "comma separated list of target op names to be wrapped. Only used in tests">,
  ];
}

def ReduceWhileOperandsPass : Pass<"tfl-reduce-while", "mlir::func::FuncOp"> {
  let summary = "Reduce the number of operands and results of a whlieOp..";
  let constructor = "CreateReduceWhileOperandsPass()";