pytorch/torchgen/api
2023-02-03 03:12:43 +00:00
..
types Revert "[Reland] Add sym_size/stride/numel/storage_offset to native_function.yaml (#91… (#92402)" 2023-02-03 03:12:43 +00:00
__init__.py
autograd.py [mta] Backward of unary foreach functions (#89591) 2023-01-23 08:28:06 +00:00
cpp.py Revert "[Reland] Add sym_size/stride/numel/storage_offset to native_function.yaml (#91… (#92402)" 2023-02-03 03:12:43 +00:00
dispatcher.py Use presence of _symint in kernel name to generate symint sig or not (#84579) 2022-09-09 18:31:56 +00:00
functionalization.py Apply ufmt linter to all py files under torchgen (#81570) 2022-07-16 03:52:25 +00:00
lazy.py More fixes for LTC symint interlock. (#86043) 2022-10-02 02:53:49 +00:00
meta.py
native.py Reland 2 Many symintifications (#87604) (#87980) 2022-10-28 13:40:11 +00:00
python.py [SDPA] Wire up FlashAttention's backward (#92917) 2023-02-02 04:02:30 +00:00
structured.py Change API type Tensor[] for structured kernels. (#73350) 2022-09-26 21:46:38 +00:00
translate.py Add macro C10_AS_INTARRAYREF_SLOW (#90675) 2022-12-16 15:10:35 -08:00
ufunc.py Back out "Revert D38984222: Don't introduce new overload for SymInt (#83628)" (#84173) 2022-08-29 18:01:07 +00:00
unboxing.py Change API type Tensor[] for structured kernels. (#73350) 2022-09-26 21:46:38 +00:00