search

tenstorrent / tt-metal

:metal: TT-NN operator library, and TT-Metalium low level kernel programming model.
Apache License 2.0
395 stars 48 forks source link

YoloV5 Ops Unit tests #6867

Open saichandax opened 5 months ago

saichandax commented 5 months ago

To create unittests for all the ops in the YoloV5 model on GS device. unit tests to be created under https://github.com/tenstorrent-metal/tt-metal/blob/main/tests/ttnn/unit_tests/operations

saichandax commented 5 months ago

Commit for the unit tests is here

saichandax commented 5 months ago

PR: #6898 (Need to clean code). @ankitmcw