keras/src/ops/core.py/AssociativeScan/call
class AssociativeScan:    def call(self, f, elems, axis=0):
        return backend.core.associative_scan(
            f, elems, reverse=self.reverse, axis=axis
        )