pytorch

1c0d6ff8 - [fx][const fold] Allow to set up a function to modify const_nodes for split_const_subgraphs (#67784)

[fx][const fold] Allow to set up a function to modify const_nodes for split_const_subgraphs (#67784) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/67784 FX model generates quant/dequant layers for INT8 explicit mode support. However, if the inputs of quant/dequant layers are constant, the layer will be put into constant subgraph and optimized out. Hence TensorRT will fails to parse the left over graph. It is better to set up an optional function (skip_folding_node_fn) to skip folding nodes for split_const_subgraphs. Reviewed By: jfix71 Differential Revision: D32076970 fbshipit-source-id: 7dcbb4f02386f8c831d09a2f0e40bcdba904471c