TTNN demo for UNet

[dvartaniansTT]

Is your feature request related to a problem? Please describe. have a demo for UNet running on device via TTNN

Describe the solution you'd like Have a demo of UNet carvana and/or UNet Brain-segmentation running on device via TTNN. carvana and Brain MRI segmentation via UNet

[dvartaniansTT]

currently blocked by: 5857 & 6326

[dvartaniansTT]

a quick summary for the brain-mri variant here:

• we have unit tests for all required convs: conv2d brain unit test issue 5857

• we have unit test for all required maxpool_2d: maxpool2d brain UNet unit test (all pass :rocket:)

• I can't write unit tests for the concat ops yet as I'm blocked by convTraspose2D (the op before concats) to know the sharding shapes needed for the concats.

• I have filed an issue for convTranspose2d bringup: #6326

• I also added a unit test for log sigmoid in TTNN: log sigmoid TTNN unit test (pass :rocket:)

[dvartaniansTT]

currently blocked by: https://github.com/tenstorrent-metal/tt-metal/issues/5857 & https://github.com/tenstorrent-metal/tt-metal/issues/6326 once these 2 issues are resolved, I can test concats and then the model end to end.