} else if (auto while_op = llvm::dyn_cast<WhileOp>(op)) {
      if (failed(ConvertWhileOp(while_op, allow_passthrough_args_))) {
        op->emitOpError() << "failed to convert to region form";
        return WalkResult::interrupt();
      }
    } else if (auto generator_op = llvm::dyn_cast<GeneratorDatasetOp>(op)) {
      if (allow_passthrough_args_) {
        if (failed(ConvertGeneratorDatasetOp(generator_op))) {

        }) {is_stateless = false} : (tensor<i1>) -> tensor<*xf32>
    ```
  }];

  let constructor = "TF::CreateTFFunctionalControlFlowToRegions()";

  let options = [
    Option<"allow_passthrough_args_", "allow-passthrough-args", "bool",
           /*default=*/"false",
           "Allows to switch on passthrough args for control flow logic. This "

// their region based counterparts.
std::unique_ptr<OperationPass<ModuleOp>>
CreateTFFunctionalControlFlowToRegions();
std::unique_ptr<OperationPass<ModuleOp>> CreateTFFunctionalControlFlowToRegions(
    bool allow_passthrough_args);

// Transforms region bases control flow operations in the TensorFlow dialect to
// their functional counterparts.
std::unique_ptr<OperationPass<ModuleOp>>
CreateTFRegionControlFlowToFunctional();