https://github.com/MaheshRavishankar requested changes to this pull request. Thanks @shahidact for refactoring it and also adding support for broadcasting behavior. Please look at comments below. https://github.com/llvm/llvm-project/pull/104783