}

// Only legalize TensorFlow TensorList ops if all TensorList ops are supported
// natively.
class LegalizeTensorListPass
    : public impl::LegalizeTensorListPassBase<LegalizeTensorListPass> {
 public:
  MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(LegalizeTensorListPass)

  void runOnOperation() override {
    mlir::ModuleOp module = getOperation();

  let summary = "Hoist while op regions into functions";
  let constructor = "CreateWhileOutlinePass()";
  let dependentDialects = ["TF::TensorFlowDialect"];
}

def LegalizeTensorListPass : Pass<"tfl-legalize-tensorlist", "mlir::ModuleOp"> {
  let summary = "Legalize TensorFlow tensorlist ops to TensorFlow Lite custom.";
  let constructor = "CreateLegalizeTensorListPass()";