saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fixed slight bug in FX docs (#51779)

Browse source 
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/51779

Reviewed By: ngimel

Differential Revision: D26279623

Pulled By: Chillee

fbshipit-source-id: 0cd2a487ce6b80ce0d3f81e2b2334ade20d816bb
This commit is contained in:
Horace He 2021-02-05 11:23:42 -08:00 committed by Facebook GitHub Bot
parent aa1fd6b45a
commit 9c2dd5775a
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
docs/source/fx.rst
View file

@ -269,6 +269,7 @@ on them and append them to the :class:`Graph`.
"""
graph : fx.Graph = tracer_class().trace(model)
new_graph = fx.Graph()
env = {}
for node in graph.nodes:
if node.op == 'call_function' and node.target in decomposition_rules:
# By wrapping the arguments with proxies,