| /external/pytorch/torch/_functorch/_aot_autograd/ |
| D | subclass_utils.py | 16 from .schemas import MutationType, SubclassCreationMeta, ViewAndMutationMeta
23 def requires_subclass_dispatch(args, fw_metadata: ViewAndMutationMeta) -> bool: argument
222 unwrapped_args, *, is_joint_structure: bool, meta: ViewAndMutationMeta argument
246 def create_metadata_for_subclass(meta: ViewAndMutationMeta) -> ViewAndMutationMeta: argument
285 metadata = ViewAndMutationMeta(
299 fw_metadata: ViewAndMutationMeta, argument
300 inner_metadata: ViewAndMutationMeta, argument
|
| D | runtime_wrappers.py | 51 ViewAndMutationMeta,
90 fw_metadata: ViewAndMutationMeta, argument
91 ) -> Tuple[Callable, List[Tensor], ViewAndMutationMeta]:
139 runtime_metadata: ViewAndMutationMeta, argument
244 runtime_metadata: ViewAndMutationMeta, argument
456 ) -> Tuple[Callable, List[Tensor], ViewAndMutationMeta]:
472 runtime_metadata: ViewAndMutationMeta, argument
496 metadata: ViewAndMutationMeta, argument
528 ) -> Tuple[Callable, List[Tensor], ViewAndMutationMeta]:
566 runtime_metadata: ViewAndMutationMeta, argument
[all …]
|
| D | input_output_analysis.py | 30 ViewAndMutationMeta,
39 m: ViewAndMutationMeta, argument
42 ) -> ViewAndMutationMeta:
59 return ViewAndMutationMeta(
108 m: ViewAndMutationMeta, argument
114 ) -> Tuple[ViewAndMutationMeta, List[int]]:
252 ViewAndMutationMeta(
420 fw_metadata: ViewAndMutationMeta, argument
|
| D | jit_compile_runtime_wrappers.py | 58 from .schemas import AOTConfig, MutationType, ViewAndMutationMeta
73 DispatchReturn = Tuple[Callable, ViewAndMutationMeta]
90 fw_metadata: ViewAndMutationMeta, argument
130 fw_metadata: ViewAndMutationMeta, argument
284 fw_metadata: ViewAndMutationMeta, argument
317 fw_metadata: ViewAndMutationMeta, argument
|
| D | dispatch_and_compile_graph.py | 26 from .schemas import AOTConfig, SubclassMeta, ViewAndMutationMeta
69 fw_metadata: ViewAndMutationMeta, argument
239 fw_metadata: ViewAndMutationMeta, argument
|
| D | traced_function_transforms.py | 56 ViewAndMutationMeta,
73 meta: ViewAndMutationMeta, argument
107 meta: ViewAndMutationMeta, argument
392 meta: ViewAndMutationMeta, argument
658 meta: ViewAndMutationMeta, argument
739 meta: ViewAndMutationMeta, argument
|
| D | schemas.py | 254 class ViewAndMutationMeta: class
559 if not isinstance(other, ViewAndMutationMeta):
582 fw_metadata: ViewAndMutationMeta
701 view_mutation_metadata: ViewAndMutationMeta, argument
|
| D | collect_metadata_analysis.py | 47 ViewAndMutationMeta,
134 ) -> Callable[..., ViewAndMutationMeta]:
733 metadata = ViewAndMutationMeta(
|
| D | autograd_cache.py | 41 from .schemas import AOTConfig, ViewAndMutationMeta # noqa: F401
304 runtime_metadata: ViewAndMutationMeta
|
| /external/pytorch/torch/_functorch/ |
| D | aot_autograd.py | 96 ViewAndMutationMeta,
520 ) -> Tuple[Callable, ViewAndMutationMeta]:
533 ) -> Tuple[Callable, ViewAndMutationMeta]:
674 fw_metadata = ViewAndMutationMeta(
1434 ) -> Tuple[torch.fx.GraphModule, ViewAndMutationMeta, pytree.TreeSpec, pytree.TreeSpec]:
|
| /external/pytorch/torch/_inductor/ |
| D | freezing.py | 30 fw_metadata: torch._functorch.aot_autograd.ViewAndMutationMeta, argument
|