pytorch
945c93b8 - [quant][graphmode][fx] Skip observering boolean Tensors (#57375)

Commit
3 years ago
[quant][graphmode][fx] Skip observering boolean Tensors (#57375) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/57375 Skip observing the input for masked_fill. Currently we don't have a way to query the type of Proxy in GraphModule, hopefully we should have the functionality to annotate the type, we'll need to annotate a Proxy to be a boolean Tensor to remove this hack. Test Plan: python test/test_quantization.py TestQuantizeFxOps.test_boolean_tensor Imported from OSS Reviewed By: vkuzo Differential Revision: D28126003 fbshipit-source-id: 2989766370a607579b3ea07ca36cdc2ce35893cc
Author
Parents
Loading