Home
last modified time | relevance | path

Searched refs:aot_autograd_arg_pos_to_source (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_functorch/
H A Daot_autograd.py848 aot_autograd_arg_pos_to_source=None,
982 aot_autograd_arg_pos_to_source = None
985 aot_autograd_arg_pos_to_source = []
994 aot_autograd_arg_pos_to_source.append(source)
1005 if aot_autograd_arg_pos_to_source is None:
1006 aot_autograd_arg_pos_to_source = []
1010 aot_autograd_arg_pos_to_source.append(source)
1025 if aot_autograd_arg_pos_to_source is not None:
1026 assert len(full_args) == len(aot_autograd_arg_pos_to_source)
1044 aot_autograd_arg_pos_to_source=aot_autograd_arg_pos_to_source,
[all …]
/aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/
H A Druntime_wrappers.py902 and aot_config.aot_autograd_arg_pos_to_source
911 dupe_arg_source = aot_config.aot_autograd_arg_pos_to_source[
914 kept_arg_source = aot_config.aot_autograd_arg_pos_to_source[
H A Dautograd_cache.py201 aot_config.aot_autograd_arg_pos_to_source,
H A Dschemas.py815 aot_autograd_arg_pos_to_source: Optional[List[Source]] = None variable in AOTConfig
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_aot_autograd_cache.py568 aot_autograd_arg_pos_to_source=None,