results.add<WhileResultOperandsMatchAndImplicitCapture>(context);
}

SmallVector<Region*> WhileOp::getLoopRegions() { return {&getBody()}; }

bool WhileOp::isDefinedOutsideOfLoop(Value value) {
  // TODO(jpienaar): This is to overly conservative and disables anything other
  // than constant hoisting initially.
  return false;
}

  // tf.IfOp to scf.If op first and then legalize it to tfl.if to reduce
  // code redundancy.
}

def TFL_WhileOp : Op<TFL_Dialect, "while", [
    DeclareOpInterfaceMethods<LoopLikeOpInterface, ["isDefinedOutsideOfLoop"]>,
    SingleBlockImplicitTerminator<"YieldOp">,
    DeclareOpInterfaceMethods<TFL_RuntimeVerification>]> {
  let summary = [{While loop}];

  let description = [{