OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

remove allow-untyped-defs from torch/jit/_passes/_property_propagation.py (#144132)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/144132
Approved by: https://github.com/Skylion007
This commit is contained in:
bobrenjc93 2025-01-03 08:31:30 -08:00 committed by PyTorch MergeBot
parent eb7a303d21
commit 6e09d32c00
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
torch/jit/_passes/_property_propagation.py
View File

@ -1,4 +1,3 @@
# mypy: allow-untyped-defs
"""
Tools to help with tensor property propagation.
@ -13,7 +12,7 @@ from torch import TensorType
from torch._C import Graph
def apply_input_props_using_example(graph: Graph, example_input: List[Any]):
def apply_input_props_using_example(graph: Graph, example_input: List[Any]) -> None:
"""
Applies properties for each tensor in the graph inputs
using the example supplied.