Home
last modified time | relevance | path

Searched path:functorch (Results 1 – 25 of 155) sorted by relevance

1234567

/aosp_15_r20/external/pytorch/functorch/docs/source/
H A Dfunctorch.rst
/aosp_15_r20/external/pytorch/functorch/
H A D.gitignore
H A Dwriting_batching_rules.md
H A DCOMPILE_README.md
H A DREADME.md
H A D__init__.py
H A DCMakeLists.txt
/aosp_15_r20/external/pytorch/functorch/op_analysis/
H A Dpublic_api
H A Dannotated_ops
/aosp_15_r20/external/pytorch/functorch/docs/
H A D.gitignore
H A DMakefile
/aosp_15_r20/external/pytorch/functorch/examples/
H A D.gitignore
/aosp_15_r20/external/pytorch/functorch/examples/maml_omniglot/
H A D.gitignore
/aosp_15_r20/external/pytorch/functorch/dim/
H A Dreference.py
H A Ddim.py
H A Dwrap_type.py
H A DREADME.md
/aosp_15_r20/external/pytorch/test/functorch/
H A Dtest_control_flow.py
H A Dtest_vmap_registrations.py
H A Dattn_ft.py
/aosp_15_r20/external/pytorch/functorch/benchmarks/
H A Doperator_authoring.py
/aosp_15_r20/external/pytorch/functorch/experimental/
H A Dops.py
/aosp_15_r20/external/pytorch/functorch/einops/
H A Drearrange.py
H A D_parsing.py
/aosp_15_r20/external/pytorch/functorch/notebooks/
H A Dwhirlwind_tour.ipynb

1234567