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 80 forks source link

Add TT-NN operation to re-pad tensor with new values #13132

Open ayerofieiev-tt opened 2 months ago

ayerofieiev-tt commented 2 months ago

Is your feature request related to a problem? Please describe. Many ops assume the right padding. We want to establish a standard where ops assume padding to be junk.

Describe the solution you'd like We want to make it easier for ops to re-pad tensors with valid data

Describe alternatives you've considered none

Additional context Ideally the next step we should allow to pass an additional arg which would specify the pad info (like whether tensor is padded with -inf, inf, 0 or unknown).

ntarafdar commented 2 weeks ago

@ayerofieiev-tt could you put a priority on this?