SIGN IN SIGN UP

[executorch] Ignore leading 1 dimensions when checking optimized path for op_mul (#4963)

A 1 x 1 x ... x m x n tensor can be element-wise multiplied with a m x n tensor just fine.

Pull Request resolved: https://github.com/pytorch/executorch/pull/4806

Co-authored-by: Scott Wolchok <swolchok@fb.com>
H
Hansong committed
1ae997c30334f3aeef6358f8600508150c47db29
Parent: 2553b85
Committed by GitHub <noreply@github.com> on 8/29/2024, 3:09:00 AM