@sdjordjevicTT we tested same op configuration on TTNN, and found few bugs (you can see details in linked tt-metal issues). However, we still need to map this op through MLIR in order to hit this error through compiler, rather than through manual op testing :))

Thanks for providing the context @nvukobratTT. @azecevicTT will implement this op in ttmlir.