Home
last modified time | relevance | path

Searched defs:ViewAndMutationMeta (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_functorch/_aot_autograd/
H A Druntime_wrappers.py90 fw_metadata: ViewAndMutationMeta,
139 runtime_metadata: ViewAndMutationMeta,
244 runtime_metadata: ViewAndMutationMeta,
472 runtime_metadata: ViewAndMutationMeta,
496 metadata: ViewAndMutationMeta,
566 runtime_metadata: ViewAndMutationMeta,
603 fw_metadata: ViewAndMutationMeta,
620 runtime_metadata: ViewAndMutationMeta,
655 runtime_metadata: ViewAndMutationMeta,
782 fw_metadata: ViewAndMutationMeta,
[all …]
H A Dsubclass_utils.py23 def requires_subclass_dispatch(args, fw_metadata: ViewAndMutationMeta) -> bool:
222 unwrapped_args, *, is_joint_structure: bool, meta: ViewAndMutationMeta
246 def create_metadata_for_subclass(meta: ViewAndMutationMeta) -> ViewAndMutationMeta:
299 fw_metadata: ViewAndMutationMeta,
300 inner_metadata: ViewAndMutationMeta,
H A Djit_compile_runtime_wrappers.py90 fw_metadata: ViewAndMutationMeta,
130 fw_metadata: ViewAndMutationMeta,
284 fw_metadata: ViewAndMutationMeta,
317 fw_metadata: ViewAndMutationMeta,
H A Dtraced_function_transforms.py73 meta: ViewAndMutationMeta,
107 meta: ViewAndMutationMeta,
392 meta: ViewAndMutationMeta,
658 meta: ViewAndMutationMeta,
739 meta: ViewAndMutationMeta,
H A Dinput_output_analysis.py39 m: ViewAndMutationMeta,
108 m: ViewAndMutationMeta,
420 fw_metadata: ViewAndMutationMeta,
H A Ddispatch_and_compile_graph.py69 fw_metadata: ViewAndMutationMeta,
239 fw_metadata: ViewAndMutationMeta,
H A Dschemas.py254 class ViewAndMutationMeta: class
701 view_mutation_metadata: ViewAndMutationMeta,
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dfreezing.py30 fw_metadata: torch._functorch.aot_autograd.ViewAndMutationMeta,