pytorch
7c182f78 - [Quant][devs] Separated implementations for quantized & non-quantized tensors for the unsqueeze function (#71648)

Commit
4 years ago
[Quant][devs] Separated implementations for quantized & non-quantized tensors for the unsqueeze function (#71648) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/71648 This PR is part of a series of PRs addressing https://github.com/pytorch/pytorch/issues/54150, related to using dispatcher for calls to quantized backends as opposed to if/else conditionals. This particular PR separates the calls to quantized & non-quantized backends for unsqueeze using a dispatcher. Test Plan: Additional testing was not implemented to test this change because test cases in the existing test suite already make use of the squeeze function for various backends. Additional testing was not implemented to test this change because test cases in the existing test suite already make use of the squeeze function for various backends. Differential Revision: D33809041 D33809041 Reviewed By: albanD, jerryzh168 Pulled By: dzdang fbshipit-source-id: 304d3311bc88e9bdc0ebc600e4da8e3e661134ad (cherry picked from commit 978604a03e95f2ec7b542fad60264b61c440e9b9)
Author
Committer
Parents
Loading