tenstorrent / tt-metal

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

Generate pytorch traces for the `Upsample` Op #14175

Open mywoodstock opened 3 weeks ago

boris-drazic commented 1 week ago

1D input, nearest

Aten ops: arrange, add, mul, _to_copy, and _unsafe_index

Image

1D input, linear

Aten ops: upsample_linear1d

Image

boris-drazic commented 1 week ago

2D input, nearest

Aten ops: arrange, add, mul, _to_copy, unsqueeze, and _unsafe_index

Image

2D input, bilinear

Aten ops: arrange, add, mul, sub, clamp, ceil, _to_copy, rsub, unsqueeze, and _unsafe_index

Image

2D input, bicubic Aten ops: add, sub, rsub, mul, clamp, floor, _to_copy, view, arrange, _unsafe_index

Image

boris-drazic commented 1 week ago

3D input, nearest

Aten ops: arrange, unsqueeze, _to_copy, mul, add, _unsafe_index

Image

3D input, trilinear

Aten ops: upsample_trilinear3d

Image