mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-17 12:26:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #15289 from cgarciae:add-missing-api-references

Browse Source 
PiperOrigin-RevId: 521617419
This commit is contained in:
jax authors 2023-04-03 18:23:04 -07:00
commit 8a6c929678
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
docs/jax.tree_util.rst
View File

@ -16,10 +16,13 @@ List of Functions
build_tree
register_pytree_node
register_pytree_node_class
register_pytree_with_keys
register_pytree_with_keys_class
tree_all
tree_flatten
tree_leaves
tree_map
tree_map_with_path
tree_reduce
tree_structure
tree_transpose