pytorch
bbd2b1bd - [quant][graphmode][fx] Add shape to nontensor op list (#55529)

Commit
3 years ago
[quant][graphmode][fx] Add shape to nontensor op list (#55529) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/55529 x.shape outputs a non-Tensor, add this to the all_node_args_have_no_tensors function to avoid inserting observer for the getattr "shape" node. Test Plan: Imported from OSS Reviewed By: wat3rBro Differential Revision: D27628145 fbshipit-source-id: 4729294ab80c0a1e72440396d31e7e82257b1092
Author
Parents
Loading