pytorch/functorch (path) in projects: external

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

/external/pytorch/functorch/docs/source/
D	functorch.rst
/external/pytorch/functorch/
D	.gitignore
D	writing_batching_rules.md
D	CMakeLists.txt
D	README.md
D	__init__.py
D	COMPILE_README.md
/external/pytorch/functorch/examples/
D	.gitignore
/external/pytorch/functorch/op_analysis/
D	annotated_ops
D	public_api
/external/pytorch/functorch/docs/
D	.gitignore
D	Makefile
/external/pytorch/functorch/examples/maml_omniglot/
D	.gitignore
/external/pytorch/test/functorch/
D	discover_coverage.py
D	test_logging.py
D	test_memory_efficient_fusion.py
D	test_rearrange.py
/external/pytorch/functorch/benchmarks/
D	operator_authoring.py
/external/pytorch/functorch/notebooks/
D	neural_tangent_kernels.ipynb
D	whirlwind_tour.ipynb
/external/pytorch/functorch/dim/
D	tree_map.py
/external/pytorch/functorch/experimental/
D	__init__.py
D	control_flow.py
/external/pytorch/functorch/csrc/
D	init_dim_only.cpp
/external/pytorch/functorch/compile/
D	__init__.py