// more efficient computationally.
  bool shouldMaterializeInto(Region* region) const final {
    return isa<WhileOp>(region->getParentOp());
  }
};

void TFLDialect::printType(Type type, DialectAsmPrinter& os) const {
  if (type.isa<ControlType>()) {
    os << "control";
    return;
  }
  os << "<unknown TFL type>";
}

Type TFLDialect::parseType(DialectAsmParser& parser) const {