pytorch
7884c2bb - [ONNX] Add Concat to Scalar type analysis JIT pass (#69227) (#69548)

Commit
2 years ago
[ONNX] Add Concat to Scalar type analysis JIT pass (#69227) (#69548) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/69548 * Add Concat to Scalar type analysis pass By using scalar type analysis for Concat, the exported model can do automatic type promotion for Concat nodes, including mixed fp16 and fp32 inputs, for example. Unit tests based on the original PR https://github.com/pytorch/pytorch/pull/24378/ * Fix UTs Test Plan: Imported from OSS Reviewed By: msaroufim Differential Revision: D32994268 Pulled By: malfet fbshipit-source-id: 0deab88b0bb1e396770690af27730accb64fcf63 (cherry picked from commit a99322cadf7b79a4548266a9d4d3af094b89bac4)
Author
Committer
Parents
Loading