Searched defs:flat_fn (Results 1 – 6 of 6) sorted by relevance
| /external/pytorch/torch/_functorch/_aot_autograd/ | ||
| D | dispatch_and_compile_graph.py | 65 flat_fn, argument |
| D | jit_compile_runtime_wrappers.py | 126 flat_fn, argument |
| D | utils.py | 176 def flat_fn(*flat_args): function |
| D | runtime_wrappers.py | 86 flat_fn, argument |
| /external/pytorch/torch/_functorch/ | ||
| D | aot_autograd.py | 515 flat_fn, argument |
| /external/pytorch/torch/_higher_order_ops/ | ||
| D | map.py | 143 def flat_fn(*flat_args): function |