tenstorrent / tt-metal

:metal: TT-NN operator library, and TT-Metalium low level kernel programming model.
https://docs.tenstorrent.com/ttnn/latest/index.html
Apache License 2.0
488 stars 81 forks source link

Delete convd_host_weights and update all tests using conv2d #14179

Open LPanosTT opened 1 month ago

LPanosTT commented 1 month ago

We have ops prepare_conv_weights_for_ttnn and prepare_conv_bias_for_ttnn that will pre-format convolution weights and biases for conv2d (without pushing them to device). In the previous implementation, you would pass the NHWC weights to ttnn.conv2d and ttnn.conv2d would return the formatted weights/bias on device for you to use in subsequent calls. The two aforementioned ops do this beforehand now.

TASKS

shwetankTT commented 3 weeks ago

Waiting for the merge of this PR. https://github.com/tenstorrent/tt-metal/pull/14566