TTNN squared difference - incomplete documentation

tenstorrent / tt-metal

:metal: TT-NN operator library, and TT-Metalium low level kernel programming model.

Apache License 2.0

399 stars 49 forks source link

TTNN squared difference - incomplete documentation #8427

Open npetrovic-tenstorrent opened 4 months ago

npetrovic-tenstorrent commented 4 months ago

For TTNN squared difference op: The documentation does not cover supported data types (BFLOAT16/BFLOAT8_B) neither the supported layouts (ROW_MAJOR/TILE). In practice, its seems that it does support both BFLOAT16 and BFLOAT8_B types, but layout ROW_MAJOR is not supported (TILE supported). Ranks 2,3,4 are supported.

KalaivaniMCW commented 1 month ago

will be revisited after TTNN migration