Home
last modified time | relevance | path

Searched refs:tree_flatten_with_path (Results 1 – 8 of 8) sorted by relevance

/external/pytorch/torch/_inductor/
D__init__.py84 flat_args_with_path, received_spec = pytree.tree_flatten_with_path(
/external/pytorch/test/
Dtest_pytree.py1174 key_leaves, spec = py_pytree.tree_flatten_with_path(
1205 key_leaves, spec = py_pytree.tree_flatten_with_path(pytree)
1235 flat_out, _ = py_pytree.tree_flatten_with_path(pytree)
1245 flat, _ = py_pytree.tree_flatten_with_path(tree)
1281 flat_with_path, spec1 = py_pytree.tree_flatten_with_path(example)
1293 flat, _ = py_pytree.tree_flatten_with_path(tree)
/external/pytorch/torch/export/
D_unlift.py24 flat_args_with_path, received_spec = pytree.tree_flatten_with_path(args)
Dexported_program.py830 flat_args_with_path, received_spec = pytree.tree_flatten_with_path(
Dunflatten.py444 flat_args_with_path, in_spec = pytree.tree_flatten_with_path(
/external/pytorch/torch/_export/
Dutils.py33 tree_flatten_with_path,
755 flat_args_with_path, _ = tree_flatten_with_path(combined_args)
/external/pytorch/torch/utils/
D_cxx_pytree.py919 def tree_flatten_with_path( function
D_pytree.py1502 def tree_flatten_with_path( function
1605 keypath_leaves, treespec = tree_flatten_with_path(tree, is_leaf)