keras/src/ops/core.py/ForiLoop/__init__
class ForiLoop:    def __init__(self, lower, upper, body_fun):
        super().__init__()
        self.lower = lower
        self.upper = upper
        self.body_fun = body_fun