Home
last modified time | relevance | path

Searched refs:substitute_in_graph (Results 1 – 6 of 6) sorted by relevance

/external/pytorch/torch/_dynamo/polyfills/
Ditertools.py10 from ..decorators import substitute_in_graph
25 @substitute_in_graph(itertools.chain, is_embedded_type=True) # type: ignore[arg-type]
31 @substitute_in_graph(itertools.chain.from_iterable) # type: ignore[arg-type]
40 @substitute_in_graph(itertools.islice, is_embedded_type=True) # type: ignore[arg-type]
69 @substitute_in_graph(itertools.tee)
Dbuiltins.py10 from ..decorators import substitute_in_graph
23 @substitute_in_graph(builtins.all, can_constant_fold_through=True)
31 @substitute_in_graph(builtins.any, can_constant_fold_through=True)
39 @substitute_in_graph(builtins.enumerate, is_embedded_type=True) # type: ignore[arg-type]
Dos.py10 from ..decorators import substitute_in_graph
17 @substitute_in_graph(os.fspath, can_constant_fold_through=True)
/external/pytorch/torch/compiler/
D__init__.py123 def substitute_in_graph( function
179 return torch._dynamo.substitute_in_graph(
/external/pytorch/docs/source/
Dtorch.compiler_api.rst19 substitute_in_graph
/external/pytorch/test/dynamo/
Dtest_decorators.py269 @torch._dynamo.substitute_in_graph(operator.indexOf)
276 @torch._dynamo.substitute_in_graph(operator.indexOf)