file 1 file 2 # same commits # commits file 1 # commits file 2 functorch/csrc/BatchRulesViews.cpp functorch/csrc/BatchingRegistrations.cpp 20 45 41 functorch/csrc/BatchRulesUnaryOps.cpp functorch/csrc/BatchRulesBinaryOps.cpp 13 32 48 functorch/csrc/BatchRulesBinaryOps.cpp functorch/csrc/BatchRulesHelper.h 11 48 31 functorch/__init__.py functorch/_src/python_key.py 11 43 41 functorch/csrc/init.cpp functorch/csrc/DynamicLayer.cpp 9 22 36 functorch/csrc/init.cpp functorch/__init__.py 9 22 43 functorch/csrc/BatchRulesLinearAlgebra.cpp functorch/csrc/BatchRulesBinaryOps.cpp 9 33 48 functorch/csrc/DynamicLayer.h functorch/csrc/DynamicLayer.cpp 9 9 36 functorch/csrc/BatchRulesUnaryOps.cpp functorch/csrc/BatchRulesLinearAlgebra.cpp 9 32 33 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/BatchRulesUnaryOps.cpp 9 50 32 functorch/_src/decompositions.py functorch/_src/aot_autograd.py 8 17 25 functorch/_src/eager_transforms.py functorch/__init__.py 8 51 43 functorch/_src/eager_transforms.py functorch/_src/vmap.py 8 51 18 functorch/_src/make_functional.py functorch/__init__.py 8 13 43 functorch/csrc/BatchRulesViews.cpp functorch/csrc/BatchRulesHelper.h 8 45 31 functorch/csrc/BatchRulesUnaryOps.cpp functorch/csrc/BatchRulesHelper.h 8 32 31 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/BatchRulesHelper.h 8 50 31 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/BatchRulesLinearAlgebra.cpp 8 50 33 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/BatchingRegistrations.cpp 8 50 41 functorch/_src/aot_autograd.py functorch/_src/compilers.py 7 25 11 functorch/_src/python_key.py functorch/_src/compilers.py 7 41 11 functorch/_src/python_key.py functorch/_src/aot_autograd.py 7 41 25 functorch/_src/python_key.py functorch/_src/decompositions.py 7 41 17 functorch/csrc/BatchRulesLinearAlgebra.cpp functorch/csrc/BatchRulesHelper.h 7 33 31 functorch/csrc/TensorWrapper.cpp functorch/csrc/DynamicLayer.cpp 7 12 36 functorch/csrc/BatchRulesViews.cpp functorch/csrc/BatchRulesBinaryOps.cpp 7 45 48 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/BatchRulesViews.cpp 7 50 45 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/BatchRulesBinaryOps.cpp 7 50 48 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/BatchRulesHelper.h 7 10 31 functorch/_src/eager_transforms.py functorch/_src/pytree_hacks.py 6 51 7 functorch/csrc/BatchingRegistrations.cpp functorch/csrc/BatchRulesHelper.h 6 41 31 functorch/csrc/BatchRulesModules.cpp functorch/csrc/BatchRulesHelper.h 6 30 31 functorch/csrc/DynamicLayer.cpp functorch/csrc/BatchedFallback.cpp 6 36 15 functorch/csrc/DynamicLayer.cpp functorch/csrc/BatchingRegistrations.cpp 6 36 41 functorch/csrc/DynamicLayer.cpp functorch/_src/eager_transforms.py 6 36 51 functorch/csrc/DynamicLayer.cpp functorch/__init__.py 6 36 43 functorch/csrc/BatchRulesViews.cpp functorch/csrc/BatchRulesLinearAlgebra.cpp 6 45 33 functorch/csrc/BatchedTensorImpl.h functorch/csrc/BatchedTensorImpl.cpp 6 6 8 functorch/csrc/BatchRulesPooling.cpp functorch/csrc/BatchRulesHelper.h 6 13 31 functorch/_src/vmap.py functorch/_src/pytree_hacks.py 5 18 7 functorch/compile/__init__.py functorch/_src/aot_autograd.py 5 18 25 functorch/_src/python_key.py functorch/compile/__init__.py 5 41 18 functorch/csrc/BatchRulesHelper.h functorch/csrc/BatchRulesLoss.cpp 5 31 15 functorch/__init__.py functorch/_src/vmap.py 5 43 18 functorch/csrc/BatchingRegistrations.cpp functorch/_src/eager_transforms.py 5 41 51 functorch/csrc/BatchRulesModules.cpp functorch/csrc/BatchRulesBinaryOps.cpp 5 30 48 functorch/_src/make_functional.py functorch/_src/eager_transforms.py 5 13 51 functorch/csrc/DynamicLayer.cpp functorch/_src/vmap.py 5 36 18 functorch/csrc/init.cpp functorch/csrc/BatchedFallback.cpp 5 22 15 functorch/csrc/init.cpp functorch/csrc/BatchingRegistrations.cpp 5 22 41 functorch/csrc/init.cpp functorch/_src/eager_transforms.py 5 22 51 functorch/csrc/init.cpp functorch/_src/vmap.py 5 22 18 functorch/csrc/DynamicLayer.h functorch/csrc/init.cpp 5 9 22 setup.py functorch/csrc/BatchingRegistrations.cpp 5 13 41 functorch/csrc/BatchRulesViews.cpp functorch/csrc/BatchRulesLoss.cpp 5 45 15 functorch/csrc/BatchedTensorImpl.cpp functorch/csrc/BatchingRegistrations.cpp 5 8 41 functorch/csrc/BatchedTensorImpl.cpp functorch/csrc/DynamicLayer.cpp 5 8 36 functorch/csrc/BatchedTensorImpl.cpp functorch/csrc/TensorWrapper.cpp 5 8 12 functorch/csrc/BatchRulesUnaryOps.cpp functorch/csrc/BatchRulesModules.cpp 5 32 30 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/BatchRulesLoss.cpp 5 50 15 functorch/csrc/BatchedFallback.h functorch/csrc/BatchedFallback.cpp 5 8 15 functorch/csrc/TensorWrapper.h functorch/csrc/TensorWrapper.cpp 5 5 12 functorch/_src/decompositions.py functorch/_src/compilers.py 4 17 11 functorch/csrc/BatchRulesHelper.h functorch/csrc/BatchRulesNorm.cpp 4 31 10 functorch/csrc/BatchingRegistrations.cpp functorch/_src/vmap.py 4 41 18 functorch/csrc/BatchRulesModules.cpp functorch/csrc/BatchRulesDecompositions.cpp 4 30 22 functorch/csrc/DynamicLayer.cpp functorch/csrc/BatchRulesHelper.h 4 36 31 functorch/csrc/BatchingMetaprogramming.h functorch/csrc/BatchingRegistrations.cpp 4 6 41 functorch/csrc/BatchingMetaprogramming.h functorch/csrc/DynamicLayer.cpp 4 6 36 functorch/csrc/BatchRulesLinearAlgebra.cpp functorch/csrc/BatchingRegistrations.cpp 4 33 41 functorch/csrc/DynamicLayer.h functorch/_src/vmap.py 4 9 18 functorch/csrc/TensorWrapper.cpp functorch/csrc/BatchedFallback.cpp 4 12 15 functorch/csrc/TensorWrapper.cpp functorch/csrc/BatchingRegistrations.cpp 4 12 41 functorch/csrc/TensorWrapper.cpp functorch/csrc/init.cpp 4 12 22 functorch/csrc/BatchedTensorImpl.cpp functorch/csrc/BatchedFallback.cpp 4 8 15 functorch/csrc/BatchedTensorImpl.cpp functorch/csrc/init.cpp 4 8 22 functorch/csrc/BatchedTensorImpl.h functorch/csrc/BatchingRegistrations.cpp 4 6 41 functorch/csrc/BatchedTensorImpl.h functorch/csrc/TensorWrapper.cpp 4 6 12 functorch/csrc/BatchRulesUnaryOps.cpp functorch/csrc/BatchRulesViews.cpp 4 32 45 functorch/csrc/BatchRulesUnaryOps.cpp functorch/csrc/BatchingRegistrations.cpp 4 32 41 functorch/csrc/VmapTransforms.cpp functorch/csrc/BatchedFallback.cpp 4 5 15 functorch/csrc/VmapTransforms.cpp functorch/csrc/DynamicLayer.cpp 4 5 36 functorch/csrc/VmapTransforms.cpp functorch/csrc/TensorWrapper.cpp 4 5 12 functorch/csrc/VmapTransforms.cpp functorch/csrc/BatchedTensorImpl.cpp 4 5 8 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/BatchRulesViews.cpp 4 10 45 functorch/csrc/BatchedFallback.h functorch/csrc/init.cpp 4 8 22 functorch/csrc/BatchRulesNorm.cpp functorch/csrc/BatchRulesDecompositions.cpp 3 10 22 functorch/compile/__init__.py functorch/_src/compilers.py 3 18 11 functorch/_src/fx_minifier.py functorch/_src/decompositions.py 3 7 17 functorch/_src/fx_minifier.py functorch/_src/aot_autograd.py 3 7 25 functorch/csrc/BatchRulesScatterOps.cpp functorch/csrc/BatchRulesHelper.h 3 21 31 functorch/csrc/BatchRulesBinaryOps.cpp functorch/csrc/BatchRulesDecompositions.cpp 3 48 22 functorch/csrc/BatchRulesBinaryOps.cpp functorch/csrc/BatchRulesNorm.cpp 3 48 10 functorch/csrc/BatchRulesBinaryOps.cpp functorch/csrc/BatchRulesLoss.cpp 3 48 15 functorch/_src/eager_transforms.py functorch/_src/python_key.py 3 51 41 functorch/csrc/BatchingRegistrations.cpp functorch/csrc/BatchedFallback.cpp 3 41 15 functorch/csrc/BatchingRegistrations.cpp functorch/csrc/BatchRulesBinaryOps.cpp 3 41 48 functorch/csrc/BatchingRegistrations.cpp functorch/__init__.py 3 41 43 functorch/csrc/BatchRulesModules.cpp functorch/csrc/BatchRulesScatterOps.cpp 3 30 21 functorch/csrc/BatchRulesFactory.cpp functorch/csrc/BatchRulesBinaryOps.cpp 3 11 48 codegen/codegen_outofplacebatching.py functorch/csrc/BatchRulesHelper.h 3 13 31 codegen/codegen_outofplacebatching.py functorch/csrc/BatchRulesBinaryOps.cpp 3 13 48 functorch/_src/make_functional.py functorch/_src/vmap.py 3 13 18 functorch/_src/make_functional.py functorch/csrc/BatchedFallback.cpp 3 13 15 functorch/_src/make_functional.py functorch/csrc/BatchingRegistrations.cpp 3 13 41 functorch/_src/operator_authoring.py functorch/__init__.py 3 6 43 functorch/csrc/DynamicLayer.cpp functorch/csrc/BatchRulesFactory.cpp 3 36 11 functorch/csrc/DynamicLayer.cpp functorch/_src/make_functional.py 3 36 13 functorch/csrc/init.cpp functorch/csrc/CompileCache.cpp 3 22 15 functorch/csrc/init.cpp functorch/_src/make_functional.py 3 22 13 functorch/csrc/init.cpp functorch/_src/python_key.py 3 22 41 functorch/csrc/BatchingMetaprogramming.h codegen/codegen_outofplacebatching.py 3 6 13 functorch/csrc/BatchingMetaprogramming.h functorch/csrc/BatchedFallback.cpp 3 6 15 functorch/csrc/BatchingMetaprogramming.h functorch/_src/eager_transforms.py 3 6 51 functorch/csrc/BatchingMetaprogramming.h functorch/csrc/init.cpp 3 6 22 functorch/csrc/BatchRulesLinearAlgebra.cpp functorch/csrc/BatchRulesModules.cpp 3 33 30 functorch/csrc/BatchRulesLinearAlgebra.cpp functorch/csrc/BatchRulesLoss.cpp 3 33 15 functorch/csrc/DynamicLayer.h functorch/csrc/BatchedFallback.cpp 3 9 15 functorch/csrc/DynamicLayer.h functorch/_src/eager_transforms.py 3 9 51 functorch/csrc/DynamicLayer.h functorch/__init__.py 3 9 43 setup.py functorch/csrc/BatchedFallback.cpp 3 13 15 setup.py functorch/_src/eager_transforms.py 3 13 51 setup.py functorch/csrc/DynamicLayer.cpp 3 13 36 setup.py functorch/csrc/init.cpp 3 13 22 setup.py functorch/csrc/BatchingMetaprogramming.h 3 13 6 functorch/csrc/CompileCache.h functorch/csrc/CompileCache.cpp 3 4 15 functorch/csrc/CompileCache.h functorch/csrc/init.cpp 3 4 22 functorch/csrc/TensorWrapper.cpp functorch/csrc/DynamicLayer.h 3 12 9 functorch/csrc/TensorWrapper.cpp functorch/_src/make_functional.py 3 12 13 functorch/csrc/TensorWrapper.cpp functorch/__init__.py 3 12 43 functorch/csrc/TensorWrapper.cpp functorch/_src/eager_transforms.py 3 12 51 functorch/csrc/BatchRulesViews.cpp setup.py 3 45 13 functorch/csrc/BatchRulesViews.cpp functorch/csrc/DynamicLayer.cpp 3 45 36 functorch/csrc/BatchedTensorImpl.cpp functorch/_src/make_functional.py 3 8 13 functorch/csrc/BatchedTensorImpl.cpp functorch/_src/eager_transforms.py 3 8 51 functorch/csrc/BatchedTensorImpl.cpp functorch/csrc/BatchingMetaprogramming.h 3 8 6 functorch/csrc/BatchedTensorImpl.cpp setup.py 3 8 13 functorch/csrc/BatchedTensorImpl.h functorch/csrc/BatchedFallback.cpp 3 6 15 functorch/csrc/BatchedTensorImpl.h functorch/_src/make_functional.py 3 6 13 functorch/csrc/BatchedTensorImpl.h functorch/csrc/DynamicLayer.cpp 3 6 36 functorch/csrc/BatchedTensorImpl.h functorch/csrc/init.cpp 3 6 22 functorch/csrc/BatchRulesUnaryOps.cpp functorch/csrc/BatchRulesLoss.cpp 3 32 15 functorch/csrc/BatchRulesUnaryOps.cpp codegen/codegen_outofplacebatching.py 3 32 13 functorch/csrc/Constants.h functorch/csrc/BatchRulesHelper.h 3 5 31 functorch/csrc/Constants.h functorch/csrc/BatchingRegistrations.cpp 3 5 41 functorch/csrc/Constants.h functorch/csrc/DynamicLayer.cpp 3 5 36 functorch/csrc/Constants.h functorch/csrc/TensorWrapper.cpp 3 5 12 functorch/csrc/PlumbingHelper.cpp functorch/csrc/BatchingRegistrations.cpp 3 5 41 functorch/csrc/PlumbingHelper.cpp codegen/codegen_outofplacebatching.py 3 5 13 functorch/csrc/PlumbingHelper.cpp functorch/csrc/BatchRulesViews.cpp 3 5 45 functorch/csrc/PlumbingHelper.cpp functorch/csrc/BatchRulesHelper.h 3 5 31 functorch/csrc/PlumbingHelper.cpp functorch/csrc/BatchRulesBinaryOps.cpp 3 5 48 functorch/csrc/VmapTransforms.cpp functorch/csrc/BatchingRegistrations.cpp 3 5 41 functorch/csrc/VmapTransforms.cpp functorch/_src/make_functional.py 3 5 13 functorch/csrc/VmapTransforms.cpp functorch/csrc/init.cpp 3 5 22 functorch/csrc/VmapTransforms.cpp functorch/csrc/BatchedTensorImpl.h 3 5 6 benchmarks/operator_authoring.py functorch/__init__.py 3 3 43 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/BatchRulesNorm.cpp 3 50 10 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/BatchingRegistrations.cpp 3 10 41 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/BatchRulesBinaryOps.cpp 3 10 48 functorch/csrc/BatchRulesHelper.cpp codegen/codegen_outofplacebatching.py 3 10 13 functorch/csrc/PlumbingHelper.h codegen/codegen_outofplacebatching.py 3 3 13 functorch/csrc/PlumbingHelper.h functorch/csrc/PlumbingHelper.cpp 3 3 5 functorch/csrc/BatchedFallback.h functorch/csrc/BatchingRegistrations.cpp 3 8 41 functorch/csrc/TensorWrapper.h functorch/_src/eager_transforms.py 3 5 51 functorch/csrc/TensorWrapper.h functorch/csrc/DynamicLayer.cpp 3 5 36 functorch/_src/decompositions.py functorch/compile/__init__.py 2 17 18 functorch/_src/python_key.py functorch/_src/pytree_hacks.py 2 41 7 functorch/_src/python_key.py functorch/_src/vmap.py 2 41 18 functorch/csrc/CompileCache.cpp functorch/_src/aot_autograd.py 2 15 25 functorch/csrc/BatchRulesLoss.cpp functorch/csrc/BatchRulesNorm.cpp 2 15 10 functorch/csrc/PyTorchOperatorHacks.cpp functorch/csrc/BatchRulesDecompositions.cpp 2 4 22 functorch/_src/fx_minifier.py functorch/compile/__init__.py 2 7 18 functorch/_src/fx_minifier.py functorch/_src/compilers.py 2 7 11 functorch/_src/fx_minifier.py functorch/_src/python_key.py 2 7 41 functorch/csrc/BatchRulesHelper.h functorch/csrc/BatchRulesDecompositions.cpp 2 31 22 setup.cfg functorch/_src/compilers.py 2 4 11 setup.cfg functorch/_src/aot_autograd.py 2 4 25 setup.cfg functorch/_src/decompositions.py 2 4 17 setup.cfg functorch/_src/python_key.py 2 4 41 setup.cfg functorch/_src/fx_minifier.py 2 4 7 functorch/csrc/BatchedFallback.cpp functorch/_src/vmap.py 2 15 18 functorch/csrc/BatchedFallback.cpp functorch/csrc/BatchRulesHelper.h 2 15 31 functorch/csrc/BatchRulesBinaryOps.cpp functorch/csrc/BatchRulesScatterOps.cpp 2 48 21 functorch/csrc/BatchRulesBinaryOps.cpp functorch/csrc/BatchedFallback.cpp 2 48 15 functorch/__init__.py functorch/_src/pytree_hacks.py 2 43 7 functorch/__init__.py functorch/compile/__init__.py 2 43 18 functorch/__init__.py functorch/csrc/BatchRulesHelper.h 2 43 31 functorch/__init__.py functorch/csrc/BatchedFallback.cpp 2 43 15 functorch/_src/eager_transforms.py functorch/csrc/BatchedFallback.cpp 2 51 15 functorch/_src/eager_transforms.py functorch/csrc/BatchRulesBinaryOps.cpp 2 51 48 functorch/csrc/BatchingRegistrations.cpp functorch/csrc/BatchRulesLoss.cpp 2 41 15 functorch/csrc/BatchRulesFactory.cpp functorch/csrc/BatchRulesHelper.h 2 11 31 functorch/csrc/BatchRulesFactory.cpp functorch/_src/pytree_hacks.py 2 11 7 functorch/csrc/BatchRulesFactory.cpp functorch/_src/eager_transforms.py 2 11 51 codegen/codegen_outofplacebatching.py functorch/_src/pytree_hacks.py 2 13 7 codegen/codegen_outofplacebatching.py functorch/_src/vmap.py 2 13 18 codegen/codegen_outofplacebatching.py functorch/_src/python_key.py 2 13 41 codegen/codegen_outofplacebatching.py functorch/__init__.py 2 13 43 codegen/codegen_outofplacebatching.py functorch/_src/eager_transforms.py 2 13 51 codegen/codegen_outofplacebatching.py functorch/csrc/BatchingRegistrations.cpp 2 13 41 codegen/codegen_outofplacebatching.py functorch/csrc/BatchedFallback.cpp 2 13 15 codegen/gen_plumbing.py codegen/codegen_outofplacebatching.py 2 2 13 functorch/_src/make_functional.py functorch/_src/pytree_hacks.py 2 13 7 functorch/_src/make_functional.py functorch/_src/python_key.py 2 13 41 functorch/_src/make_functional.py codegen/codegen_outofplacebatching.py 2 13 13 op_analysis/gen_data.py functorch/_src/decompositions.py 2 11 17 op_analysis/gen_data.py functorch/_src/python_key.py 2 11 41 op_analysis/gen_data.py functorch/__init__.py 2 11 43 functorch/csrc/DynamicLayer.cpp functorch/_src/python_key.py 2 36 41 functorch/csrc/DynamicLayer.cpp functorch/csrc/BatchRulesBinaryOps.cpp 2 36 48 functorch/csrc/DynamicLayer.cpp codegen/codegen_outofplacebatching.py 2 36 13 functorch/csrc/init.cpp functorch/_src/operator_authoring.py 2 22 6 functorch/csrc/init.cpp functorch/csrc/BatchRulesHelper.h 2 22 31 functorch/csrc/BatchingMetaprogramming.h functorch/_src/vmap.py 2 6 18 functorch/csrc/BatchingMetaprogramming.h functorch/csrc/BatchRulesHelper.h 2 6 31 functorch/csrc/BatchingMetaprogramming.h functorch/csrc/BatchRulesBinaryOps.cpp 2 6 48 functorch/csrc/BatchingMetaprogramming.h functorch/__init__.py 2 6 43 functorch/csrc/BatchingMetaprogramming.h functorch/_src/make_functional.py 2 6 13 functorch/csrc/BatchRulesLinearAlgebra.cpp functorch/csrc/BatchRulesNorm.cpp 2 33 10 functorch/csrc/BatchRulesLinearAlgebra.cpp functorch/csrc/BatchRulesFactory.cpp 2 33 11 functorch/csrc/BatchRulesLinearAlgebra.cpp functorch/csrc/DynamicLayer.cpp 2 33 36 functorch/csrc/BatchRulesLinearAlgebra.cpp codegen/codegen_outofplacebatching.py 2 33 13 functorch/csrc/BatchRulesLinearAlgebra.cpp functorch/_src/python_key.py 2 33 41 functorch/csrc/BatchRulesLinearAlgebra.cpp functorch/__init__.py 2 33 43 functorch/csrc/BatchRulesLinearAlgebra.cpp functorch/_src/eager_transforms.py 2 33 51 functorch/csrc/DynamicLayer.h functorch/csrc/BatchingRegistrations.cpp 2 9 41 functorch/csrc/DynamicLayer.h functorch/_src/make_functional.py 2 9 13 functorch/csrc/DynamicLayer.h functorch/csrc/BatchingMetaprogramming.h 2 9 6 setup.py functorch/_src/vmap.py 2 13 18 setup.py functorch/__init__.py 2 13 43 setup.py functorch/_src/make_functional.py 2 13 13 setup.py functorch/csrc/DynamicLayer.h 2 13 9 functorch/csrc/TensorWrapper.cpp functorch/_src/python_key.py 2 12 41 functorch/csrc/TensorWrapper.cpp functorch/_src/vmap.py 2 12 18 functorch/csrc/TensorWrapper.cpp functorch/csrc/BatchRulesHelper.h 2 12 31 functorch/csrc/TensorWrapper.cpp functorch/csrc/BatchingMetaprogramming.h 2 12 6 functorch/csrc/TensorWrapper.cpp setup.py 2 12 13 functorch/csrc/BatchRulesViews.cpp functorch/csrc/BatchedFallback.cpp 2 45 15 functorch/csrc/BatchRulesViews.cpp codegen/codegen_outofplacebatching.py 2 45 13 functorch/csrc/BatchRulesViews.cpp functorch/csrc/BatchingMetaprogramming.h 2 45 6 functorch/csrc/BatchRulesViews.cpp functorch/csrc/BatchRulesModules.cpp 2 45 30 functorch/csrc/BatchRulesViews.cpp functorch/csrc/BatchRulesNorm.cpp 2 45 10 functorch/csrc/BatchRulesViews.cpp functorch/_src/eager_transforms.py 2 45 51 functorch/csrc/BatchRulesViews.cpp functorch/csrc/TensorWrapper.cpp 2 45 12 functorch/csrc/BatchedTensorImpl.cpp functorch/csrc/BatchRulesViews.cpp 2 8 45 functorch/csrc/BatchedTensorImpl.cpp functorch/_src/vmap.py 2 8 18 functorch/csrc/BatchedTensorImpl.cpp functorch/__init__.py 2 8 43 functorch/csrc/BatchedTensorImpl.cpp functorch/csrc/DynamicLayer.h 2 8 9 functorch/csrc/BatchedTensorImpl.h functorch/csrc/BatchRulesViews.cpp 2 6 45 functorch/csrc/BatchedTensorImpl.h functorch/_src/vmap.py 2 6 18 functorch/csrc/BatchedTensorImpl.h functorch/__init__.py 2 6 43 functorch/csrc/BatchedTensorImpl.h functorch/_src/eager_transforms.py 2 6 51 functorch/csrc/BatchedTensorImpl.h functorch/csrc/BatchingMetaprogramming.h 2 6 6 functorch/csrc/BatchedTensorImpl.h functorch/csrc/DynamicLayer.h 2 6 9 functorch/csrc/BatchedTensorImpl.h setup.py 2 6 13 functorch/csrc/BatchRulesUnaryOps.cpp functorch/csrc/BatchRulesNorm.cpp 2 32 10 functorch/csrc/BatchRulesUnaryOps.cpp functorch/csrc/BatchRulesFactory.cpp 2 32 11 functorch/csrc/BatchRulesPooling.cpp functorch/csrc/BatchRulesDecompositions.cpp 2 13 22 functorch/csrc/BatchRulesPooling.cpp functorch/csrc/BatchingRegistrations.cpp 2 13 41 functorch/csrc/BatchRulesPooling.cpp functorch/csrc/BatchRulesUnaryOps.cpp 2 13 32 functorch/csrc/BatchRulesPooling.cpp functorch/csrc/BatchRulesBinaryOps.cpp 2 13 48 functorch/csrc/BatchRulesPooling.cpp functorch/csrc/BatchRulesLinearAlgebra.cpp 2 13 33 functorch/csrc/BatchRulesPooling.cpp functorch/csrc/BatchRulesViews.cpp 2 13 45 functorch/experimental/__init__.py functorch/_src/eager_transforms.py 2 2 51 functorch/csrc/Constants.h functorch/_src/vmap.py 2 5 18 functorch/csrc/Constants.h functorch/csrc/BatchedFallback.cpp 2 5 15 functorch/csrc/Constants.h functorch/csrc/BatchRulesBinaryOps.cpp 2 5 48 functorch/csrc/Constants.h functorch/__init__.py 2 5 43 functorch/csrc/Constants.h functorch/_src/eager_transforms.py 2 5 51 functorch/csrc/Constants.h codegen/codegen_outofplacebatching.py 2 5 13 functorch/csrc/Constants.h functorch/_src/make_functional.py 2 5 13 functorch/csrc/Constants.h functorch/csrc/init.cpp 2 5 22 functorch/csrc/Constants.h functorch/csrc/BatchingMetaprogramming.h 2 5 6 functorch/csrc/Constants.h functorch/csrc/DynamicLayer.h 2 5 9 functorch/csrc/Constants.h setup.py 2 5 13 functorch/csrc/Constants.h functorch/csrc/BatchRulesViews.cpp 2 5 45 functorch/csrc/Constants.h functorch/csrc/BatchedTensorImpl.cpp 2 5 8 functorch/csrc/Constants.h functorch/csrc/BatchedTensorImpl.h 2 5 6 functorch/csrc/Constants.h functorch/csrc/BatchRulesUnaryOps.cpp 2 5 32 functorch/csrc/PlumbingHelper.cpp functorch/csrc/BatchedFallback.cpp 2 5 15 functorch/csrc/PlumbingHelper.cpp functorch/_src/make_functional.py 2 5 13 functorch/csrc/PlumbingHelper.cpp functorch/csrc/DynamicLayer.cpp 2 5 36 functorch/csrc/PlumbingHelper.cpp functorch/csrc/init.cpp 2 5 22 functorch/csrc/PlumbingHelper.cpp functorch/csrc/TensorWrapper.cpp 2 5 12 functorch/csrc/PlumbingHelper.cpp functorch/csrc/BatchedTensorImpl.cpp 2 5 8 functorch/csrc/PlumbingHelper.cpp functorch/csrc/BatchedTensorImpl.h 2 5 6 functorch/csrc/PlumbingHelper.cpp functorch/csrc/BatchingMetaprogramming.h 2 5 6 functorch/csrc/PlumbingHelper.cpp functorch/csrc/BatchRulesLoss.cpp 2 5 15 functorch/csrc/PlumbingHelper.cpp functorch/csrc/BatchRulesLinearAlgebra.cpp 2 5 33 functorch/csrc/PlumbingHelper.cpp functorch/csrc/BatchRulesUnaryOps.cpp 2 5 32 functorch/csrc/PlumbingHelper.cpp functorch/csrc/Constants.h 2 5 5 functorch/csrc/VmapTransforms.cpp functorch/csrc/PlumbingHelper.cpp 2 5 5 functorch/csrc/VmapTransforms.cpp functorch/_src/vmap.py 2 5 18 functorch/csrc/VmapTransforms.cpp functorch/__init__.py 2 5 43 functorch/csrc/VmapTransforms.cpp functorch/_src/eager_transforms.py 2 5 51 functorch/csrc/VmapTransforms.cpp functorch/csrc/BatchingMetaprogramming.h 2 5 6 functorch/csrc/VmapTransforms.cpp functorch/csrc/DynamicLayer.h 2 5 9 functorch/csrc/VmapTransforms.cpp setup.py 2 5 13 functorch/csrc/VmapTransforms.cpp functorch/csrc/Constants.h 2 5 5 benchmarks/operator_authoring.py functorch/_src/operator_authoring.py 2 3 6 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/BatchedFallback.cpp 2 50 15 functorch/csrc/BatchRulesReduceOps.cpp setup.py 2 50 13 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/BatchRulesPooling.cpp 2 50 13 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/BatchRulesModules.cpp 2 50 30 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/PlumbingHelper.cpp 2 50 5 functorch/csrc/BatchRulesReduceOps.cpp functorch/_src/vmap.py 2 50 18 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/BatchRulesLoss.cpp 2 10 15 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/BatchedFallback.cpp 2 10 15 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/DynamicLayer.cpp 2 10 36 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/init.cpp 2 10 22 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/BatchingMetaprogramming.h 2 10 6 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/BatchRulesUnaryOps.cpp 2 10 32 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/Constants.h 2 10 5 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/PlumbingHelper.cpp 2 10 5 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/BatchRulesReduceOps.cpp 2 10 50 functorch/csrc/PlumbingHelper.h functorch/csrc/BatchingRegistrations.cpp 2 3 41 functorch/csrc/PlumbingHelper.h functorch/csrc/BatchingMetaprogramming.h 2 3 6 functorch/csrc/PlumbingHelper.h functorch/csrc/BatchRulesViews.cpp 2 3 45 functorch/csrc/PlumbingHelper.h functorch/csrc/BatchRulesHelper.h 2 3 31 functorch/csrc/PlumbingHelper.h functorch/csrc/BatchRulesBinaryOps.cpp 2 3 48 functorch/csrc/PlumbingHelper.h functorch/csrc/BatchRulesUnaryOps.cpp 2 3 32 functorch/csrc/PlumbingHelper.h functorch/csrc/Constants.h 2 3 5 functorch/csrc/PlumbingHelper.h functorch/csrc/BatchRulesHelper.cpp 2 3 10 functorch/csrc/BatchedFallback.h functorch/_src/vmap.py 2 8 18 functorch/csrc/BatchedFallback.h functorch/csrc/BatchRulesLoss.cpp 2 8 15 functorch/csrc/BatchedFallback.h functorch/__init__.py 2 8 43 functorch/csrc/BatchedFallback.h functorch/_src/eager_transforms.py 2 8 51 functorch/csrc/BatchedFallback.h functorch/_src/make_functional.py 2 8 13 functorch/csrc/BatchedFallback.h functorch/csrc/DynamicLayer.cpp 2 8 36 functorch/csrc/BatchedFallback.h functorch/csrc/BatchingMetaprogramming.h 2 8 6 functorch/csrc/BatchedFallback.h functorch/csrc/DynamicLayer.h 2 8 9 functorch/csrc/BatchedFallback.h setup.py 2 8 13 functorch/csrc/BatchedFallback.h functorch/csrc/TensorWrapper.cpp 2 8 12 functorch/csrc/BatchedFallback.h functorch/csrc/BatchedTensorImpl.cpp 2 8 8 functorch/csrc/BatchedFallback.h functorch/csrc/BatchedTensorImpl.h 2 8 6 functorch/csrc/BatchedFallback.h functorch/csrc/Constants.h 2 8 5 functorch/csrc/BatchedFallback.h functorch/csrc/VmapTransforms.cpp 2 8 5 functorch/csrc/TensorWrapper.h functorch/_src/vmap.py 2 5 18 functorch/csrc/TensorWrapper.h functorch/csrc/BatchedFallback.cpp 2 5 15 functorch/csrc/TensorWrapper.h functorch/__init__.py 2 5 43 functorch/csrc/TensorWrapper.h functorch/csrc/BatchingRegistrations.cpp 2 5 41 functorch/csrc/TensorWrapper.h functorch/_src/make_functional.py 2 5 13 functorch/csrc/TensorWrapper.h functorch/csrc/init.cpp 2 5 22 functorch/csrc/TensorWrapper.h functorch/csrc/BatchingMetaprogramming.h 2 5 6 functorch/csrc/TensorWrapper.h functorch/csrc/DynamicLayer.h 2 5 9 functorch/csrc/TensorWrapper.h setup.py 2 5 13 functorch/csrc/TensorWrapper.h functorch/csrc/BatchedTensorImpl.cpp 2 5 8 functorch/csrc/TensorWrapper.h functorch/csrc/BatchedTensorImpl.h 2 5 6 functorch/csrc/TensorWrapper.h functorch/csrc/Constants.h 2 5 5 functorch/csrc/TensorWrapper.h functorch/csrc/VmapTransforms.cpp 2 5 5 functorch/csrc/TensorWrapper.h functorch/csrc/BatchedFallback.h 2 5 8 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/BatchRulesReduceOps.cpp 2 7 50 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/BatchRulesHelper.h 2 7 31 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/BatchRulesViews.cpp 2 7 45 functorch/csrc/VmapModeRegistrations.cpp functorch/_src/vmap.py 2 7 18 functorch/csrc/VmapModeRegistrations.cpp functorch/_src/eager_transforms.py 2 7 51 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/DynamicLayer.cpp 2 7 36 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/init.cpp 2 7 22 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/DynamicLayer.h 2 7 9 functorch/_src/__init__.py functorch/_src/vmap.py 2 2 18 functorch/_src/__init__.py functorch/csrc/BatchedFallback.cpp 2 2 15 functorch/_src/__init__.py functorch/__init__.py 2 2 43 functorch/_src/__init__.py functorch/_src/eager_transforms.py 2 2 51 functorch/_src/__init__.py functorch/csrc/BatchingRegistrations.cpp 2 2 41 functorch/_src/__init__.py functorch/_src/make_functional.py 2 2 13 functorch/_src/__init__.py functorch/csrc/DynamicLayer.cpp 2 2 36 functorch/_src/__init__.py functorch/csrc/init.cpp 2 2 22 functorch/_src/__init__.py functorch/csrc/BatchingMetaprogramming.h 2 2 6 functorch/_src/__init__.py functorch/csrc/DynamicLayer.h 2 2 9 functorch/_src/__init__.py setup.py 2 2 13 functorch/_src/__init__.py functorch/csrc/TensorWrapper.cpp 2 2 12 functorch/_src/__init__.py functorch/csrc/BatchedTensorImpl.cpp 2 2 8 functorch/_src/__init__.py functorch/csrc/BatchedTensorImpl.h 2 2 6 functorch/_src/__init__.py functorch/csrc/Constants.h 2 2 5 functorch/_src/__init__.py functorch/csrc/VmapTransforms.cpp 2 2 5 functorch/_src/__init__.py functorch/csrc/BatchedFallback.h 2 2 8 functorch/_src/__init__.py functorch/csrc/TensorWrapper.h 2 2 5 functorch/csrc/InPlacePlumbing.h functorch/csrc/BatchRulesHelper.h 2 2 31 functorch/csrc/InPlacePlumbing.h functorch/csrc/BatchRulesBinaryOps.cpp 2 2 48 functorch/csrc/InPlacePlumbing.h codegen/codegen_outofplacebatching.py 2 2 13 functorch/csrc/InPlacePlumbing.h functorch/csrc/BatchRulesUnaryOps.cpp 2 2 32 functorch/csrc/InPlacePlumbing.h functorch/csrc/Constants.h 2 2 5 functorch/csrc/InPlacePlumbing.h functorch/csrc/PlumbingHelper.cpp 2 2 5 functorch/csrc/InPlacePlumbing.h functorch/csrc/BatchRulesHelper.cpp 2 2 10 functorch/csrc/InPlacePlumbing.h functorch/csrc/PlumbingHelper.h 2 2 3 functorch/csrc/OutOfPlacePlumbing.h functorch/_src/eager_transforms.py 2 2 51 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/BatchingRegistrations.cpp 2 2 41 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/DynamicLayer.cpp 2 2 36 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/init.cpp 2 2 22 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/BatchingMetaprogramming.h 2 2 6 functorch/csrc/OutOfPlacePlumbing.h setup.py 2 2 13 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/BatchedTensorImpl.cpp 2 2 8 functorch/csrc/VmapTransforms.h functorch/_src/vmap.py 2 2 18 functorch/csrc/VmapTransforms.h functorch/csrc/BatchedFallback.cpp 2 2 15 functorch/csrc/VmapTransforms.h functorch/__init__.py 2 2 43 functorch/csrc/VmapTransforms.h functorch/_src/eager_transforms.py 2 2 51 functorch/csrc/VmapTransforms.h functorch/csrc/BatchingRegistrations.cpp 2 2 41 functorch/csrc/VmapTransforms.h functorch/_src/make_functional.py 2 2 13 functorch/csrc/VmapTransforms.h functorch/csrc/DynamicLayer.cpp 2 2 36 functorch/csrc/VmapTransforms.h functorch/csrc/init.cpp 2 2 22 functorch/csrc/VmapTransforms.h functorch/csrc/BatchingMetaprogramming.h 2 2 6 functorch/csrc/VmapTransforms.h functorch/csrc/DynamicLayer.h 2 2 9 functorch/csrc/VmapTransforms.h setup.py 2 2 13 functorch/csrc/VmapTransforms.h functorch/csrc/TensorWrapper.cpp 2 2 12 functorch/csrc/VmapTransforms.h functorch/csrc/BatchedTensorImpl.cpp 2 2 8 functorch/csrc/VmapTransforms.h functorch/csrc/BatchedTensorImpl.h 2 2 6 functorch/csrc/VmapTransforms.h functorch/csrc/Constants.h 2 2 5 functorch/csrc/VmapTransforms.h functorch/csrc/VmapTransforms.cpp 2 2 5 functorch/csrc/VmapTransforms.h functorch/csrc/BatchedFallback.h 2 2 8 functorch/csrc/VmapTransforms.h functorch/csrc/TensorWrapper.h 2 2 5 functorch/csrc/VmapTransforms.h functorch/_src/__init__.py 2 2 2 functorch/_src/pytree_hacks.py functorch/_src/compilers.py 1 7 11 functorch/_src/pytree_hacks.py functorch/_src/aot_autograd.py 1 7 25 functorch/_src/vmap.py functorch/_src/compilers.py 1 18 11 functorch/_src/vmap.py functorch/_src/aot_autograd.py 1 18 25 functorch/csrc/BatchRulesNorm.cpp functorch/_src/aot_autograd.py 1 10 25 benchmarks/transformer_fusion_patterns/__init__.py functorch/_src/compilers.py 1 1 11 benchmarks/transformer_fusion_patterns/benchmark.py functorch/_src/compilers.py 1 1 11 benchmarks/transformer_fusion_patterns/benchmark.py benchmarks/transformer_fusion_patterns/__init__.py 1 1 1 benchmarks/transformer_fusion_patterns/benchmark_helper.py functorch/_src/compilers.py 1 1 11 benchmarks/transformer_fusion_patterns/benchmark_helper.py benchmarks/transformer_fusion_patterns/__init__.py 1 1 1 benchmarks/transformer_fusion_patterns/benchmark_helper.py benchmarks/transformer_fusion_patterns/benchmark.py 1 1 1 benchmarks/transformer_fusion_patterns/bias_gelu_dropout.py functorch/_src/compilers.py 1 1 11 benchmarks/transformer_fusion_patterns/bias_gelu_dropout.py benchmarks/transformer_fusion_patterns/__init__.py 1 1 1 benchmarks/transformer_fusion_patterns/bias_gelu_dropout.py benchmarks/transformer_fusion_patterns/benchmark.py 1 1 1 benchmarks/transformer_fusion_patterns/bias_gelu_dropout.py benchmarks/transformer_fusion_patterns/benchmark_helper.py 1 1 1 functorch/compile/__init__.py benchmarks/transformer_fusion_patterns/__init__.py 1 18 1 functorch/compile/__init__.py benchmarks/transformer_fusion_patterns/benchmark.py 1 18 1 functorch/compile/__init__.py benchmarks/transformer_fusion_patterns/benchmark_helper.py 1 18 1 functorch/compile/__init__.py benchmarks/transformer_fusion_patterns/bias_gelu_dropout.py 1 18 1 functorch/_src/decompositions.py functorch/_src/pytree_hacks.py 1 17 7 functorch/_src/decompositions.py functorch/_src/vmap.py 1 17 18 functorch/csrc/CompileCache.cpp functorch/compile/__init__.py 1 15 18 functorch/csrc/BatchRulesLoss.cpp functorch/csrc/BatchRulesDecompositions.cpp 1 15 22 functorch/csrc/BatchRulesLoss.cpp functorch/_src/pytree_hacks.py 1 15 7 functorch/csrc/BatchRulesLoss.cpp functorch/_src/vmap.py 1 15 18 functorch/csrc/BatchRulesLoss.cpp functorch/_src/python_key.py 1 15 41 functorch/csrc/PyTorchOperatorHacks.cpp functorch/csrc/BatchRulesLoss.cpp 1 4 15 functorch/_src/fx_minifier.py functorch/_src/pytree_hacks.py 1 7 7 functorch/_src/fx_minifier.py functorch/_src/vmap.py 1 7 18 functorch/csrc/BatchRulesHelper.h functorch/_src/pytree_hacks.py 1 31 7 functorch/csrc/BatchRulesHelper.h functorch/_src/vmap.py 1 31 18 functorch/csrc/BatchRulesHelper.h functorch/_src/python_key.py 1 31 41 setup.cfg functorch/_src/pytree_hacks.py 1 4 7 setup.cfg functorch/_src/vmap.py 1 4 18 notebooks/_src/plot_ensembling.py setup.cfg 1 1 4 notebooks/_src/plot_jacobians_and_hessians.py setup.cfg 1 1 4 notebooks/_src/plot_jacobians_and_hessians.py notebooks/_src/plot_ensembling.py 1 1 1 notebooks/_src/plot_per_sample_gradients.py setup.cfg 1 1 4 notebooks/_src/plot_per_sample_gradients.py notebooks/_src/plot_ensembling.py 1 1 1 notebooks/_src/plot_per_sample_gradients.py notebooks/_src/plot_jacobians_and_hessians.py 1 1 1 functorch/csrc/BatchRulesScatterOps.cpp functorch/csrc/BatchRulesLoss.cpp 1 21 15 functorch/csrc/BatchedFallback.cpp functorch/csrc/BatchRulesScatterOps.cpp 1 15 21 functorch/csrc/BatchedFallback.cpp functorch/_src/pytree_hacks.py 1 15 7 functorch/csrc/BatchedFallback.cpp functorch/_src/python_key.py 1 15 41 functorch/csrc/BatchedFallback.cpp functorch/csrc/BatchRulesLoss.cpp 1 15 15 functorch/csrc/BatchRulesBinaryOps.cpp functorch/_src/pytree_hacks.py 1 48 7 functorch/csrc/BatchRulesBinaryOps.cpp functorch/_src/vmap.py 1 48 18 functorch/csrc/BatchRulesBinaryOps.cpp functorch/_src/python_key.py 1 48 41 functorch/__init__.py functorch/_src/compilers.py 1 43 11 functorch/__init__.py functorch/_src/aot_autograd.py 1 43 25 functorch/__init__.py functorch/_src/decompositions.py 1 43 17 functorch/__init__.py functorch/_src/fx_minifier.py 1 43 7 functorch/__init__.py setup.cfg 1 43 4 functorch/__init__.py functorch/csrc/CompileCache.cpp 1 43 15 functorch/__init__.py functorch/csrc/BatchRulesLoss.cpp 1 43 15 functorch/__init__.py functorch/csrc/BatchRulesBinaryOps.cpp 1 43 48 functorch/_src/eager_transforms.py functorch/_src/compilers.py 1 51 11 functorch/_src/eager_transforms.py functorch/_src/aot_autograd.py 1 51 25 functorch/_src/eager_transforms.py functorch/_src/decompositions.py 1 51 17 functorch/_src/eager_transforms.py functorch/_src/fx_minifier.py 1 51 7 functorch/_src/eager_transforms.py setup.cfg 1 51 4 functorch/_src/eager_transforms.py functorch/csrc/BatchRulesDecompositions.cpp 1 51 22 functorch/_src/eager_transforms.py functorch/csrc/BatchRulesLoss.cpp 1 51 15 functorch/_src/eager_transforms.py functorch/csrc/BatchRulesHelper.h 1 51 31 functorch/csrc/BatchingRegistrations.cpp functorch/csrc/BatchRulesScatterOps.cpp 1 41 21 functorch/csrc/BatchingRegistrations.cpp functorch/csrc/BatchRulesNorm.cpp 1 41 10 functorch/csrc/BatchingRegistrations.cpp functorch/_src/pytree_hacks.py 1 41 7 functorch/csrc/BatchingRegistrations.cpp functorch/_src/python_key.py 1 41 41 functorch/csrc/BatchRulesModules.cpp functorch/csrc/BatchRulesConvolution.cpp 1 30 2 functorch/csrc/BatchRulesModules.cpp functorch/csrc/BatchingRegistrations.cpp 1 30 41 functorch/csrc/BatchRulesModules.cpp functorch/csrc/BatchRulesNorm.cpp 1 30 10 functorch/csrc/BatchRulesModules.cpp functorch/csrc/BatchRulesLoss.cpp 1 30 15 functorch/csrc/BatchRulesFactory.cpp functorch/csrc/BatchRulesScatterOps.cpp 1 11 21 functorch/csrc/BatchRulesFactory.cpp functorch/csrc/BatchRulesModules.cpp 1 11 30 functorch/csrc/BatchRulesFactory.cpp functorch/_src/vmap.py 1 11 18 functorch/csrc/BatchRulesFactory.cpp functorch/_src/python_key.py 1 11 41 functorch/csrc/BatchRulesFactory.cpp functorch/csrc/BatchRulesLoss.cpp 1 11 15 functorch/csrc/BatchRulesFactory.cpp functorch/csrc/BatchedFallback.cpp 1 11 15 functorch/csrc/BatchRulesFactory.cpp functorch/__init__.py 1 11 43 functorch/csrc/BatchRulesFactory.cpp functorch/csrc/BatchingRegistrations.cpp 1 11 41 codegen/codegen_outofplacebatching.py functorch/_src/compilers.py 1 13 11 codegen/codegen_outofplacebatching.py functorch/_src/aot_autograd.py 1 13 25 codegen/codegen_outofplacebatching.py functorch/_src/decompositions.py 1 13 17 codegen/codegen_outofplacebatching.py functorch/_src/fx_minifier.py 1 13 7 codegen/codegen_outofplacebatching.py setup.cfg 1 13 4 codegen/codegen_outofplacebatching.py functorch/csrc/BatchRulesScatterOps.cpp 1 13 21 codegen/codegen_outofplacebatching.py functorch/csrc/BatchRulesModules.cpp 1 13 30 codegen/codegen_outofplacebatching.py functorch/csrc/BatchRulesLoss.cpp 1 13 15 codegen/codegen_outofplacebatching.py functorch/csrc/BatchRulesFactory.cpp 1 13 11 codegen/gen_plumbing.py functorch/_src/compilers.py 1 2 11 codegen/gen_plumbing.py functorch/_src/aot_autograd.py 1 2 25 codegen/gen_plumbing.py functorch/_src/pytree_hacks.py 1 2 7 codegen/gen_plumbing.py functorch/_src/vmap.py 1 2 18 codegen/gen_plumbing.py functorch/_src/decompositions.py 1 2 17 codegen/gen_plumbing.py functorch/_src/python_key.py 1 2 41 codegen/gen_plumbing.py functorch/_src/fx_minifier.py 1 2 7 codegen/gen_plumbing.py setup.cfg 1 2 4 codegen/gen_plumbing.py functorch/__init__.py 1 2 43 codegen/gen_plumbing.py functorch/_src/eager_transforms.py 1 2 51 functorch/_src/custom_function.py functorch/_src/compilers.py 1 2 11 functorch/_src/custom_function.py functorch/_src/aot_autograd.py 1 2 25 functorch/_src/custom_function.py functorch/_src/pytree_hacks.py 1 2 7 functorch/_src/custom_function.py functorch/_src/vmap.py 1 2 18 functorch/_src/custom_function.py functorch/_src/decompositions.py 1 2 17 functorch/_src/custom_function.py functorch/_src/python_key.py 1 2 41 functorch/_src/custom_function.py functorch/_src/fx_minifier.py 1 2 7 functorch/_src/custom_function.py setup.cfg 1 2 4 functorch/_src/custom_function.py functorch/__init__.py 1 2 43 functorch/_src/custom_function.py functorch/_src/eager_transforms.py 1 2 51 functorch/_src/custom_function.py codegen/codegen_outofplacebatching.py 1 2 13 functorch/_src/custom_function.py codegen/gen_plumbing.py 1 2 2 functorch/_src/make_functional.py functorch/_src/compilers.py 1 13 11 functorch/_src/make_functional.py functorch/_src/aot_autograd.py 1 13 25 functorch/_src/make_functional.py functorch/_src/decompositions.py 1 13 17 functorch/_src/make_functional.py functorch/_src/fx_minifier.py 1 13 7 functorch/_src/make_functional.py setup.cfg 1 13 4 functorch/_src/make_functional.py codegen/gen_plumbing.py 1 13 2 functorch/_src/make_functional.py functorch/_src/custom_function.py 1 13 2 functorch/_src/make_functional.py functorch/csrc/BatchRulesLoss.cpp 1 13 15 functorch/_src/make_functional.py functorch/csrc/BatchRulesHelper.h 1 13 31 functorch/_src/make_functional.py functorch/csrc/BatchRulesBinaryOps.cpp 1 13 48 functorch/_src/make_functional.py functorch/csrc/BatchRulesFactory.cpp 1 13 11 functorch/_src/operator_authoring.py functorch/_src/compilers.py 1 6 11 functorch/_src/operator_authoring.py functorch/_src/aot_autograd.py 1 6 25 functorch/_src/operator_authoring.py functorch/_src/pytree_hacks.py 1 6 7 functorch/_src/operator_authoring.py functorch/_src/vmap.py 1 6 18 functorch/_src/operator_authoring.py functorch/_src/decompositions.py 1 6 17 functorch/_src/operator_authoring.py functorch/_src/python_key.py 1 6 41 functorch/_src/operator_authoring.py functorch/_src/fx_minifier.py 1 6 7 functorch/_src/operator_authoring.py setup.cfg 1 6 4 functorch/_src/operator_authoring.py functorch/_src/eager_transforms.py 1 6 51 functorch/_src/operator_authoring.py codegen/codegen_outofplacebatching.py 1 6 13 functorch/_src/operator_authoring.py codegen/gen_plumbing.py 1 6 2 functorch/_src/operator_authoring.py functorch/_src/custom_function.py 1 6 2 functorch/_src/operator_authoring.py functorch/_src/make_functional.py 1 6 13 functorch/_src/operator_authoring.py functorch/csrc/CompileCache.cpp 1 6 15 functorch/_src/top_operators_github_usage.py functorch/_src/compilers.py 1 4 11 functorch/_src/top_operators_github_usage.py functorch/_src/aot_autograd.py 1 4 25 functorch/_src/top_operators_github_usage.py functorch/_src/pytree_hacks.py 1 4 7 functorch/_src/top_operators_github_usage.py functorch/_src/vmap.py 1 4 18 functorch/_src/top_operators_github_usage.py functorch/_src/decompositions.py 1 4 17 functorch/_src/top_operators_github_usage.py functorch/_src/python_key.py 1 4 41 functorch/_src/top_operators_github_usage.py functorch/_src/fx_minifier.py 1 4 7 functorch/_src/top_operators_github_usage.py setup.cfg 1 4 4 functorch/_src/top_operators_github_usage.py functorch/__init__.py 1 4 43 functorch/_src/top_operators_github_usage.py functorch/_src/eager_transforms.py 1 4 51 functorch/_src/top_operators_github_usage.py codegen/codegen_outofplacebatching.py 1 4 13 functorch/_src/top_operators_github_usage.py codegen/gen_plumbing.py 1 4 2 functorch/_src/top_operators_github_usage.py functorch/_src/custom_function.py 1 4 2 functorch/_src/top_operators_github_usage.py functorch/_src/make_functional.py 1 4 13 functorch/_src/top_operators_github_usage.py functorch/_src/operator_authoring.py 1 4 6 op_analysis/gen_data.py functorch/_src/compilers.py 1 11 11 op_analysis/gen_data.py functorch/_src/aot_autograd.py 1 11 25 op_analysis/gen_data.py functorch/_src/pytree_hacks.py 1 11 7 op_analysis/gen_data.py functorch/_src/vmap.py 1 11 18 op_analysis/gen_data.py functorch/_src/fx_minifier.py 1 11 7 op_analysis/gen_data.py setup.cfg 1 11 4 op_analysis/gen_data.py functorch/_src/eager_transforms.py 1 11 51 op_analysis/gen_data.py codegen/codegen_outofplacebatching.py 1 11 13 op_analysis/gen_data.py codegen/gen_plumbing.py 1 11 2 op_analysis/gen_data.py functorch/_src/custom_function.py 1 11 2 op_analysis/gen_data.py functorch/_src/make_functional.py 1 11 13 op_analysis/gen_data.py functorch/_src/operator_authoring.py 1 11 6 op_analysis/gen_data.py functorch/_src/top_operators_github_usage.py 1 11 4 op_analysis/gen_data.py functorch/compile/__init__.py 1 11 18 op_analysis/gen_data.py functorch/csrc/BatchRulesBinaryOps.cpp 1 11 48 functorch/csrc/DynamicLayer.cpp functorch/_src/pytree_hacks.py 1 36 7 functorch/csrc/DynamicLayer.cpp functorch/csrc/BatchRulesLoss.cpp 1 36 15 functorch/csrc/init.cpp functorch/_src/custom_function.py 1 22 2 functorch/csrc/init.cpp functorch/compile/__init__.py 1 22 18 functorch/csrc/init.cpp functorch/_src/pytree_hacks.py 1 22 7 functorch/csrc/init.cpp functorch/csrc/BatchRulesLoss.cpp 1 22 15 functorch/csrc/init.cpp functorch/csrc/BatchRulesBinaryOps.cpp 1 22 48 functorch/csrc/init.cpp functorch/csrc/BatchRulesFactory.cpp 1 22 11 functorch/csrc/init.cpp codegen/codegen_outofplacebatching.py 1 22 13 functorch/csrc/BatchingMetaprogramming.h functorch/_src/pytree_hacks.py 1 6 7 functorch/csrc/BatchingMetaprogramming.h functorch/_src/python_key.py 1 6 41 functorch/csrc/BatchingMetaprogramming.h functorch/csrc/BatchRulesLoss.cpp 1 6 15 functorch/csrc/BatchingMetaprogramming.h functorch/csrc/BatchRulesFactory.cpp 1 6 11 functorch/csrc/BatchRulesLinearAlgebra.cpp functorch/_src/pytree_hacks.py 1 33 7 functorch/csrc/BatchRulesLinearAlgebra.cpp functorch/_src/vmap.py 1 33 18 functorch/csrc/BatchRulesLinearAlgebra.cpp functorch/csrc/BatchedFallback.cpp 1 33 15 functorch/csrc/BatchRulesLinearAlgebra.cpp functorch/_src/make_functional.py 1 33 13 functorch/csrc/BatchRulesLinearAlgebra.cpp functorch/csrc/init.cpp 1 33 22 functorch/csrc/BatchRulesLinearAlgebra.cpp functorch/csrc/BatchingMetaprogramming.h 1 33 6 functorch/csrc/DynamicLayer.h functorch/_src/pytree_hacks.py 1 9 7 functorch/csrc/DynamicLayer.h functorch/_src/python_key.py 1 9 41 functorch/csrc/DynamicLayer.h functorch/csrc/BatchRulesLoss.cpp 1 9 15 functorch/csrc/DynamicLayer.h functorch/csrc/BatchRulesHelper.h 1 9 31 functorch/csrc/DynamicLayer.h functorch/csrc/BatchRulesBinaryOps.cpp 1 9 48 functorch/csrc/DynamicLayer.h functorch/csrc/BatchRulesFactory.cpp 1 9 11 functorch/csrc/DynamicLayer.h codegen/codegen_outofplacebatching.py 1 9 13 functorch/csrc/DynamicLayer.h functorch/csrc/BatchRulesLinearAlgebra.cpp 1 9 33 setup.py functorch/_src/pytree_hacks.py 1 13 7 setup.py functorch/_src/python_key.py 1 13 41 setup.py functorch/csrc/BatchRulesLoss.cpp 1 13 15 setup.py functorch/csrc/BatchRulesHelper.h 1 13 31 setup.py functorch/csrc/BatchRulesBinaryOps.cpp 1 13 48 setup.py functorch/csrc/BatchRulesFactory.cpp 1 13 11 setup.py codegen/codegen_outofplacebatching.py 1 13 13 setup.py functorch/csrc/BatchRulesLinearAlgebra.cpp 1 13 33 functorch/csrc/CustomFunction.cpp functorch/_src/custom_function.py 1 1 2 functorch/csrc/CustomFunction.cpp functorch/csrc/init.cpp 1 1 22 functorch/csrc/CustomFunction.h functorch/_src/custom_function.py 1 1 2 functorch/csrc/CustomFunction.h functorch/csrc/init.cpp 1 1 22 functorch/csrc/CustomFunction.h functorch/csrc/CustomFunction.cpp 1 1 1 functorch/csrc/CompileCache.h functorch/_src/aot_autograd.py 1 4 25 functorch/csrc/CompileCache.h functorch/csrc/BatchRulesNorm.cpp 1 4 10 functorch/csrc/CompileCache.h functorch/compile/__init__.py 1 4 18 functorch/csrc/CompileCache.h functorch/__init__.py 1 4 43 functorch/csrc/CompileCache.h functorch/_src/operator_authoring.py 1 4 6 functorch/csrc/TensorWrapper.cpp functorch/_src/pytree_hacks.py 1 12 7 functorch/csrc/TensorWrapper.cpp functorch/csrc/BatchRulesLoss.cpp 1 12 15 functorch/csrc/TensorWrapper.cpp functorch/csrc/BatchRulesBinaryOps.cpp 1 12 48 functorch/csrc/TensorWrapper.cpp functorch/csrc/BatchRulesFactory.cpp 1 12 11 functorch/csrc/TensorWrapper.cpp codegen/codegen_outofplacebatching.py 1 12 13 functorch/csrc/TensorWrapper.cpp functorch/csrc/BatchRulesLinearAlgebra.cpp 1 12 33 functorch/csrc/BatchRulesDynamic.cpp functorch/csrc/BatchRulesBinaryOps.cpp 1 5 48 functorch/csrc/BatchRulesDynamic.cpp functorch/csrc/BatchRulesFactory.cpp 1 5 11 functorch/csrc/BatchRulesDynamic.cpp functorch/csrc/BatchRulesDecompositions.cpp 1 5 22 functorch/csrc/BatchRulesViews.cpp functorch/csrc/BatchRulesScatterOps.cpp 1 45 21 functorch/csrc/BatchRulesViews.cpp functorch/_src/pytree_hacks.py 1 45 7 functorch/csrc/BatchRulesViews.cpp functorch/_src/vmap.py 1 45 18 functorch/csrc/BatchRulesViews.cpp functorch/_src/python_key.py 1 45 41 functorch/csrc/BatchRulesViews.cpp functorch/__init__.py 1 45 43 functorch/csrc/BatchRulesViews.cpp functorch/csrc/BatchRulesFactory.cpp 1 45 11 functorch/csrc/BatchRulesViews.cpp functorch/_src/make_functional.py 1 45 13 functorch/csrc/BatchRulesViews.cpp functorch/csrc/init.cpp 1 45 22 functorch/csrc/BatchRulesViews.cpp functorch/csrc/DynamicLayer.h 1 45 9 functorch/csrc/BatchedTensorImpl.cpp functorch/_src/pytree_hacks.py 1 8 7 functorch/csrc/BatchedTensorImpl.cpp functorch/_src/python_key.py 1 8 41 functorch/csrc/BatchedTensorImpl.cpp functorch/csrc/BatchRulesLoss.cpp 1 8 15 functorch/csrc/BatchedTensorImpl.cpp functorch/csrc/BatchRulesHelper.h 1 8 31 functorch/csrc/BatchedTensorImpl.cpp functorch/csrc/BatchRulesBinaryOps.cpp 1 8 48 functorch/csrc/BatchedTensorImpl.cpp functorch/csrc/BatchRulesFactory.cpp 1 8 11 functorch/csrc/BatchedTensorImpl.cpp codegen/codegen_outofplacebatching.py 1 8 13 functorch/csrc/BatchedTensorImpl.cpp functorch/csrc/BatchRulesLinearAlgebra.cpp 1 8 33 functorch/csrc/BatchedTensorImpl.h functorch/_src/pytree_hacks.py 1 6 7 functorch/csrc/BatchedTensorImpl.h functorch/_src/python_key.py 1 6 41 functorch/csrc/BatchedTensorImpl.h functorch/csrc/BatchRulesLoss.cpp 1 6 15 functorch/csrc/BatchedTensorImpl.h functorch/csrc/BatchRulesHelper.h 1 6 31 functorch/csrc/BatchedTensorImpl.h functorch/csrc/BatchRulesBinaryOps.cpp 1 6 48 functorch/csrc/BatchedTensorImpl.h functorch/csrc/BatchRulesFactory.cpp 1 6 11 functorch/csrc/BatchedTensorImpl.h codegen/codegen_outofplacebatching.py 1 6 13 functorch/csrc/BatchedTensorImpl.h functorch/csrc/BatchRulesLinearAlgebra.cpp 1 6 33 functorch/csrc/BatchRulesUnaryOps.cpp functorch/csrc/BatchRulesScatterOps.cpp 1 32 21 functorch/csrc/BatchRulesUnaryOps.cpp op_analysis/gen_data.py 1 32 11 functorch/csrc/BatchRulesUnaryOps.cpp functorch/_src/pytree_hacks.py 1 32 7 functorch/csrc/BatchRulesUnaryOps.cpp functorch/_src/vmap.py 1 32 18 functorch/csrc/BatchRulesUnaryOps.cpp functorch/_src/python_key.py 1 32 41 functorch/csrc/BatchRulesUnaryOps.cpp functorch/csrc/BatchedFallback.cpp 1 32 15 functorch/csrc/BatchRulesUnaryOps.cpp functorch/__init__.py 1 32 43 functorch/csrc/BatchRulesUnaryOps.cpp functorch/_src/eager_transforms.py 1 32 51 functorch/csrc/BatchRulesUnaryOps.cpp functorch/_src/make_functional.py 1 32 13 functorch/csrc/BatchRulesUnaryOps.cpp functorch/csrc/DynamicLayer.cpp 1 32 36 functorch/csrc/BatchRulesUnaryOps.cpp functorch/csrc/init.cpp 1 32 22 functorch/csrc/BatchRulesUnaryOps.cpp functorch/csrc/BatchingMetaprogramming.h 1 32 6 functorch/csrc/BatchRulesUnaryOps.cpp functorch/csrc/DynamicLayer.h 1 32 9 functorch/csrc/BatchRulesUnaryOps.cpp setup.py 1 32 13 functorch/csrc/BatchRulesUnaryOps.cpp functorch/csrc/TensorWrapper.cpp 1 32 12 functorch/csrc/BatchRulesUnaryOps.cpp functorch/csrc/BatchedTensorImpl.cpp 1 32 8 functorch/csrc/BatchRulesUnaryOps.cpp functorch/csrc/BatchedTensorImpl.h 1 32 6 functorch/csrc/BatchRulesPooling.cpp functorch/csrc/BatchRulesScatterOps.cpp 1 13 21 functorch/csrc/BatchRulesPooling.cpp functorch/csrc/BatchRulesModules.cpp 1 13 30 functorch/csrc/BatchRulesPooling.cpp functorch/csrc/BatchRulesNorm.cpp 1 13 10 functorch/csrc/BatchRulesPooling.cpp functorch/csrc/BatchRulesLoss.cpp 1 13 15 functorch/experimental/__init__.py functorch/csrc/BatchingRegistrations.cpp 1 2 41 functorch/csrc/Constants.h functorch/_src/top_operators_github_usage.py 1 5 4 functorch/csrc/Constants.h functorch/_src/pytree_hacks.py 1 5 7 functorch/csrc/Constants.h functorch/_src/python_key.py 1 5 41 functorch/csrc/Constants.h functorch/csrc/BatchRulesLoss.cpp 1 5 15 functorch/csrc/Constants.h functorch/csrc/BatchRulesFactory.cpp 1 5 11 functorch/csrc/Constants.h functorch/csrc/BatchRulesLinearAlgebra.cpp 1 5 33 functorch/csrc/PlumbingHelper.cpp functorch/csrc/BatchRulesNorm.cpp 1 5 10 functorch/csrc/PlumbingHelper.cpp functorch/csrc/BatchRulesPooling.cpp 1 5 13 functorch/csrc/PlumbingHelper.cpp functorch/_src/pytree_hacks.py 1 5 7 functorch/csrc/PlumbingHelper.cpp functorch/_src/vmap.py 1 5 18 functorch/csrc/PlumbingHelper.cpp functorch/_src/python_key.py 1 5 41 functorch/csrc/PlumbingHelper.cpp functorch/__init__.py 1 5 43 functorch/csrc/PlumbingHelper.cpp functorch/_src/eager_transforms.py 1 5 51 functorch/csrc/PlumbingHelper.cpp functorch/csrc/BatchRulesFactory.cpp 1 5 11 functorch/csrc/PlumbingHelper.cpp functorch/csrc/DynamicLayer.h 1 5 9 functorch/csrc/PlumbingHelper.cpp setup.py 1 5 13 functorch/csrc/VmapTransforms.cpp functorch/_src/pytree_hacks.py 1 5 7 functorch/csrc/VmapTransforms.cpp functorch/_src/python_key.py 1 5 41 functorch/csrc/VmapTransforms.cpp functorch/csrc/BatchRulesLoss.cpp 1 5 15 functorch/csrc/VmapTransforms.cpp functorch/csrc/BatchRulesHelper.h 1 5 31 functorch/csrc/VmapTransforms.cpp functorch/csrc/BatchRulesBinaryOps.cpp 1 5 48 functorch/csrc/VmapTransforms.cpp functorch/csrc/BatchRulesFactory.cpp 1 5 11 functorch/csrc/VmapTransforms.cpp codegen/codegen_outofplacebatching.py 1 5 13 functorch/csrc/VmapTransforms.cpp functorch/csrc/BatchRulesLinearAlgebra.cpp 1 5 33 functorch/csrc/VmapTransforms.cpp functorch/csrc/BatchRulesViews.cpp 1 5 45 functorch/csrc/VmapTransforms.cpp functorch/csrc/BatchRulesUnaryOps.cpp 1 5 32 functorch/csrc/PointwiseOperatorCompileCache.cpp functorch/_src/operator_authoring.py 1 1 6 functorch/csrc/PointwiseOperatorCompileCache.cpp functorch/csrc/init.cpp 1 1 22 functorch/csrc/PointwiseOperatorCompileCache.h functorch/_src/operator_authoring.py 1 1 6 functorch/csrc/PointwiseOperatorCompileCache.h functorch/csrc/init.cpp 1 1 22 functorch/csrc/PointwiseOperatorCompileCache.h functorch/csrc/PointwiseOperatorCompileCache.cpp 1 1 1 benchmarks/operator_authoring.py functorch/compile/__init__.py 1 3 18 benchmarks/operator_authoring.py functorch/csrc/CompileCache.cpp 1 3 15 benchmarks/operator_authoring.py functorch/csrc/init.cpp 1 3 22 benchmarks/operator_authoring.py functorch/csrc/CompileCache.h 1 3 4 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/BatchRulesDynamic.cpp 1 50 5 functorch/csrc/BatchRulesReduceOps.cpp functorch/_src/pytree_hacks.py 1 50 7 functorch/csrc/BatchRulesReduceOps.cpp functorch/_src/python_key.py 1 50 41 functorch/csrc/BatchRulesReduceOps.cpp functorch/__init__.py 1 50 43 functorch/csrc/BatchRulesReduceOps.cpp functorch/_src/eager_transforms.py 1 50 51 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/BatchRulesFactory.cpp 1 50 11 functorch/csrc/BatchRulesReduceOps.cpp codegen/codegen_outofplacebatching.py 1 50 13 functorch/csrc/BatchRulesReduceOps.cpp functorch/_src/make_functional.py 1 50 13 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/DynamicLayer.cpp 1 50 36 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/init.cpp 1 50 22 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/BatchingMetaprogramming.h 1 50 6 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/DynamicLayer.h 1 50 9 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/TensorWrapper.cpp 1 50 12 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/BatchedTensorImpl.cpp 1 50 8 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/BatchedTensorImpl.h 1 50 6 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/Constants.h 1 50 5 functorch/csrc/BatchRulesReduceOps.cpp functorch/csrc/VmapTransforms.cpp 1 50 5 functorch/csrc/BatchRulesHelper.cpp functorch/_src/pytree_hacks.py 1 10 7 functorch/csrc/BatchRulesHelper.cpp functorch/_src/vmap.py 1 10 18 functorch/csrc/BatchRulesHelper.cpp functorch/_src/python_key.py 1 10 41 functorch/csrc/BatchRulesHelper.cpp functorch/__init__.py 1 10 43 functorch/csrc/BatchRulesHelper.cpp functorch/_src/eager_transforms.py 1 10 51 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/BatchRulesFactory.cpp 1 10 11 functorch/csrc/BatchRulesHelper.cpp functorch/_src/make_functional.py 1 10 13 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/BatchRulesLinearAlgebra.cpp 1 10 33 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/DynamicLayer.h 1 10 9 functorch/csrc/BatchRulesHelper.cpp setup.py 1 10 13 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/TensorWrapper.cpp 1 10 12 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/BatchedTensorImpl.cpp 1 10 8 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/BatchedTensorImpl.h 1 10 6 functorch/csrc/BatchRulesHelper.cpp functorch/csrc/VmapTransforms.cpp 1 10 5 benchmarks/process_scorecard.py benchmarks/pointwise_scorecard.py 1 1 2 functorch/csrc/PlumbingHelper.h functorch/_src/pytree_hacks.py 1 3 7 functorch/csrc/PlumbingHelper.h functorch/_src/vmap.py 1 3 18 functorch/csrc/PlumbingHelper.h functorch/_src/python_key.py 1 3 41 functorch/csrc/PlumbingHelper.h functorch/csrc/BatchRulesLoss.cpp 1 3 15 functorch/csrc/PlumbingHelper.h functorch/csrc/BatchedFallback.cpp 1 3 15 functorch/csrc/PlumbingHelper.h functorch/__init__.py 1 3 43 functorch/csrc/PlumbingHelper.h functorch/_src/eager_transforms.py 1 3 51 functorch/csrc/PlumbingHelper.h functorch/csrc/BatchRulesFactory.cpp 1 3 11 functorch/csrc/PlumbingHelper.h functorch/_src/make_functional.py 1 3 13 functorch/csrc/PlumbingHelper.h functorch/csrc/DynamicLayer.cpp 1 3 36 functorch/csrc/PlumbingHelper.h functorch/csrc/init.cpp 1 3 22 functorch/csrc/PlumbingHelper.h functorch/csrc/BatchRulesLinearAlgebra.cpp 1 3 33 functorch/csrc/PlumbingHelper.h functorch/csrc/DynamicLayer.h 1 3 9 functorch/csrc/PlumbingHelper.h setup.py 1 3 13 functorch/csrc/PlumbingHelper.h functorch/csrc/TensorWrapper.cpp 1 3 12 functorch/csrc/PlumbingHelper.h functorch/csrc/BatchedTensorImpl.cpp 1 3 8 functorch/csrc/PlumbingHelper.h functorch/csrc/BatchedTensorImpl.h 1 3 6 functorch/csrc/PlumbingHelper.h functorch/csrc/VmapTransforms.cpp 1 3 5 functorch/csrc/PlumbingHelper.h functorch/csrc/BatchRulesReduceOps.cpp 1 3 50 functorch/csrc/BatchedFallback.h functorch/csrc/BatchRulesNorm.cpp 1 8 10 functorch/csrc/BatchedFallback.h functorch/_src/pytree_hacks.py 1 8 7 functorch/csrc/BatchedFallback.h functorch/_src/python_key.py 1 8 41 functorch/csrc/BatchedFallback.h functorch/csrc/BatchRulesHelper.h 1 8 31 functorch/csrc/BatchedFallback.h functorch/csrc/BatchRulesBinaryOps.cpp 1 8 48 functorch/csrc/BatchedFallback.h functorch/csrc/BatchRulesFactory.cpp 1 8 11 functorch/csrc/BatchedFallback.h codegen/codegen_outofplacebatching.py 1 8 13 functorch/csrc/BatchedFallback.h functorch/csrc/BatchRulesLinearAlgebra.cpp 1 8 33 functorch/csrc/BatchedFallback.h functorch/csrc/BatchRulesViews.cpp 1 8 45 functorch/csrc/BatchedFallback.h functorch/csrc/BatchRulesUnaryOps.cpp 1 8 32 functorch/csrc/BatchedFallback.h functorch/csrc/PlumbingHelper.cpp 1 8 5 functorch/csrc/BatchedFallback.h functorch/csrc/BatchRulesReduceOps.cpp 1 8 50 functorch/csrc/BatchedFallback.h functorch/csrc/BatchRulesHelper.cpp 1 8 10 functorch/csrc/BatchedFallback.h functorch/csrc/PlumbingHelper.h 1 8 3 functorch/csrc/TensorWrapper.h functorch/_src/pytree_hacks.py 1 5 7 functorch/csrc/TensorWrapper.h functorch/_src/python_key.py 1 5 41 functorch/csrc/TensorWrapper.h functorch/csrc/BatchRulesLoss.cpp 1 5 15 functorch/csrc/TensorWrapper.h functorch/csrc/BatchRulesHelper.h 1 5 31 functorch/csrc/TensorWrapper.h functorch/csrc/BatchRulesBinaryOps.cpp 1 5 48 functorch/csrc/TensorWrapper.h functorch/csrc/BatchRulesFactory.cpp 1 5 11 functorch/csrc/TensorWrapper.h codegen/codegen_outofplacebatching.py 1 5 13 functorch/csrc/TensorWrapper.h functorch/csrc/BatchRulesLinearAlgebra.cpp 1 5 33 functorch/csrc/TensorWrapper.h functorch/csrc/BatchRulesViews.cpp 1 5 45 functorch/csrc/TensorWrapper.h functorch/csrc/BatchRulesUnaryOps.cpp 1 5 32 functorch/csrc/TensorWrapper.h functorch/csrc/PlumbingHelper.cpp 1 5 5 functorch/csrc/TensorWrapper.h functorch/csrc/BatchRulesReduceOps.cpp 1 5 50 functorch/csrc/TensorWrapper.h functorch/csrc/BatchRulesHelper.cpp 1 5 10 functorch/csrc/TensorWrapper.h functorch/csrc/PlumbingHelper.h 1 5 3 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/BatchRulesDynamic.cpp 1 7 5 functorch/csrc/VmapModeRegistrations.cpp functorch/_src/pytree_hacks.py 1 7 7 functorch/csrc/VmapModeRegistrations.cpp functorch/_src/python_key.py 1 7 41 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/BatchRulesLoss.cpp 1 7 15 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/BatchedFallback.cpp 1 7 15 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/BatchRulesBinaryOps.cpp 1 7 48 functorch/csrc/VmapModeRegistrations.cpp functorch/__init__.py 1 7 43 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/BatchingRegistrations.cpp 1 7 41 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/BatchRulesFactory.cpp 1 7 11 functorch/csrc/VmapModeRegistrations.cpp codegen/codegen_outofplacebatching.py 1 7 13 functorch/csrc/VmapModeRegistrations.cpp functorch/_src/make_functional.py 1 7 13 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/BatchingMetaprogramming.h 1 7 6 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/BatchRulesLinearAlgebra.cpp 1 7 33 functorch/csrc/VmapModeRegistrations.cpp setup.py 1 7 13 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/TensorWrapper.cpp 1 7 12 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/BatchedTensorImpl.cpp 1 7 8 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/BatchedTensorImpl.h 1 7 6 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/BatchRulesUnaryOps.cpp 1 7 32 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/Constants.h 1 7 5 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/PlumbingHelper.cpp 1 7 5 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/VmapTransforms.cpp 1 7 5 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/BatchRulesHelper.cpp 1 7 10 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/PlumbingHelper.h 1 7 3 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/BatchedFallback.h 1 7 8 functorch/csrc/VmapModeRegistrations.cpp functorch/csrc/TensorWrapper.h 1 7 5 codegen/gen_functorch_lagging_op_db.py functorch/csrc/BatchRulesReduceOps.cpp 1 3 50 functorch/_src/__init__.py functorch/_src/pytree_hacks.py 1 2 7 functorch/_src/__init__.py functorch/_src/python_key.py 1 2 41 functorch/_src/__init__.py functorch/csrc/BatchRulesLoss.cpp 1 2 15 functorch/_src/__init__.py functorch/csrc/BatchRulesHelper.h 1 2 31 functorch/_src/__init__.py functorch/csrc/BatchRulesBinaryOps.cpp 1 2 48 functorch/_src/__init__.py functorch/csrc/BatchRulesFactory.cpp 1 2 11 functorch/_src/__init__.py codegen/codegen_outofplacebatching.py 1 2 13 functorch/_src/__init__.py functorch/csrc/BatchRulesLinearAlgebra.cpp 1 2 33 functorch/_src/__init__.py functorch/csrc/BatchRulesViews.cpp 1 2 45 functorch/_src/__init__.py functorch/csrc/BatchRulesUnaryOps.cpp 1 2 32 functorch/_src/__init__.py functorch/csrc/PlumbingHelper.cpp 1 2 5 functorch/_src/__init__.py functorch/csrc/BatchRulesReduceOps.cpp 1 2 50 functorch/_src/__init__.py functorch/csrc/BatchRulesHelper.cpp 1 2 10 functorch/_src/__init__.py functorch/csrc/PlumbingHelper.h 1 2 3 functorch/_src/__init__.py functorch/csrc/VmapModeRegistrations.cpp 1 2 7 functorch/csrc/InPlacePlumbing.h functorch/_src/pytree_hacks.py 1 2 7 functorch/csrc/InPlacePlumbing.h functorch/_src/vmap.py 1 2 18 functorch/csrc/InPlacePlumbing.h functorch/_src/python_key.py 1 2 41 functorch/csrc/InPlacePlumbing.h functorch/csrc/BatchRulesLoss.cpp 1 2 15 functorch/csrc/InPlacePlumbing.h functorch/csrc/BatchedFallback.cpp 1 2 15 functorch/csrc/InPlacePlumbing.h functorch/__init__.py 1 2 43 functorch/csrc/InPlacePlumbing.h functorch/_src/eager_transforms.py 1 2 51 functorch/csrc/InPlacePlumbing.h functorch/csrc/BatchingRegistrations.cpp 1 2 41 functorch/csrc/InPlacePlumbing.h functorch/csrc/BatchRulesFactory.cpp 1 2 11 functorch/csrc/InPlacePlumbing.h functorch/_src/make_functional.py 1 2 13 functorch/csrc/InPlacePlumbing.h functorch/csrc/DynamicLayer.cpp 1 2 36 functorch/csrc/InPlacePlumbing.h functorch/csrc/init.cpp 1 2 22 functorch/csrc/InPlacePlumbing.h functorch/csrc/BatchingMetaprogramming.h 1 2 6 functorch/csrc/InPlacePlumbing.h functorch/csrc/BatchRulesLinearAlgebra.cpp 1 2 33 functorch/csrc/InPlacePlumbing.h functorch/csrc/DynamicLayer.h 1 2 9 functorch/csrc/InPlacePlumbing.h setup.py 1 2 13 functorch/csrc/InPlacePlumbing.h functorch/csrc/TensorWrapper.cpp 1 2 12 functorch/csrc/InPlacePlumbing.h functorch/csrc/BatchRulesViews.cpp 1 2 45 functorch/csrc/InPlacePlumbing.h functorch/csrc/BatchedTensorImpl.cpp 1 2 8 functorch/csrc/InPlacePlumbing.h functorch/csrc/BatchedTensorImpl.h 1 2 6 functorch/csrc/InPlacePlumbing.h functorch/csrc/VmapTransforms.cpp 1 2 5 functorch/csrc/InPlacePlumbing.h functorch/csrc/BatchRulesReduceOps.cpp 1 2 50 functorch/csrc/InPlacePlumbing.h functorch/csrc/BatchedFallback.h 1 2 8 functorch/csrc/InPlacePlumbing.h functorch/csrc/TensorWrapper.h 1 2 5 functorch/csrc/InPlacePlumbing.h functorch/csrc/VmapModeRegistrations.cpp 1 2 7 functorch/csrc/InPlacePlumbing.h functorch/_src/__init__.py 1 2 2 functorch/csrc/OutOfPlacePlumbing.h functorch/_src/pytree_hacks.py 1 2 7 functorch/csrc/OutOfPlacePlumbing.h functorch/_src/vmap.py 1 2 18 functorch/csrc/OutOfPlacePlumbing.h functorch/_src/python_key.py 1 2 41 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/BatchRulesLoss.cpp 1 2 15 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/BatchRulesHelper.h 1 2 31 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/BatchedFallback.cpp 1 2 15 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/BatchRulesBinaryOps.cpp 1 2 48 functorch/csrc/OutOfPlacePlumbing.h functorch/__init__.py 1 2 43 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/BatchRulesFactory.cpp 1 2 11 functorch/csrc/OutOfPlacePlumbing.h codegen/codegen_outofplacebatching.py 1 2 13 functorch/csrc/OutOfPlacePlumbing.h functorch/_src/make_functional.py 1 2 13 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/BatchRulesLinearAlgebra.cpp 1 2 33 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/DynamicLayer.h 1 2 9 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/TensorWrapper.cpp 1 2 12 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/BatchRulesViews.cpp 1 2 45 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/BatchedTensorImpl.h 1 2 6 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/BatchRulesUnaryOps.cpp 1 2 32 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/Constants.h 1 2 5 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/PlumbingHelper.cpp 1 2 5 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/VmapTransforms.cpp 1 2 5 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/BatchRulesReduceOps.cpp 1 2 50 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/BatchRulesHelper.cpp 1 2 10 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/PlumbingHelper.h 1 2 3 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/BatchedFallback.h 1 2 8 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/TensorWrapper.h 1 2 5 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/VmapModeRegistrations.cpp 1 2 7 functorch/csrc/OutOfPlacePlumbing.h functorch/_src/__init__.py 1 2 2 functorch/csrc/OutOfPlacePlumbing.h functorch/csrc/InPlacePlumbing.h 1 2 2 functorch/csrc/VmapTransforms.h functorch/_src/pytree_hacks.py 1 2 7 functorch/csrc/VmapTransforms.h functorch/_src/python_key.py 1 2 41 functorch/csrc/VmapTransforms.h functorch/csrc/BatchRulesLoss.cpp 1 2 15 functorch/csrc/VmapTransforms.h functorch/csrc/BatchRulesHelper.h 1 2 31 functorch/csrc/VmapTransforms.h functorch/csrc/BatchRulesBinaryOps.cpp 1 2 48 functorch/csrc/VmapTransforms.h functorch/csrc/BatchRulesFactory.cpp 1 2 11 functorch/csrc/VmapTransforms.h codegen/codegen_outofplacebatching.py 1 2 13 functorch/csrc/VmapTransforms.h functorch/csrc/BatchRulesLinearAlgebra.cpp 1 2 33 functorch/csrc/VmapTransforms.h functorch/csrc/BatchRulesViews.cpp 1 2 45 functorch/csrc/VmapTransforms.h functorch/csrc/BatchRulesUnaryOps.cpp 1 2 32 functorch/csrc/VmapTransforms.h functorch/csrc/PlumbingHelper.cpp 1 2 5 functorch/csrc/VmapTransforms.h functorch/csrc/BatchRulesReduceOps.cpp 1 2 50 functorch/csrc/VmapTransforms.h functorch/csrc/BatchRulesHelper.cpp 1 2 10 functorch/csrc/VmapTransforms.h functorch/csrc/PlumbingHelper.h 1 2 3 functorch/csrc/VmapTransforms.h functorch/csrc/VmapModeRegistrations.cpp 1 2 7 functorch/csrc/VmapTransforms.h functorch/csrc/InPlacePlumbing.h 1 2 2 functorch/csrc/VmapTransforms.h functorch/csrc/OutOfPlacePlumbing.h 1 2 2