llvm-project
729f958c - [TOSA] Add SameOperandsAndResultRank to TOSA Ops (#104501)

Commit
324 days ago
[TOSA] Add SameOperandsAndResultRank to TOSA Ops (#104501) [note: this is blocked by: https://github.com/tensorflow/tensorflow/pull/73891 otherwise tensorflow may have lit test failures] This patch adds SameOperandsAndResultRank trait to TOSA operators with ResultsBroadcastableShape trait. SameOperandsAndResultRank trait requiring that all operands and results have matching ranks unless the operand/result is unranked. This also renders the TosaMakeBroadcastable pass unnecessary - but this pass is left in for now just in case it is still used in some flows. The lit test, broadcast.mlir, is removed. This also adds verify of the SameOperandsAndResultRank trait in the TosaInferShapes pass to validate inferred shapes. Signed-off-by: Tai Ly <tai.ly@arm.com>
Author
Parents
Loading