Name	Size	Mode	Actions
experimental/	-	0755	rm
passes/	-	0755	rm
__pycache__/	-	0755	rm
annotate.py	929	0644	editdlrm
graph.py	49098	0644	editdlrm
graph_module.py	28764	0644	editdlrm
immutable_collections.py	1312	0644	editdlrm
interpreter.py	18541	0644	editdlrm
node.py	25953	0644	editdlrm
operator_schemas.py	18169	0644	editdlrm
proxy.py	14807	0644	editdlrm
subgraph_rewriter.py	16983	0644	editdlrm
tensor_type.py	2889	0644	editdlrm
_compatibility.py	1000	0644	editdlrm
_pytree.py	1898	0644	editdlrm
_symbolic_trace.py	38082	0644	editdlrm
__init__.py	3769	0644	editdlrm

from torch.fx.proxy import Proxy from ._compatibility import compatibility @compatibility(is_backward_compatible=False) def annotate(val, type): # val could be either a regular value (not tracing) # or fx.Proxy (tracing) if isinstance(val, Proxy): if val.node.type: raise RuntimeError(f"Tried to annotate a value that already had a type on it!" f" Existing type is {val.node.type} " f"and new type is {type}. " f"This could happen if you tried to annotate a function parameter " f"value (in which case you should use the type slot " f"on the function signature) or you called " f"annotate on the same value twice") else: val.node.type = type return val else: return val