pytorch
f325c29b - [fx] Make NormalizeArgs preserve node type (#85637)

Commit
3 years ago
[fx] Make NormalizeArgs preserve node type (#85637) Summary: Make `NormalizeArgs` preserve node types when transforming the graph. This bug is preventing me from scripting a graph that goes through the fx2trt `acc_tracer`. Test Plan: New unit test Reviewed By: ipiszy Differential Revision: D39753021 Pull Request resolved: https://github.com/pytorch/pytorch/pull/85637 Approved by: https://github.com/Chillee
Author
Mike Iovine
Committer
Parents
Loading