tenstorrent / tt-metal

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

support convs with groups>1 in TTNN needed for MobilenetV2 #7544

Closed saichandax closed 1 week ago

saichandax commented 4 months ago

Describe the solution you'd like We need to add support for groups>1 for the convs in MobilenetV2.

to test, from saichand/mobilenet_v2_ops_unit_tests run: pytest tests/ttnn/unit_tests/operations/test_conv2d.py::test_mobilenetv2_conv_gs_groups

Additional context customer feature GS - e150 OS: [e.g. Ubuntu 20.04]

jliangTT commented 4 months ago

related to #4903 that is currently being worked on

keerthana-r-mcw commented 1 week ago

Resolved with new conv-api