pytorch/torchgen
2023-07-14 21:15:50 +00:00
..
api Revert "Add forward mode AD to out-place foreach functions (#102409)" 2023-06-30 22:46:57 +00:00
decompositions
dest [BE] Use Literal from typing (#98846) 2023-04-12 05:49:37 +00:00
executorch [torchgen] Rename executorch's RuntimeContext to KernelRuntimeContext (#104892) 2023-07-14 21:15:50 +00:00
operator_versions
selective_build [Reland][ET] Select used et_kernel_metadata only (#104005) 2023-06-23 14:38:45 +00:00
shape_functions
static_runtime sparse.mm backward: performance improvements (#94991) 2023-06-12 20:57:29 +00:00
__init__.py
BUCK.oss
BUILD.bazel
build.bzl update rules_python and let bazel install its own pip dependencies (#101405) 2023-05-23 06:20:33 +00:00
code_template.py
context.py [Specialized Kernel] Propagate Specialized Kernel Support through ComputeCodegenUnboxedKernels (#103113) 2023-06-14 10:18:16 +00:00
gen.py [Specialized Kernel] Translate Kernel Assignment Logic from function.yaml to native_functions.yaml (#102576) 2023-06-08 23:42:24 +00:00
gen_backend_stubs.py fix missing-prototypes warnings in torch_cpu (Part 6) (#101845) 2023-06-15 16:48:28 +00:00
gen_executorch.py [Reland][ET] Select used et_kernel_metadata only (#104005) 2023-06-23 14:38:45 +00:00
gen_functionalization_type.py REDO of dropout support for mem eff #102038 (#103704) 2023-06-26 23:05:03 +00:00
gen_lazy_tensor.py Excise yaml dependency in torchgen.model (#100203) 2023-04-28 13:45:39 +00:00
gen_vmap_plumbing.py
local.py
model.py Add capturable/maximize tests to Adam(W) optim configs (#104669) 2023-07-10 17:38:46 +00:00
native_function_generation.py [RFC]: Create aten native op for constrain_range (#103346) 2023-06-16 14:55:40 +00:00
utils.py Excise yaml dependency in torchgen.model (#100203) 2023-04-28 13:45:39 +00:00
yaml_utils.py [Reland] Update mypy to 1.4.1 (#105227) 2023-07-14 20:45:12 +00:00