class_docstr: _LayerStack impl with no per-layer inputs provided to the function.
class_docstr_tok: ["_LayerStack","impl","with","no","per-layer","inputs","provided","to","the","function","."]
code: def _call_wrapped(self, x, **kwargs):
      ret = self._f(*x, **kwargs)
      if len(x) == 1:
          # If the function takes a single argument, the wrapped function receives
          # a tuple of length 1, and therefore it must return a tuple of length 1.
          ret = (ret,)
      return ret, None
  
imports: import collections
  import functools
  import inspect
  from typing import Any, Callable, Optional, Protocol, Union
  from haiku._src import base
  from haiku._src import lift
  from haiku._src import module
  from haiku._src import transform
  import jax
  import jax.numpy
label_desc: Use image node_id 2 for calling the _LayerStackNoPerLayer obj's underlying member method code with example usage: obj._call_wrapped(x) and returns: ret,  None