pytorch
a74c8df7 - [quant][fx][pt2e][be] Store node_name_to_target_dtype to node.meta["target_dtype_info"] (#92574)

Commit
1 year ago
[quant][fx][pt2e][be] Store node_name_to_target_dtype to node.meta["target_dtype_info"] (#92574) Summary: This is in preparation for quantize_pt2e API where we allow programability for users to set how they want to quantize their model Test Plan: python test/test_quantization.py TestQuantizeFx python test/test_quantization.py TestQuantizePT2E Reviewers: Subscribers: Tasks: Tags: Pull Request resolved: https://github.com/pytorch/pytorch/pull/92574 Approved by: https://github.com/jcaip
Author
Committer
Parents
Loading