Searched refs:_tree_map_with_path (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/export/ |
| H A D | dynamic_shapes.py | 433 def _tree_map_with_path( function
634 dynamic_shapes = _tree_map_with_path(find_shape, combined_args)
764 _tree_map_with_path(check_shape, combined_args, dynamic_shapes, tree_name="inputs")
911 result = _tree_map_with_path(
1043 _tree_map_with_path(assoc_shape, combined_args, dynamic_shapes, tree_name="inputs")
1220 return _tree_map_with_path(apply_fixes, dynamic_shapes, dynamic_shapes)
|
| /aosp_15_r20/external/pytorch/torch/_export/serde/ |
| H A D | dynamic_shapes.py | 11 _tree_map_with_path,
193 tree_shapes = _tree_map_with_path(
|
| /aosp_15_r20/external/pytorch/torch/_export/ |
| H A D | non_strict_utils.py | 30 _tree_map_with_path,
239 _tree_map_with_path(_tree_map_helper, combined_args, dynamic_shapes)
|