diff --git a/keras/src/layers/layer.py b/keras/src/layers/layer.py index 6dd94555760..f405173aee5 100644 --- a/keras/src/layers/layer.py +++ b/keras/src/layers/layer.py @@ -1797,6 +1797,22 @@ def call(self, x): ) + def register_call_context_args(self, *names): + """Public wrapper for `_register_call_context_args`. + + This method registers call-context arguments for this layer. It simply + forwards the provided argument names to the private method + `_register_call_context_args`. See `_register_call_context_args` for + details. + + Args: + *names: Names of call-context arguments to register. + + Returns: + None + """ + return self._register_call_context_args(*names) + def is_backend_tensor_or_symbolic(x, allow_none=False): if allow_none and x is None: return True