Home
last modified time | relevance | path

Searched defs:PassType (Results 1 – 10 of 10) sorted by relevance

/aosp_15_r20/external/executorch/exir/program/
H A D_program.py181 def _transform(self, *passes: PassType) -> "ExportedProgram":
289 def transform_exported_program(ep, *passes: PassType) -> ExportedProgram:
393 def transform(self, *passes: PassType) -> "ExirExportedProgram":
963 Union[Sequence[PassType], Dict[str, Sequence[PassType]]]
1211 passes: Union[Sequence[PassType], Dict[str, Sequence[PassType]]],
/aosp_15_r20/external/executorch/exir/
H A Dpass_manager.py18 PassType: TypeAlias = Callable[[torch.fx.GraphModule], Optional[PassResult]] variable
34 passes: Optional[Union[List[PassType], List[List[PassType]]]] = None,
H A Dpass_base.py52 PassType = Callable[[torch.fx.GraphModule], Optional[PassResult]] variable
/aosp_15_r20/external/pytorch/test/distributed/fsdp/
H A Dtest_fsdp_comm.py46 class PassType(Enum): class
124 pass_type: PassType,
177 pass_type: PassType,
/aosp_15_r20/external/executorch/exir/passes/
H A Dpass_registry.py44 def wrapper(one_pass: PassType) -> PassType:
55 def register_list(cls, pass_name: str, pass_list: List[PassType]) -> None:
/aosp_15_r20/external/pytorch/torch/export/
H A Dexported_program.py85 PassType = Callable[[torch.fx.GraphModule], Optional[PassResult]] variable
1010 def _transform_do_not_use(self, *passes: PassType) -> "ExportedProgram":
H A D__init__.py70 PassType = Callable[[torch.fx.GraphModule], Optional[PassResult]] variable
/aosp_15_r20/external/executorch/backends/xnnpack/_passes/
H A D__init__.py46 passes: Optional[List[Type[PassType]]] = None,
/aosp_15_r20/external/pytorch/torch/_export/
H A Dpass_base.py31 PassType = Callable[[torch.fx.GraphModule], Optional[PassResult]] variable
/aosp_15_r20/external/executorch/backends/xnnpack/test/tester/
H A Dtester.py235 pass_list: Optional[List[Type[PassType]]] = None,