mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
c855f8632e
Adds suppressions to pyrefly will typecheck clean: https://github.com/pytorch/pytorch/issues/163283 Almost there! Test plan: dmypy restart && python3 scripts/lintrunner.py -a pyrefly check step 1: delete lines in the pyrefly.toml file from the project-excludes field step 2: run pyrefly check step 3: add suppressions, clean up unused suppressions before: https://gist.github.com/maggiemoss/4b3bf2037014e116bc00706a16aef199 after: INFO 0 errors (6,884 ignored) Pull Request resolved: https://github.com/pytorch/pytorch/pull/164913 Approved by: https://github.com/oulgen
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
from . import parametrizations, parametrize, rnn, stateless
|
|
from .clip_grad import ( # pyrefly: ignore # deprecated
|
|
_clip_grads_with_norm_ as clip_grads_with_norm_,
|
|
_get_total_norm as get_total_norm,
|
|
clip_grad_norm,
|
|
clip_grad_norm_,
|
|
clip_grad_value_,
|
|
)
|
|
from .convert_parameters import parameters_to_vector, vector_to_parameters
|
|
from .fusion import (
|
|
fuse_conv_bn_eval,
|
|
fuse_conv_bn_weights,
|
|
fuse_linear_bn_eval,
|
|
fuse_linear_bn_weights,
|
|
)
|
|
from .init import skip_init
|
|
from .memory_format import (
|
|
convert_conv2d_weight_memory_format,
|
|
convert_conv3d_weight_memory_format,
|
|
)
|
|
from .spectral_norm import remove_spectral_norm, spectral_norm
|
|
from .weight_norm import remove_weight_norm, weight_norm
|
|
|
|
|
|
__all__ = [
|
|
"clip_grad_norm",
|
|
"clip_grad_norm_",
|
|
"clip_grads_with_norm_",
|
|
"clip_grad_value_",
|
|
"convert_conv2d_weight_memory_format",
|
|
"convert_conv3d_weight_memory_format",
|
|
"fuse_conv_bn_eval",
|
|
"fuse_conv_bn_weights",
|
|
"fuse_linear_bn_eval",
|
|
"fuse_linear_bn_weights",
|
|
"get_total_norm",
|
|
"parameters_to_vector",
|
|
"parametrizations",
|
|
"parametrize",
|
|
"remove_spectral_norm",
|
|
"remove_weight_norm",
|
|
"rnn",
|
|
"skip_init",
|
|
"spectral_norm",
|
|
"stateless",
|
|
"vector_to_parameters",
|
|
"weight_norm",
|
|
]
|