keras/src/ops/core.py/WhileLoop/__init__
class WhileLoop:    def __init__(self, cond, body, maximum_iterations):
        super().__init__()
        self.cond = cond
        self.body = body
        self.maximum_iterations = maximum_iterations