pytorch
c004346c - ns for fx: support binary ops when adding unshadowed loggers for inputs (#56408)

Commit
3 years ago
ns for fx: support binary ops when adding unshadowed loggers for inputs (#56408) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/56408 Adds the ability to log unshadowed inputs of binary ops such as `add` and `mul`, when indices 0, 1, or 0 and 1 are tensors. Note: making shadowing support this is saved for a future PR. Test Plan: ``` python test/test_quantization.py TestFXNumericSuiteCoreAPIs.test_add_mul_inputs_activations ``` Imported from OSS Reviewed By: jerryzh168 Differential Revision: D27864296 fbshipit-source-id: 3cbeb728297aa192d1ea17e815299709fd9db056
Author
Parents
Loading