Searched defs:compiled_fn (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/ |
| H A D | runtime_wrappers.py | 102 def post_compile(self, compiled_fn, aot_config, *, runtime_metadata) -> Callable: argument
136 compiled_fn, argument
242 compiled_fn, argument
469 compiled_fn, argument
563 compiled_fn, argument
617 compiled_fn, argument
652 compiled_fn, argument
940 compiled_fn, argument
1134 compiled_fn, argument
2082 compiled_fn, argument
|
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_aot_autograd_cache.py | 321 def expect_miss(compiled_fn, a, b): argument
333 def expect_hit(compiled_fn, a, b): argument
348 def expect_guard_miss(compiled_fn, a, b): argument
|
| H A D | test_repros.py | 3288 def compiled_fn(x): function
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/utils/ |
| H A D | lambda_post_hook.h | 21 LambdaPostHook(fn_type fn, compiled_fn_type compiled_fn) in LambdaPostHook()
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | compiled_autograd.py | 333 def runtime_wrapper(compiled_fn, inputs, sizes, scalars, hooks): argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/dynamo/ |
| H A D | python_compiled_autograd.cpp | 307 THPObjectPtr compiled_fn; member
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/python/ |
| H A D | python_sugared_value.cpp | 1367 auto compiled_fn = py::module::import("torch.jit._recursive") in toSugaredValue() local
|