Spenser Bauman 98aad40806 [tosa] Improve inferred shapes of TOSA operations ...

The TosaInferShapes pass avoids updating the shapes of tensor operators
when the consumers are not TOSA operations, limiting the efficacy of
TosaInferShapes when the IR is a mix of TOSA and other operations.
This change attempts to update the result shapes when the consumers
themselves have reasonable type/shape inference methods.

Reviewed By: eric-k256

Differential Revision: https://reviews.llvm.org/D151228

2023-05-25 16:27:13 -07:00

11 KiB

Raw Blame History

View Raw