Searched refs:FunctionalModuleWithBuffers (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/_functorch/ |
| H A D | make_functional.py | 256 class FunctionalModuleWithBuffers(nn.Module): class
289 FunctionalModuleWithBuffers(
422 ) -> Tuple[FunctionalModuleWithBuffers, Tuple[Tensor, ...], Tuple[Tensor, ...]]:
476 return FunctionalModuleWithBuffers._create_from(
493 ) -> Tuple[FunctionalModuleWithBuffers, Tuple[Tensor, ...], Tuple[Tensor, ...]]:
|
| /aosp_15_r20/external/pytorch/functorch/ |
| H A D | __init__.py | 25 FunctionalModuleWithBuffers,
|
| /aosp_15_r20/external/pytorch/functorch/notebooks/ |
| H A D | per_sample_grads.ipynb | 258 …"Let's review the changes - first, the model has become the stateless FunctionalModuleWithBuffers:"
283 "FunctionalModuleWithBuffers(\n",
|