pytorch
a0a7a2d6 - [quant][fx] store dtype, axis as literals in the graph (#54624)

Commit
3 years ago
[quant][fx] store dtype, axis as literals in the graph (#54624) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/54624 previously we were creating setattr nodes for dtype and axis. The FX convention is that primitive types are embedded as literals in args/kwargs. With this change we won't see getattr nodes in the graph anymore for dtype/axis Test Plan: python test/test_quantization.py TestQuantizeFx Imported from OSS Reviewed By: jerryzh168 Differential Revision: D27306898 fbshipit-source-id: a7c91c7cb21ee96015c7f8830b38d943ada65358
Author
Parents
Loading