Searched defs:pytree (Results 1 – 5 of 5) sorted by relevance
25 def ravel_pytree(pytree): argument
1047 def _std_basis(pytree): argument1055 def _unravel_array_into_pytree(pytree, axis, arr): argument
57 pytree = xla_client._xla.pytree variable
313 def msgpack_serialize(pytree): argument
126 def _shapes(pytree): argument