Path Lines of Code benchmarks/per_sample_grads.py 71 codegen/codegen_outofplacebatching.py 185 functorch/__init__.py 71 functorch/_src/aot_autograd.py 430 functorch/_src/custom_function.py 13 functorch/_src/eager_transforms.py 386 functorch/_src/make_functional.py 187 functorch/_src/operator_authoring.py 360 functorch/csrc/BatchRulesConvolution.cpp 347 functorch/csrc/BatchRulesDecompositions.cpp 214 functorch/csrc/BatchRulesLinearAlgebra.cpp 149 functorch/csrc/BatchRulesModules.cpp 324 functorch/csrc/BatchRulesReduceOps.cpp 262 functorch/csrc/BatchRulesViews.cpp 329 functorch/csrc/BatchedFallback.cpp 276 functorch/csrc/BatchedFallback.h 40 functorch/csrc/BatchedTensorImpl.cpp 124 functorch/csrc/CustomFunction.cpp 242 functorch/csrc/DynamicLayer.cpp 517 functorch/csrc/DynamicLayer.h 42 functorch/csrc/PointwiseOperatorCompileCache.cpp 584 functorch/csrc/PyTorchOperatorHacks.cpp 157 functorch/csrc/TensorWrapper.cpp 154 functorch/csrc/init.cpp 213 notebooks/_src/plot_jacobians_and_hessians.py 66