explicate eltwise shapes

tenstorrent · Nov 6, 2024 · b34a7b8 · b34a7b8
1 parent 0ca59f1
commit b34a7b8
Show file tree

Hide file tree

Showing 12 changed files with 1,441 additions and 1,439 deletions.
diff --git a/docs/ops/stablehlo.add.md b/docs/ops/stablehlo.add.md
diff --git a/docs/ops/stablehlo.and.md b/docs/ops/stablehlo.and.md
@@ -4,6 +4,6 @@
 
 ||STABLE HLO Input Variations|ttnn op|Torch Name|Status|
 | :--- | :--- | :--- | :--- | :--- |
-|0|Tensor<[19]>,<br>|ttnn.and|aten::logical_and|5|
-|1|Tensor<[197]>,<br>|ttnn.and|aten::logical_and|5|
+|0|Tensor<[19]>,<br>Tensor<[19]>,<br>|ttnn.and|aten::logical_and|5|
+|1|Tensor<[197]>,<br>Tensor<[197]>,<br>|ttnn.and|aten::logical_and|5|
 
diff --git a/docs/ops/stablehlo.clamp.md b/docs/ops/stablehlo.clamp.md
@@ -4,23 +4,23 @@
 
 ||STABLE HLO Input Variations|ttnn op|Torch Name|Status|
 | :--- | :--- | :--- | :--- | :--- |
-|0|Tensor<[1,1024,512]>,<br>|ttnn.clamp|aten::gelu|4|
-|1|Tensor<[1,256,256]>,<br>|ttnn.clamp|aten::gelu|4|
-|2|Tensor<[1,10,3072]>,<br>|ttnn.clamp|aten::gelu|4|
-|3|Tensor<[1,10,768]>,<br>|ttnn.clamp|aten::gelu|4|
-|4|Tensor<[1,4096,1280]>,<br>|ttnn.clamp|aten::gelu|4|
-|5|Tensor<[1,1024,2560]>,<br>|ttnn.clamp|aten::gelu|4|
-|6|Tensor<[1,256,5120]>,<br>|ttnn.clamp|aten::gelu|4|
-|7|Tensor<[1,64,5120]>,<br>|ttnn.clamp|aten::gelu|4|
-|8|Tensor<[1,25,3072]>,<br>|ttnn.clamp|aten::gelu|4|
-|9|Tensor<[1,1445,768]>,<br>|ttnn.clamp|aten::gelu|4|
-|10|Tensor<[1,3072,8]>,<br>|ttnn.clamp|aten::gelu|4|
-|11|Tensor<[1,256,1280]>,<br>|ttnn.clamp|aten::gelu|4|
-|12|Tensor<[1,2048,768]>,<br>|ttnn.clamp|aten::gelu|4|
-|13|Tensor<[1,201,3072]>,<br>|ttnn.clamp|aten::gelu|4|
-|14|Tensor<[1,1536]>,<br>|ttnn.clamp|aten::gelu|4|
-|15|Tensor<[1,19,4096]>,<br>|ttnn.clamp|aten::gelu|4|
-|16|Tensor<[1,16,3072]>,<br>|ttnn.clamp|aten::gelu|4|
+|0|Tensor<[1,1024,512]>,<br>Tensor<[1,1024,512]>,<br>|ttnn.clamp|aten::gelu|4|
+|1|Tensor<[1,256,256]>,<br>Tensor<[1,256,256]>,<br>|ttnn.clamp|aten::gelu|4|
+|2|Tensor<[1,10,3072]>,<br>Tensor<[1,10,3072]>,<br>|ttnn.clamp|aten::gelu|4|
+|3|Tensor<[1,10,768]>,<br>Tensor<[1,10,768]>,<br>|ttnn.clamp|aten::gelu|4|
+|4|Tensor<[1,4096,1280]>,<br>Tensor<[1,4096,1280]>,<br>|ttnn.clamp|aten::gelu|4|
+|5|Tensor<[1,1024,2560]>,<br>Tensor<[1,1024,2560]>,<br>|ttnn.clamp|aten::gelu|4|
+|6|Tensor<[1,256,5120]>,<br>Tensor<[1,256,5120]>,<br>|ttnn.clamp|aten::gelu|4|
+|7|Tensor<[1,64,5120]>,<br>Tensor<[1,64,5120]>,<br>|ttnn.clamp|aten::gelu|4|
+|8|Tensor<[1,25,3072]>,<br>Tensor<[1,25,3072]>,<br>|ttnn.clamp|aten::gelu|4|
+|9|Tensor<[1,1445,768]>,<br>Tensor<[1,1445,768]>,<br>|ttnn.clamp|aten::gelu|4|
+|10|Tensor<[1,3072,8]>,<br>Tensor<[1,3072,8]>,<br>|ttnn.clamp|aten::gelu|4|
+|11|Tensor<[1,256,1280]>,<br>Tensor<[1,256,1280]>,<br>|ttnn.clamp|aten::gelu|4|
+|12|Tensor<[1,2048,768]>,<br>Tensor<[1,2048,768]>,<br>|ttnn.clamp|aten::gelu|4|
+|13|Tensor<[1,201,3072]>,<br>Tensor<[1,201,3072]>,<br>|ttnn.clamp|aten::gelu|4|
+|14|Tensor<[1,1536]>,<br>Tensor<[1,1536]>,<br>|ttnn.clamp|aten::gelu|4|
+|15|Tensor<[1,19,4096]>,<br>Tensor<[1,19,4096]>,<br>|ttnn.clamp|aten::gelu|4|
+|16|Tensor<[1,16,3072]>,<br>Tensor<[1,16,3072]>,<br>|ttnn.clamp|aten::gelu|4|
 |17|Scalar,<br>Tensor<[30]>,<br>Scalar,<br>|ttnn.clamp|aten::clamp|4|
 |18|Scalar,<br>Tensor<[30,1]>,<br>Scalar,<br>|ttnn.clamp|aten::clamp|4|
 |19|Scalar,<br>Tensor<[40]>,<br>Scalar,<br>|ttnn.clamp|aten::clamp|4|
