pytorch/torch/_C
2024-04-23 00:33:25 +00:00
..
_dynamo [dynamo] Save/restore cublas_allow_tf32 in convert_frame (#123509) 2024-04-07 03:37:47 +00:00
__init__.pyi.in preferred blas library; cublaslt gemm implementation (#122106) 2024-04-22 15:38:22 +00:00
_aoti.pyi [Inductor] Support custom op in JIT with cpp wrapper (#122554) 2024-03-26 18:48:45 +00:00
_autograd.pyi rename ort to maia (#123265) 2024-04-23 00:33:25 +00:00
_cpu.pyi
_cudnn.pyi
_distributed_autograd.pyi
_distributed_c10d.pyi [c10d/nccl-pg] allow user to pass process group description (#123472) 2024-04-12 08:44:21 +00:00
_distributed_rpc_testing.pyi
_distributed_rpc.pyi Enable import following in MYPYNOFOLLOW (now MYPYINDUCTOR) (#113830) 2023-11-17 18:24:21 +00:00
_functions.pyi
_functorch.pyi Teach dynamo about torch.func.jvp (#119926) 2024-03-22 20:25:47 +00:00
_itt.pyi
_lazy_ts_backend.pyi
_lazy.pyi
_monitor.pyi
_nn.pyi.in Enable import following in MYPYNOFOLLOW (now MYPYINDUCTOR) (#113830) 2023-11-17 18:24:21 +00:00
_nvtx.pyi
_onnx.pyi additional support for float8_e4m3fnuz and _e5m2fnuz (#115214) 2024-01-22 18:33:41 +00:00
_profiler.pyi [Profiler] Unify the device(CUDA, XPU, PrivateUse1) in torch profiler post processing (#123247) 2024-04-22 01:26:55 +00:00
_VariableFunctions.pyi.in Enable import following in MYPYNOFOLLOW (now MYPYINDUCTOR) (#113830) 2023-11-17 18:24:21 +00:00
_verbose.pyi
build.bzl
return_types.pyi.in register torch.return_types in torch.fx._pytree (#120027) 2024-02-23 21:52:42 +00:00