Searched refs:compiled_function (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/ |
| H A D | autograd_cache.py | 373 compiled_function = AOTDispatchAutograd.post_compile(
387 compiled_function = RuntimeWrapper(
395 compiled_function, _ = post_compile(
397 compiled_function,
402 return compiled_function
|
| H A D | runtime_wrappers.py | 2063 compiled_function = RuntimeWrapper(
2073 return compiled_function
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/optests/ |
| H A D | aot_autograd.py | 6 from functorch.compile import compiled_function, min_cut_rematerialization_partition, nop
66 compiled_f = compiled_function(
|
| /aosp_15_r20/external/pytorch/functorch/compile/ |
| H A D | __init__.py | 6 compiled_function,
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/eager/ |
| H A D | benchmarks_test.py | 834 def compiled_function(x, tangent): function
841 func = lambda: compiled_function(m, tangent)
853 def compiled_function(x, tangent): function
860 func = lambda: compiled_function(m, tangent)
|
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | aot_autograd.py | 1556 compiled_function = aot_function variable
|
| /aosp_15_r20/external/pytorch/test/functorch/ |
| H A D | test_aotdispatch.py | 29 compiled_function,
5057 compiled_fn = compiled_function(
|