Home
last modified time | relevance | path

Searched refs:FunctionalModule (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/pytorch/torch/_functorch/
H A Dmake_functional.py312 class FunctionalModule(nn.Module): class
338 return FunctionalModule(model_copy, param_names, names_map), params
352 ) -> Tuple[FunctionalModule, Tuple[Tensor, ...]]:
415 return FunctionalModule._create_from(
/aosp_15_r20/external/pytorch/functorch/
H A D__init__.py24 FunctionalModule,
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Dcommon_nn.py167 class FunctionalModule(nn.Module): class
170 return FunctionalModule