pytorch
821a9759 - fx quant: improve performance of all_node_args_have_no_tensors (#58461)

Commit
3 years ago
fx quant: improve performance of all_node_args_have_no_tensors (#58461) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/58461 Improves the logic which calculates whether a node has any tensors in its arguments by terminating the recursion early when possible. In a future PR, we should probably ditch this entire approach and switch to using dtype propagation. Test Plan: ``` python test/test_quantization.py TestQuantizeFx python test/test_quantization.py TestQuantizeFxOps ``` Imported from OSS Reviewed By: jerryzh168 Differential Revision: D28499455 fbshipit-source-id: bedd844022b90e1fcb7d7a3cb4cc65440dc9cc59
Author
Parents
Loading