@@ -36,19 +36,19 @@
 |29|Scalar,<br>Tensor<[480]>,<br>Scalar,<br>|ttnn.clamp|aten::clamp|4|
 |30|Scalar,<br>Tensor<[480,1]>,<br>Scalar,<br>|ttnn.clamp|aten::clamp|4|
 |31|Scalar,<br>Tensor<[640]>,<br>Scalar,<br>|ttnn.clamp|aten::clamp|4|
-|32|Tensor<[1,19200,256]>,<br>|ttnn.clamp|aten::gelu|4|
-|33|Tensor<[1,4800,512]>,<br>|ttnn.clamp|aten::gelu|4|
-|34|Tensor<[1,1200,1280]>,<br>|ttnn.clamp|aten::gelu|4|
-|35|Tensor<[1,300,2048]>,<br>|ttnn.clamp|aten::gelu|4|
-|36|Tensor<[1,197,3072]>,<br>|ttnn.clamp|aten::gelu|4|
+|32|Tensor<[1,19200,256]>,<br>Tensor<[1,19200,256]>,<br>|ttnn.clamp|aten::gelu|4|
+|33|Tensor<[1,4800,512]>,<br>Tensor<[1,4800,512]>,<br>|ttnn.clamp|aten::gelu|4|
+|34|Tensor<[1,1200,1280]>,<br>Tensor<[1,1200,1280]>,<br>|ttnn.clamp|aten::gelu|4|
+|35|Tensor<[1,300,2048]>,<br>Tensor<[1,300,2048]>,<br>|ttnn.clamp|aten::gelu|4|
+|36|Tensor<[1,197,3072]>,<br>Tensor<[1,197,3072]>,<br>|ttnn.clamp|aten::gelu|4|
 |37|Scalar,<br>Tensor<[128]>,<br>Scalar,<br>|ttnn.clamp|aten::clamp|4|
 |38|Scalar,<br>Tensor<[128,1]>,<br>Scalar,<br>|ttnn.clamp|aten::clamp|4|
-|39|Tensor<[1,16384,128]>,<br>|ttnn.clamp|aten::gelu|4|
-|40|Tensor<[1,4096,256]>,<br>|ttnn.clamp|aten::gelu|4|
-|41|Tensor<[1,1024,640]>,<br>|ttnn.clamp|aten::gelu|4|
-|42|Tensor<[1,256,1024]>,<br>|ttnn.clamp|aten::gelu|4|
-|43|Tensor<[1,7,18176]>,<br>|ttnn.clamp|aten::gelu|4|
+|39|Tensor<[1,16384,128]>,<br>Tensor<[1,16384,128]>,<br>|ttnn.clamp|aten::gelu|4|
+|40|Tensor<[1,4096,256]>,<br>Tensor<[1,4096,256]>,<br>|ttnn.clamp|aten::gelu|4|
+|41|Tensor<[1,1024,640]>,<br>Tensor<[1,1024,640]>,<br>|ttnn.clamp|aten::gelu|4|
+|42|Tensor<[1,256,1024]>,<br>Tensor<[1,256,1024]>,<br>|ttnn.clamp|aten::gelu|4|
+|43|Tensor<[1,7,18176]>,<br>Tensor<[1,7,18176]>,<br>|ttnn.clamp|aten::gelu|4|
 |44|Scalar,<br>Tensor<[27]>,<br>Scalar,<br>|ttnn.clamp|aten::clamp|4|
 |45|Scalar,<br>Tensor<[27,1]>,<br>Scalar,<br>|ttnn.clamp|aten::clamp|4|
-|46|Tensor<[1,197,4096]>,<br>|ttnn.clamp|aten::gelu|4|
+|46|Tensor<[1,197,4096]>,<br>Tensor<[1,197,4096]>,<br>|ttnn.clamp|aten::gelu|4|