diff --git a/docs/ops/stablehlo/arith.constant.md b/docs/ops/stablehlo/arith.constant.md new file mode 100644 index 00000000..11231e22 --- /dev/null +++ b/docs/ops/stablehlo/arith.constant.md @@ -0,0 +1,7 @@ + +### arith.constant + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1]>,
||aten::_safe_softmax|4| diff --git a/docs/ops/stablehlo/stablehlo.add.md b/docs/ops/stablehlo/stablehlo.add.md new file mode 100644 index 00000000..c2d43a9d --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.add.md @@ -0,0 +1,458 @@ + +### stablehlo.add::ttnn.add + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[256,256]>,
Tensor<[256,256]>,
|ttnn.add|aten::add.Tensor|6| +|1|Tensor<[1,1,32,32]>,
Tensor<[1,1,32,32]>,
|ttnn.add|aten::add.Tensor|4| +|2|Tensor<[1,32,1]>,
Tensor<[1,32,1]>,
|ttnn.add|aten::add.Tensor|4| +|3|Tensor<[1,32,32,128]>,
Tensor<[1,32,32,128]>,
|ttnn.add|aten::add.Tensor|5| +|4|Tensor<[1,32,32,32]>,
Tensor<[1,32,32,32]>,
|ttnn.add|aten::add.Tensor|4| +|5|Tensor<[1,32,4096]>,
Tensor<[1,32,4096]>,
|ttnn.add|aten::add.Tensor|5| +|6|Tensor<[32]>,
Tensor<[32]>,
|ttnn.add|aten::arange|4| +|7|Tensor<[32,1]>,
Tensor<[32,1]>,
|ttnn.add|aten::triu|4| +|8|Tensor<[1,7,768]>,
Tensor<[1,7,768]>,
|ttnn.add|aten::add.Tensor|5| +|9|Tensor<[7]>,
Tensor<[7]>,
|ttnn.add|aten::add.Tensor|4| +|10|Tensor<[1,7,1]>,
Tensor<[1,7,1]>,
|ttnn.add|aten::add.Tensor|4| +|11|Tensor<[7,2304]>,
Tensor<[7,2304]>,
|ttnn.add|aten::add.Tensor|4| +|12|Tensor<[1,12,7,7]>,
Tensor<[1,12,7,7]>,
|ttnn.add|aten::add.Tensor|4| +|13|Tensor<[7,768]>,
Tensor<[7,768]>,
|ttnn.add|aten::add.Tensor|4| +|14|Tensor<[7,3072]>,
Tensor<[7,3072]>,
|ttnn.add|aten::add.Tensor|4| +|15|Tensor<[1,7,3072]>,
Tensor<[1,7,3072]>,
|ttnn.add|aten::add.Tensor|5| +|16|Tensor<[1]>,
Tensor<[1]>,
|ttnn.add|aten::arange|4| +|17|Tensor<[1,32,112,112]>,
Tensor<[1,32,112,112]>,
|ttnn.add|aten::add.Tensor|4| +|18|Tensor<[64]>,
Tensor<[64]>,
|ttnn.add|aten::add.Tensor|4| +|19|Tensor<[1,64,112,112]>,
Tensor<[1,64,112,112]>,
|ttnn.add|aten::add.Tensor|4| +|20|Tensor<[1,64,56,56]>,
Tensor<[1,64,56,56]>,
|ttnn.add|aten::add.Tensor|4| +|21|Tensor<[128]>,
Tensor<[128]>,
|ttnn.add|aten::add.Tensor|4| +|22|Tensor<[1,128,56,56]>,
Tensor<[1,128,56,56]>,
|ttnn.add|aten::add.Tensor|4| +|23|Tensor<[1,128,28,28]>,
Tensor<[1,128,28,28]>,
|ttnn.add|aten::add.Tensor|4| +|24|Tensor<[256]>,
Tensor<[256]>,
|ttnn.add|aten::add.Tensor|4| +|25|Tensor<[1,256,28,28]>,
Tensor<[1,256,28,28]>,
|ttnn.add|aten::add.Tensor|4| +|26|Tensor<[512]>,
Tensor<[512]>,
|ttnn.add|aten::add.Tensor|4| +|27|Tensor<[1,512,28,28]>,
Tensor<[1,512,28,28]>,
|ttnn.add|aten::add.Tensor|4| +|28|Tensor<[1,19,28,28]>,
Tensor<[1,19,28,28]>,
|ttnn.add|aten::convolution|4| +|29|Tensor<[1,38,28,28]>,
Tensor<[1,38,28,28]>,
|ttnn.add|aten::convolution|4| +|30|Tensor<[256,512]>,
Tensor<[256,512]>,
|ttnn.add|aten::add.Tensor|4| +|31|Tensor<[1,256,1]>,
Tensor<[1,256,1]>,
|ttnn.add|aten::add.Tensor|4| +|32|Tensor<[1,256,512]>,
Tensor<[1,256,512]>,
|ttnn.add|aten::add.Tensor|4| +|33|Tensor<[1,1000]>,
Tensor<[1,1000]>,
|ttnn.add|aten::add.Tensor|4| +|34|Tensor<[1,1024,512]>,
Tensor<[1,1024,512]>,
|ttnn.add|aten::convolution|4| +|35|Tensor<[1,256,256]>,
Tensor<[1,256,256]>,
|ttnn.add|aten::gelu|4| +|36|Tensor<[1,64,1,1]>,
Tensor<[1,64,1,1]>,
|ttnn.add|aten::add.Tensor|4| +|37|Tensor<[1,64,360,640]>,
Tensor<[1,64,360,640]>,
|ttnn.add|aten::add.Tensor|4| +|38|Tensor<[1,64,180,320]>,
Tensor<[1,64,180,320]>,
|ttnn.add|aten::add.Tensor|4| +|39|Tensor<[1,256,1,1]>,
Tensor<[1,256,1,1]>,
|ttnn.add|aten::add.Tensor|4| +|40|Tensor<[1,256,180,320]>,
Tensor<[1,256,180,320]>,
|ttnn.add|aten::add.Tensor|4| +|41|Tensor<[1,128,1,1]>,
Tensor<[1,128,1,1]>,
|ttnn.add|aten::add.Tensor|4| +|42|Tensor<[1,128,180,320]>,
Tensor<[1,128,180,320]>,
|ttnn.add|aten::add.Tensor|4| +|43|Tensor<[1,128,90,160]>,
Tensor<[1,128,90,160]>,
|ttnn.add|aten::add.Tensor|4| +|44|Tensor<[1,512,1,1]>,
Tensor<[1,512,1,1]>,
|ttnn.add|aten::add.Tensor|4| +|45|Tensor<[1,512,90,160]>,
Tensor<[1,512,90,160]>,
|ttnn.add|aten::add.Tensor|4| +|46|Tensor<[1,256,90,160]>,
Tensor<[1,256,90,160]>,
|ttnn.add|aten::add.Tensor|4| +|47|Tensor<[1,256,45,80]>,
Tensor<[1,256,45,80]>,
|ttnn.add|aten::add.Tensor|4| +|48|Tensor<[1,1024,1,1]>,
Tensor<[1,1024,1,1]>,
|ttnn.add|aten::add.Tensor|4| +|49|Tensor<[1,1024,45,80]>,
Tensor<[1,1024,45,80]>,
|ttnn.add|aten::add.Tensor|4| +|50|Tensor<[1,512,45,80]>,
Tensor<[1,512,45,80]>,
|ttnn.add|aten::add.Tensor|4| +|51|Tensor<[1,512,23,40]>,
Tensor<[1,512,23,40]>,
|ttnn.add|aten::add.Tensor|4| +|52|Tensor<[1,2048,1,1]>,
Tensor<[1,2048,1,1]>,
|ttnn.add|aten::add.Tensor|4| +|53|Tensor<[1,2048,23,40]>,
Tensor<[1,2048,23,40]>,
|ttnn.add|aten::add.Tensor|4| +|54|Tensor<[23]>,
Tensor<[23]>,
|ttnn.add|aten::add.Tensor|4| +|55|Tensor<[40]>,
Tensor<[40]>,
|ttnn.add|aten::add.Tensor|4| +|56|Tensor<[1,1,40]>,
Tensor<[1,1,40]>,
|ttnn.add|aten::add.Tensor|4| +|57|Tensor<[1,23,1]>,
Tensor<[1,23,1]>,
|ttnn.add|aten::add.Tensor|4| +|58|Tensor<[920,1,256]>,
Tensor<[920,1,256]>,
|ttnn.add|aten::add.Tensor|5| +|59|Tensor<[920,256]>,
Tensor<[920,256]>,
|ttnn.add|aten::add.Tensor|4| +|60|Tensor<[920,1,1]>,
Tensor<[920,1,1]>,
|ttnn.add|aten::add.Tensor|4| +|61|Tensor<[920,2048]>,
Tensor<[920,2048]>,
|ttnn.add|aten::add.Tensor|4| +|62|Tensor<[100,1,256]>,
Tensor<[100,1,256]>,
|ttnn.add|aten::add.Tensor|5| +|63|Tensor<[100,256]>,
Tensor<[100,256]>,
|ttnn.add|aten::add.Tensor|4| +|64|Tensor<[100,1,1]>,
Tensor<[100,1,1]>,
|ttnn.add|aten::add.Tensor|4| +|65|Tensor<[100,2048]>,
Tensor<[100,2048]>,
|ttnn.add|aten::add.Tensor|4| +|66|Tensor<[6,1,100,92]>,
Tensor<[6,1,100,92]>,
|ttnn.add|aten::add.Tensor|4| +|67|Tensor<[6,1,100,256]>,
Tensor<[6,1,100,256]>,
|ttnn.add|aten::add.Tensor|4| +|68|Tensor<[6,1,100,4]>,
Tensor<[6,1,100,4]>,
|ttnn.add|aten::add.Tensor|4| +|69|Tensor<[8,920,920]>,
Tensor<[8,920,920]>,
|ttnn.add|aten::baddbmm|4| +|70|Tensor<[8,100,920]>,
Tensor<[8,100,920]>,
|ttnn.add|aten::baddbmm|4| +|71|Tensor<[1,256,23,40]>,
Tensor<[1,256,23,40]>,
|ttnn.add|aten::convolution|4| +|72|Tensor<[1,10]>,
Tensor<[1,10]>,
|ttnn.add|aten::add.Tensor|5| +|73|Tensor<[1,10,768]>,
Tensor<[1,10,768]>,
|ttnn.add|aten::add.Tensor|5| +|74|Tensor<[1,10,1]>,
Tensor<[1,10,1]>,
|ttnn.add|aten::add.Tensor|4| +|75|Tensor<[10,768]>,
Tensor<[10,768]>,
|ttnn.add|aten::add.Tensor|4| +|76|Tensor<[1,12,10,10]>,
Tensor<[1,12,10,10]>,
|ttnn.add|aten::add.Tensor|4| +|77|Tensor<[10,3072]>,
Tensor<[10,3072]>,
|ttnn.add|aten::add.Tensor|4| +|78|Tensor<[10,250002]>,
Tensor<[10,250002]>,
|ttnn.add|aten::add.Tensor|4| +|79|Tensor<[1,10,3072]>,
Tensor<[1,10,3072]>,
|ttnn.add|aten::gelu|4| +|80|Tensor<[1,1280]>,
Tensor<[1,1280]>,
|ttnn.add|aten::add.Tensor|4| +|81|Tensor<[1,32,1,1]>,
Tensor<[1,32,1,1]>,
|ttnn.add|aten::add.Tensor|4| +|82|Tensor<[1,320,64,64]>,
Tensor<[1,320,64,64]>,
|ttnn.add|aten::add.Tensor|4| +|83|Tensor<[1,320]>,
Tensor<[1,320]>,
|ttnn.add|aten::add.Tensor|4| +|84|Tensor<[1,4096,1]>,
Tensor<[1,4096,1]>,
|ttnn.add|aten::add.Tensor|4| +|85|Tensor<[1,4096,320]>,
Tensor<[1,4096,320]>,
|ttnn.add|aten::add.Tensor|4| +|86|Tensor<[4096,320]>,
Tensor<[4096,320]>,
|ttnn.add|aten::add.Tensor|4| +|87|Tensor<[4096,2560]>,
Tensor<[4096,2560]>,
|ttnn.add|aten::add.Tensor|4| +|88|Tensor<[1,320,32,32]>,
Tensor<[1,320,32,32]>,
|ttnn.add|aten::add.Tensor|4| +|89|Tensor<[1,640]>,
Tensor<[1,640]>,
|ttnn.add|aten::add.Tensor|4| +|90|Tensor<[1,640,32,32]>,
Tensor<[1,640,32,32]>,
|ttnn.add|aten::add.Tensor|4| +|91|Tensor<[1,1024,1]>,
Tensor<[1,1024,1]>,
|ttnn.add|aten::add.Tensor|4| +|92|Tensor<[1,1024,640]>,
Tensor<[1,1024,640]>,
|ttnn.add|aten::add.Tensor|4| +|93|Tensor<[1024,640]>,
Tensor<[1024,640]>,
|ttnn.add|aten::add.Tensor|4| +|94|Tensor<[1024,5120]>,
Tensor<[1024,5120]>,
|ttnn.add|aten::add.Tensor|4| +|95|Tensor<[1,640,16,16]>,
Tensor<[1,640,16,16]>,
|ttnn.add|aten::add.Tensor|4| +|96|Tensor<[1,1280,16,16]>,
Tensor<[1,1280,16,16]>,
|ttnn.add|aten::add.Tensor|4| +|97|Tensor<[1,256,1280]>,
Tensor<[1,256,1280]>,
|ttnn.add|aten::add.Tensor|4| +|98|Tensor<[256,1280]>,
Tensor<[256,1280]>,
|ttnn.add|aten::add.Tensor|4| +|99|Tensor<[256,10240]>,
Tensor<[256,10240]>,
|ttnn.add|aten::add.Tensor|4| +|100|Tensor<[1,1280,8,8]>,
Tensor<[1,1280,8,8]>,
|ttnn.add|aten::add.Tensor|4| +|101|Tensor<[1,64,1]>,
Tensor<[1,64,1]>,
|ttnn.add|aten::add.Tensor|4| +|102|Tensor<[1,64,1280]>,
Tensor<[1,64,1280]>,
|ttnn.add|aten::add.Tensor|4| +|103|Tensor<[64,1280]>,
Tensor<[64,1280]>,
|ttnn.add|aten::add.Tensor|4| +|104|Tensor<[64,10240]>,
Tensor<[64,10240]>,
|ttnn.add|aten::add.Tensor|4| +|105|Tensor<[1,2560,8,8]>,
Tensor<[1,2560,8,8]>,
|ttnn.add|aten::add.Tensor|4| +|106|Tensor<[16]>,
Tensor<[16]>,
|ttnn.add|aten::add.Tensor|4| +|107|Tensor<[1,2560,16,16]>,
Tensor<[1,2560,16,16]>,
|ttnn.add|aten::add.Tensor|4| +|108|Tensor<[1,1920,16,16]>,
Tensor<[1,1920,16,16]>,
|ttnn.add|aten::add.Tensor|4| +|109|Tensor<[1,1920,32,32]>,
Tensor<[1,1920,32,32]>,
|ttnn.add|aten::add.Tensor|4| +|110|Tensor<[1,1280,32,32]>,
Tensor<[1,1280,32,32]>,
|ttnn.add|aten::add.Tensor|4| +|111|Tensor<[1,960,32,32]>,
Tensor<[1,960,32,32]>,
|ttnn.add|aten::add.Tensor|4| +|112|Tensor<[1,960,64,64]>,
Tensor<[1,960,64,64]>,
|ttnn.add|aten::add.Tensor|4| +|113|Tensor<[1,640,64,64]>,
Tensor<[1,640,64,64]>,
|ttnn.add|aten::add.Tensor|4| +|114|Tensor<[160]>,
Tensor<[160]>,
|ttnn.add|aten::arange.start|4| +|115|Tensor<[1,4,64,64]>,
Tensor<[1,4,64,64]>,
|ttnn.add|aten::convolution|4| +|116|Tensor<[1,4096,1280]>,
Tensor<[1,4096,1280]>,
|ttnn.add|aten::gelu|4| +|117|Tensor<[1,1024,2560]>,
Tensor<[1,1024,2560]>,
|ttnn.add|aten::gelu|4| +|118|Tensor<[1,256,5120]>,
Tensor<[1,256,5120]>,
|ttnn.add|aten::gelu|4| +|119|Tensor<[1,64,5120]>,
Tensor<[1,64,5120]>,
|ttnn.add|aten::gelu|4| +|120|Tensor<[1280]>,
Tensor<[1280]>,
|ttnn.add|aten::index.Tensor|4| +|121|Tensor<[640]>,
Tensor<[640]>,
|ttnn.add|aten::index.Tensor|4| +|122|Tensor<[1,25,768]>,
Tensor<[1,25,768]>,
|ttnn.add|aten::add.Tensor|5| +|123|Tensor<[1,25,1]>,
Tensor<[1,25,1]>,
|ttnn.add|aten::add.Tensor|4| +|124|Tensor<[25,768]>,
Tensor<[25,768]>,
|ttnn.add|aten::add.Tensor|4| +|125|Tensor<[1,12,25,25]>,
Tensor<[1,12,25,25]>,
|ttnn.add|aten::add.Tensor|4| +|126|Tensor<[25,3072]>,
Tensor<[25,3072]>,
|ttnn.add|aten::add.Tensor|4| +|127|Tensor<[25,2]>,
Tensor<[25,2]>,
|ttnn.add|aten::add.Tensor|4| +|128|Tensor<[1,1]>,
Tensor<[1,1]>,
|ttnn.add|aten::add.Tensor|4| +|129|Tensor<[1,25,3072]>,
Tensor<[1,25,3072]>,
|ttnn.add|aten::gelu|4| +|130|Tensor<[1,1445,192]>,
Tensor<[1,1445,192]>,
|ttnn.add|aten::add.Tensor|5| +|131|Tensor<[1,1445,1]>,
Tensor<[1,1445,1]>,
|ttnn.add|aten::add.Tensor|4| +|132|Tensor<[1445,192]>,
Tensor<[1445,192]>,
|ttnn.add|aten::add.Tensor|4| +|133|Tensor<[1445,768]>,
Tensor<[1445,768]>,
|ttnn.add|aten::add.Tensor|4| +|134|Tensor<[100,192]>,
Tensor<[100,192]>,
|ttnn.add|aten::add.Tensor|4| +|135|Tensor<[100,92]>,
Tensor<[100,92]>,
|ttnn.add|aten::add.Tensor|4| +|136|Tensor<[100,4]>,
Tensor<[100,4]>,
|ttnn.add|aten::add.Tensor|4| +|137|Tensor<[1,192,32,42]>,
Tensor<[1,192,32,42]>,
|ttnn.add|aten::convolution|4| +|138|Tensor<[1,1445,768]>,
Tensor<[1,1445,768]>,
|ttnn.add|aten::gelu|4| +|139|Tensor<[1,256,14,14]>,
Tensor<[1,256,14,14]>,
|ttnn.add|aten::add.Tensor|4| +|140|Tensor<[1,512,7,7]>,
Tensor<[1,512,7,7]>,
|ttnn.add|aten::add.Tensor|4| +|141|Tensor<[1,8,768]>,
Tensor<[1,8,768]>,
|ttnn.add|aten::add.Tensor|5| +|142|Tensor<[1,8,1]>,
Tensor<[1,8,1]>,
|ttnn.add|aten::add.Tensor|4| +|143|Tensor<[1,12,8,8]>,
Tensor<[1,12,8,8]>,
|ttnn.add|aten::add.Tensor|4| +|144|Tensor<[1,768,8]>,
Tensor<[1,768,8]>,
|ttnn.add|aten::add.Tensor|5| +|145|Tensor<[1,768]>,
Tensor<[1,768]>,
|ttnn.add|aten::add.Tensor|4| +|146|Tensor<[1,3]>,
Tensor<[1,3]>,
|ttnn.add|aten::add.Tensor|4| +|147|Tensor<[1,3072,8]>,
Tensor<[1,3072,8]>,
|ttnn.add|aten::convolution|4| +|148|Tensor<[1,2048,768]>,
Tensor<[1,2048,768]>,
|ttnn.add|aten::add.Tensor|4| +|149|Tensor<[1,2048,1]>,
Tensor<[1,2048,1]>,
|ttnn.add|aten::add.Tensor|4| +|150|Tensor<[2048,256]>,
Tensor<[2048,256]>,
|ttnn.add|aten::add.Tensor|4| +|151|Tensor<[2048,1280]>,
Tensor<[2048,1280]>,
|ttnn.add|aten::add.Tensor|4| +|152|Tensor<[1,8,256,2048]>,
Tensor<[1,8,256,2048]>,
|ttnn.add|aten::add.Tensor|4| +|153|Tensor<[256,768]>,
Tensor<[256,768]>,
|ttnn.add|aten::add.Tensor|4| +|154|Tensor<[2048,768]>,
Tensor<[2048,768]>,
|ttnn.add|aten::add.Tensor|4| +|155|Tensor<[2048,262]>,
Tensor<[2048,262]>,
|ttnn.add|aten::add.Tensor|4| +|156|Tensor<[2048]>,
Tensor<[2048]>,
|ttnn.add|aten::arange.start|4| +|157|Tensor<[1,256,56,56]>,
Tensor<[1,256,56,56]>,
|ttnn.add|aten::add.Tensor|4| +|158|Tensor<[1024]>,
Tensor<[1024]>,
|ttnn.add|aten::add.Tensor|4| +|159|Tensor<[1,1024,14,14]>,
Tensor<[1,1024,14,14]>,
|ttnn.add|aten::add.Tensor|4| +|160|Tensor<[1,512,14,14]>,
Tensor<[1,512,14,14]>,
|ttnn.add|aten::add.Tensor|4| +|161|Tensor<[1,2048,7,7]>,
Tensor<[1,2048,7,7]>,
|ttnn.add|aten::add.Tensor|4| +|162|Tensor<[12]>,
Tensor<[12]>,
|ttnn.add|aten::add.Tensor|4| +|163|Tensor<[1,193,768]>,
Tensor<[1,193,768]>,
|ttnn.add|aten::add.Tensor|5| +|164|Tensor<[1,201,1]>,
Tensor<[1,201,1]>,
|ttnn.add|aten::add.Tensor|4| +|165|Tensor<[1,201,768]>,
Tensor<[1,201,768]>,
|ttnn.add|aten::add.Tensor|4| +|166|Tensor<[201,768]>,
Tensor<[201,768]>,
|ttnn.add|aten::add.Tensor|4| +|167|Tensor<[1,12,201,201]>,
Tensor<[1,12,201,201]>,
|ttnn.add|aten::add.Tensor|4| +|168|Tensor<[201,3072]>,
Tensor<[201,3072]>,
|ttnn.add|aten::add.Tensor|4| +|169|Tensor<[1,1536]>,
Tensor<[1,1536]>,
|ttnn.add|aten::add.Tensor|4| +|170|Tensor<[1,3129]>,
Tensor<[1,3129]>,
|ttnn.add|aten::add.Tensor|4| +|171|Tensor<[1,768,12,16]>,
Tensor<[1,768,12,16]>,
|ttnn.add|aten::convolution|4| +|172|Tensor<[1,201,3072]>,
Tensor<[1,201,3072]>,
|ttnn.add|aten::gelu|4| +|173|Tensor<[1,128]>,
Tensor<[1,128]>,
|ttnn.add|aten::add.Tensor|4| +|174|Tensor<[1,32,26,26]>,
Tensor<[1,32,26,26]>,
|ttnn.add|aten::convolution|4| +|175|Tensor<[1,64,24,24]>,
Tensor<[1,64,24,24]>,
|ttnn.add|aten::convolution|4| +|176|Tensor<[19]>,
Tensor<[19]>,
|ttnn.add|aten::add.Tensor|4| +|177|Tensor<[1,19]>,
Tensor<[1,19]>,
|ttnn.add|aten::add.Tensor|4| +|178|Tensor<[1,19,1024]>,
Tensor<[1,19,1024]>,
|ttnn.add|aten::add.Tensor|5| +|179|Tensor<[1,19,1]>,
Tensor<[1,19,1]>,
|ttnn.add|aten::add.Tensor|4| +|180|Tensor<[19,1024]>,
Tensor<[19,1024]>,
|ttnn.add|aten::add.Tensor|4| +|181|Tensor<[1,16,19,19]>,
Tensor<[1,16,19,19]>,
|ttnn.add|aten::add.Tensor|4| +|182|Tensor<[19,4096]>,
Tensor<[19,4096]>,
|ttnn.add|aten::add.Tensor|4| +|183|Tensor<[1,19,4096]>,
Tensor<[1,19,4096]>,
|ttnn.add|aten::gelu|4| +|184|Tensor<[14]>,
Tensor<[14]>,
|ttnn.add|aten::add.Tensor|4| +|185|Tensor<[1,14,56,56]>,
Tensor<[1,14,56,56]>,
|ttnn.add|aten::add.Tensor|4| +|186|Tensor<[24]>,
Tensor<[24]>,
|ttnn.add|aten::add.Tensor|4| +|187|Tensor<[1,24,56,56]>,
Tensor<[1,24,56,56]>,
|ttnn.add|aten::add.Tensor|4| +|188|Tensor<[1,40,56,56]>,
Tensor<[1,40,56,56]>,
|ttnn.add|aten::add.Tensor|4| +|189|Tensor<[68]>,
Tensor<[68]>,
|ttnn.add|aten::add.Tensor|4| +|190|Tensor<[1,68,56,56]>,
Tensor<[1,68,56,56]>,
|ttnn.add|aten::add.Tensor|4| +|191|Tensor<[1,16,28,28]>,
Tensor<[1,16,28,28]>,
|ttnn.add|aten::add.Tensor|4| +|192|Tensor<[28]>,
Tensor<[28]>,
|ttnn.add|aten::add.Tensor|4| +|193|Tensor<[1,28,28,28]>,
Tensor<[1,28,28,28]>,
|ttnn.add|aten::add.Tensor|4| +|194|Tensor<[46]>,
Tensor<[46]>,
|ttnn.add|aten::add.Tensor|4| +|195|Tensor<[1,46,28,28]>,
Tensor<[1,46,28,28]>,
|ttnn.add|aten::add.Tensor|4| +|196|Tensor<[78]>,
Tensor<[78]>,
|ttnn.add|aten::add.Tensor|4| +|197|Tensor<[1,78,28,28]>,
Tensor<[1,78,28,28]>,
|ttnn.add|aten::add.Tensor|4| +|198|Tensor<[134]>,
Tensor<[134]>,
|ttnn.add|aten::add.Tensor|4| +|199|Tensor<[1,134,28,28]>,
Tensor<[1,134,28,28]>,
|ttnn.add|aten::add.Tensor|4| +|200|Tensor<[20]>,
Tensor<[20]>,
|ttnn.add|aten::add.Tensor|4| +|201|Tensor<[1,20,28,28]>,
Tensor<[1,20,28,28]>,
|ttnn.add|aten::add.Tensor|4| +|202|Tensor<[34]>,
Tensor<[34]>,
|ttnn.add|aten::add.Tensor|4| +|203|Tensor<[1,34,28,28]>,
Tensor<[1,34,28,28]>,
|ttnn.add|aten::add.Tensor|4| +|204|Tensor<[58]>,
Tensor<[58]>,
|ttnn.add|aten::add.Tensor|4| +|205|Tensor<[1,58,28,28]>,
Tensor<[1,58,28,28]>,
|ttnn.add|aten::add.Tensor|4| +|206|Tensor<[98]>,
Tensor<[98]>,
|ttnn.add|aten::add.Tensor|4| +|207|Tensor<[1,98,28,28]>,
Tensor<[1,98,28,28]>,
|ttnn.add|aten::add.Tensor|4| +|208|Tensor<[168]>,
Tensor<[168]>,
|ttnn.add|aten::add.Tensor|4| +|209|Tensor<[1,168,28,28]>,
Tensor<[1,168,28,28]>,
|ttnn.add|aten::add.Tensor|4| +|210|Tensor<[320]>,
Tensor<[320]>,
|ttnn.add|aten::add.Tensor|4| +|211|Tensor<[1,320,28,28]>,
Tensor<[1,320,28,28]>,
|ttnn.add|aten::add.Tensor|4| +|212|Tensor<[1,40,14,14]>,
Tensor<[1,40,14,14]>,
|ttnn.add|aten::add.Tensor|4| +|213|Tensor<[1,68,14,14]>,
Tensor<[1,68,14,14]>,
|ttnn.add|aten::add.Tensor|4| +|214|Tensor<[116]>,
Tensor<[116]>,
|ttnn.add|aten::add.Tensor|4| +|215|Tensor<[1,116,14,14]>,
Tensor<[1,116,14,14]>,
|ttnn.add|aten::add.Tensor|4| +|216|Tensor<[196]>,
Tensor<[196]>,
|ttnn.add|aten::add.Tensor|4| +|217|Tensor<[1,196,14,14]>,
Tensor<[1,196,14,14]>,
|ttnn.add|aten::add.Tensor|4| +|218|Tensor<[334]>,
Tensor<[334]>,
|ttnn.add|aten::add.Tensor|4| +|219|Tensor<[1,334,14,14]>,
Tensor<[1,334,14,14]>,
|ttnn.add|aten::add.Tensor|4| +|220|Tensor<[1,640,14,14]>,
Tensor<[1,640,14,14]>,
|ttnn.add|aten::add.Tensor|4| +|221|Tensor<[1,160,7,7]>,
Tensor<[1,160,7,7]>,
|ttnn.add|aten::add.Tensor|4| +|222|Tensor<[272]>,
Tensor<[272]>,
|ttnn.add|aten::add.Tensor|4| +|223|Tensor<[1,272,7,7]>,
Tensor<[1,272,7,7]>,
|ttnn.add|aten::add.Tensor|4| +|224|Tensor<[462]>,
Tensor<[462]>,
|ttnn.add|aten::add.Tensor|4| +|225|Tensor<[1,462,7,7]>,
Tensor<[1,462,7,7]>,
|ttnn.add|aten::add.Tensor|4| +|226|Tensor<[1,1024,7,7]>,
Tensor<[1,1024,7,7]>,
|ttnn.add|aten::add.Tensor|4| +|227|Tensor<[1,32,512,512]>,
Tensor<[1,32,512,512]>,
|ttnn.add|aten::add.Tensor|4| +|228|Tensor<[1,64,256,256]>,
Tensor<[1,64,256,256]>,
|ttnn.add|aten::add.Tensor|4| +|229|Tensor<[1,32,256,256]>,
Tensor<[1,32,256,256]>,
|ttnn.add|aten::add.Tensor|4| +|230|Tensor<[1,128,128,128]>,
Tensor<[1,128,128,128]>,
|ttnn.add|aten::add.Tensor|4| +|231|Tensor<[1,64,128,128]>,
Tensor<[1,64,128,128]>,
|ttnn.add|aten::add.Tensor|4| +|232|Tensor<[1,256,64,64]>,
Tensor<[1,256,64,64]>,
|ttnn.add|aten::add.Tensor|4| +|233|Tensor<[1,128,64,64]>,
Tensor<[1,128,64,64]>,
|ttnn.add|aten::add.Tensor|4| +|234|Tensor<[1,512,32,32]>,
Tensor<[1,512,32,32]>,
|ttnn.add|aten::add.Tensor|4| +|235|Tensor<[1,256,32,32]>,
Tensor<[1,256,32,32]>,
|ttnn.add|aten::add.Tensor|4| +|236|Tensor<[1,1024,16,16]>,
Tensor<[1,1024,16,16]>,
|ttnn.add|aten::add.Tensor|4| +|237|Tensor<[1,512,16,16]>,
Tensor<[1,512,16,16]>,
|ttnn.add|aten::add.Tensor|4| +|238|Tensor<[1,256,16,16]>,
Tensor<[1,256,16,16]>,
|ttnn.add|aten::add.Tensor|4| +|239|Tensor<[1,128,32,32]>,
Tensor<[1,128,32,32]>,
|ttnn.add|aten::add.Tensor|4| +|240|Tensor<[1,255,16,16]>,
Tensor<[1,255,16,16]>,
|ttnn.add|aten::convolution|4| +|241|Tensor<[1,255,32,32]>,
Tensor<[1,255,32,32]>,
|ttnn.add|aten::convolution|4| +|242|Tensor<[1,255,64,64]>,
Tensor<[1,255,64,64]>,
|ttnn.add|aten::convolution|4| +|243|Tensor<[1,1,256,256]>,
Tensor<[1,1,256,256]>,
|ttnn.add|aten::convolution|4| +|244|Tensor<[1,4,14,14]>,
Tensor<[1,4,14,14]>,
|ttnn.add|aten::convolution|4| +|245|Tensor<[1,16,14,14]>,
Tensor<[1,16,14,14]>,
|ttnn.add|aten::convolution|4| +|246|Tensor<[1,1,28,28]>,
Tensor<[1,1,28,28]>,
|ttnn.add|aten::convolution|4| +|247|Tensor<[1,32,1536]>,
Tensor<[1,32,1536]>,
|ttnn.add|aten::add.Tensor|4| +|248|Tensor<[32,4608]>,
Tensor<[32,4608]>,
|ttnn.add|aten::add.Tensor|4| +|249|Tensor<[1,16,32,32]>,
Tensor<[1,16,32,32]>,
|ttnn.add|aten::add.Tensor|4| +|250|Tensor<[32,1536]>,
Tensor<[32,1536]>,
|ttnn.add|aten::add.Tensor|4| +|251|Tensor<[32,6144]>,
Tensor<[32,6144]>,
|ttnn.add|aten::add.Tensor|4| +|252|Tensor<[1,32,6144]>,
Tensor<[1,32,6144]>,
|ttnn.add|aten::add.Tensor|4| +|253|Tensor<[16,32,32]>,
Tensor<[16,32,32]>,
|ttnn.add|aten::baddbmm|4| +|254|Tensor<[1,16,768]>,
Tensor<[1,16,768]>,
|ttnn.add|aten::add.Tensor|5| +|255|Tensor<[1,16,1]>,
Tensor<[1,16,1]>,
|ttnn.add|aten::add.Tensor|4| +|256|Tensor<[16,768]>,
Tensor<[16,768]>,
|ttnn.add|aten::add.Tensor|4| +|257|Tensor<[1,12,16,16]>,
Tensor<[1,12,16,16]>,
|ttnn.add|aten::add.Tensor|4| +|258|Tensor<[16,3072]>,
Tensor<[16,3072]>,
|ttnn.add|aten::add.Tensor|4| +|259|Tensor<[1,16,3072]>,
Tensor<[1,16,3072]>,
|ttnn.add|aten::gelu|4| +|260|Tensor<[1,64,224,224]>,
Tensor<[1,64,224,224]>,
|ttnn.add|aten::add.Tensor|4| +|261|Tensor<[1,128,112,112]>,
Tensor<[1,128,112,112]>,
|ttnn.add|aten::add.Tensor|4| +|262|Tensor<[1,1,224,224]>,
Tensor<[1,1,224,224]>,
|ttnn.add|aten::convolution|4| +|263|Tensor<[1,19200,1]>,
Tensor<[1,19200,1]>,
|ttnn.add|aten::add.Tensor|4| +|264|Tensor<[1,19200,64]>,
Tensor<[1,19200,64]>,
|ttnn.add|aten::add.Tensor|4| +|265|Tensor<[19200,64]>,
Tensor<[19200,64]>,
|ttnn.add|aten::add.Tensor|4| +|266|Tensor<[1,300,1]>,
Tensor<[1,300,1]>,
|ttnn.add|aten::add.Tensor|4| +|267|Tensor<[1,300,64]>,
Tensor<[1,300,64]>,
|ttnn.add|aten::add.Tensor|4| +|268|Tensor<[300,64]>,
Tensor<[300,64]>,
|ttnn.add|aten::add.Tensor|4| +|269|Tensor<[19200,256]>,
Tensor<[19200,256]>,
|ttnn.add|aten::add.Tensor|4| +|270|Tensor<[1,4800,1]>,
Tensor<[1,4800,1]>,
|ttnn.add|aten::add.Tensor|4| +|271|Tensor<[1,4800,128]>,
Tensor<[1,4800,128]>,
|ttnn.add|aten::add.Tensor|4| +|272|Tensor<[4800,128]>,
Tensor<[4800,128]>,
|ttnn.add|aten::add.Tensor|4| +|273|Tensor<[1,300,128]>,
Tensor<[1,300,128]>,
|ttnn.add|aten::add.Tensor|4| +|274|Tensor<[300,128]>,
Tensor<[300,128]>,
|ttnn.add|aten::add.Tensor|4| +|275|Tensor<[4800,512]>,
Tensor<[4800,512]>,
|ttnn.add|aten::add.Tensor|4| +|276|Tensor<[1,1200,1]>,
Tensor<[1,1200,1]>,
|ttnn.add|aten::add.Tensor|4| +|277|Tensor<[1,1200,320]>,
Tensor<[1,1200,320]>,
|ttnn.add|aten::add.Tensor|4| +|278|Tensor<[1200,320]>,
Tensor<[1200,320]>,
|ttnn.add|aten::add.Tensor|4| +|279|Tensor<[1,300,320]>,
Tensor<[1,300,320]>,
|ttnn.add|aten::add.Tensor|4| +|280|Tensor<[300,320]>,
Tensor<[300,320]>,
|ttnn.add|aten::add.Tensor|4| +|281|Tensor<[1200,1280]>,
Tensor<[1200,1280]>,
|ttnn.add|aten::add.Tensor|4| +|282|Tensor<[1,300,512]>,
Tensor<[1,300,512]>,
|ttnn.add|aten::add.Tensor|4| +|283|Tensor<[300,512]>,
Tensor<[300,512]>,
|ttnn.add|aten::add.Tensor|4| +|284|Tensor<[300,2048]>,
Tensor<[300,2048]>,
|ttnn.add|aten::add.Tensor|4| +|285|Tensor<[30]>,
Tensor<[30]>,
|ttnn.add|aten::add.Tensor|4| +|286|Tensor<[30,1]>,
Tensor<[30,1]>,
|ttnn.add|aten::add.Tensor|4| +|287|Tensor<[1,64,30,40]>,
Tensor<[1,64,30,40]>,
|ttnn.add|aten::add.Tensor|5| +|288|Tensor<[1,32,30,40]>,
Tensor<[1,32,30,40]>,
|ttnn.add|aten::add.Tensor|4| +|289|Tensor<[60]>,
Tensor<[60]>,
|ttnn.add|aten::add.Tensor|4| +|290|Tensor<[60,1]>,
Tensor<[60,1]>,
|ttnn.add|aten::add.Tensor|4| +|291|Tensor<[80]>,
Tensor<[80]>,
|ttnn.add|aten::add.Tensor|4| +|292|Tensor<[1,64,60,80]>,
Tensor<[1,64,60,80]>,
|ttnn.add|aten::add.Tensor|5| +|293|Tensor<[1,32,60,80]>,
Tensor<[1,32,60,80]>,
|ttnn.add|aten::add.Tensor|4| +|294|Tensor<[120]>,
Tensor<[120]>,
|ttnn.add|aten::add.Tensor|4| +|295|Tensor<[120,1]>,
Tensor<[120,1]>,
|ttnn.add|aten::add.Tensor|4| +|296|Tensor<[1,64,120,160]>,
Tensor<[1,64,120,160]>,
|ttnn.add|aten::add.Tensor|5| +|297|Tensor<[1,32,120,160]>,
Tensor<[1,32,120,160]>,
|ttnn.add|aten::add.Tensor|4| +|298|Tensor<[240]>,
Tensor<[240]>,
|ttnn.add|aten::add.Tensor|4| +|299|Tensor<[240,1]>,
Tensor<[240,1]>,
|ttnn.add|aten::add.Tensor|4| +|300|Tensor<[1,64,240,320]>,
Tensor<[1,64,240,320]>,
|ttnn.add|aten::add.Tensor|5| +|301|Tensor<[480]>,
Tensor<[480]>,
|ttnn.add|aten::add.Tensor|4| +|302|Tensor<[480,1]>,
Tensor<[480,1]>,
|ttnn.add|aten::add.Tensor|4| +|303|Tensor<[1,64,480,640]>,
Tensor<[1,64,480,640]>,
|ttnn.add|aten::add.Tensor|5| +|304|Tensor<[1,64,15,20]>,
Tensor<[1,64,15,20]>,
|ttnn.add|aten::convolution|4| +|305|Tensor<[1,256,120,160]>,
Tensor<[1,256,120,160]>,
|ttnn.add|aten::convolution|4| +|306|Tensor<[1,128,60,80]>,
Tensor<[1,128,60,80]>,
|ttnn.add|aten::convolution|4| +|307|Tensor<[1,128,15,20]>,
Tensor<[1,128,15,20]>,
|ttnn.add|aten::convolution|4| +|308|Tensor<[1,512,60,80]>,
Tensor<[1,512,60,80]>,
|ttnn.add|aten::convolution|4| +|309|Tensor<[1,320,30,40]>,
Tensor<[1,320,30,40]>,
|ttnn.add|aten::convolution|4| +|310|Tensor<[1,320,15,20]>,
Tensor<[1,320,15,20]>,
|ttnn.add|aten::convolution|4| +|311|Tensor<[1,1280,30,40]>,
Tensor<[1,1280,30,40]>,
|ttnn.add|aten::convolution|4| +|312|Tensor<[1,512,15,20]>,
Tensor<[1,512,15,20]>,
|ttnn.add|aten::convolution|4| +|313|Tensor<[1,2048,15,20]>,
Tensor<[1,2048,15,20]>,
|ttnn.add|aten::convolution|4| +|314|Tensor<[1,2,30,40]>,
Tensor<[1,2,30,40]>,
|ttnn.add|aten::convolution|4| +|315|Tensor<[1,2,60,80]>,
Tensor<[1,2,60,80]>,
|ttnn.add|aten::convolution|4| +|316|Tensor<[1,2,120,160]>,
Tensor<[1,2,120,160]>,
|ttnn.add|aten::convolution|4| +|317|Tensor<[1,1,480,640]>,
Tensor<[1,1,480,640]>,
|ttnn.add|aten::convolution|4| +|318|Tensor<[1,19200,256]>,
Tensor<[1,19200,256]>,
|ttnn.add|aten::gelu|4| +|319|Tensor<[1,4800,512]>,
Tensor<[1,4800,512]>,
|ttnn.add|aten::gelu|4| +|320|Tensor<[1,1200,1280]>,
Tensor<[1,1200,1280]>,
|ttnn.add|aten::gelu|4| +|321|Tensor<[1,300,2048]>,
Tensor<[1,300,2048]>,
|ttnn.add|aten::gelu|4| +|322|Tensor<[1,197,768]>,
Tensor<[1,197,768]>,
|ttnn.add|aten::add.Tensor|5| +|323|Tensor<[1,197,1]>,
Tensor<[1,197,1]>,
|ttnn.add|aten::add.Tensor|4| +|324|Tensor<[197,768]>,
Tensor<[197,768]>,
|ttnn.add|aten::add.Tensor|4| +|325|Tensor<[197,3072]>,
Tensor<[197,3072]>,
|ttnn.add|aten::add.Tensor|4| +|326|Tensor<[1,768,14,14]>,
Tensor<[1,768,14,14]>,
|ttnn.add|aten::convolution|4| +|327|Tensor<[1,197,3072]>,
Tensor<[1,197,3072]>,
|ttnn.add|aten::gelu|4| +|328|Tensor<[1,16384,1]>,
Tensor<[1,16384,1]>,
|ttnn.add|aten::add.Tensor|4| +|329|Tensor<[1,16384,32]>,
Tensor<[1,16384,32]>,
|ttnn.add|aten::add.Tensor|4| +|330|Tensor<[16384,32]>,
Tensor<[16384,32]>,
|ttnn.add|aten::add.Tensor|4| +|331|Tensor<[1,256,32]>,
Tensor<[1,256,32]>,
|ttnn.add|aten::add.Tensor|4| +|332|Tensor<[256,32]>,
Tensor<[256,32]>,
|ttnn.add|aten::add.Tensor|4| +|333|Tensor<[16384,128]>,
Tensor<[16384,128]>,
|ttnn.add|aten::add.Tensor|4| +|334|Tensor<[1,4096,64]>,
Tensor<[1,4096,64]>,
|ttnn.add|aten::add.Tensor|4| +|335|Tensor<[4096,64]>,
Tensor<[4096,64]>,
|ttnn.add|aten::add.Tensor|4| +|336|Tensor<[1,256,64]>,
Tensor<[1,256,64]>,
|ttnn.add|aten::add.Tensor|4| +|337|Tensor<[256,64]>,
Tensor<[256,64]>,
|ttnn.add|aten::add.Tensor|4| +|338|Tensor<[4096,256]>,
Tensor<[4096,256]>,
|ttnn.add|aten::add.Tensor|4| +|339|Tensor<[1,1024,160]>,
Tensor<[1,1024,160]>,
|ttnn.add|aten::add.Tensor|4| +|340|Tensor<[1024,160]>,
Tensor<[1024,160]>,
|ttnn.add|aten::add.Tensor|4| +|341|Tensor<[1,256,160]>,
Tensor<[1,256,160]>,
|ttnn.add|aten::add.Tensor|4| +|342|Tensor<[256,160]>,
Tensor<[256,160]>,
|ttnn.add|aten::add.Tensor|4| +|343|Tensor<[256,1024]>,
Tensor<[256,1024]>,
|ttnn.add|aten::add.Tensor|4| +|344|Tensor<[1,16384,256]>,
Tensor<[1,16384,256]>,
|ttnn.add|aten::add.Tensor|4| +|345|Tensor<[128,1]>,
Tensor<[128,1]>,
|ttnn.add|aten::add.Tensor|4| +|346|Tensor<[1,256,128,128]>,
Tensor<[1,256,128,128]>,
|ttnn.add|aten::add.Tensor|5| +|347|Tensor<[1,4096,256]>,
Tensor<[1,4096,256]>,
|ttnn.add|aten::add.Tensor|4| +|348|Tensor<[1,1024,256]>,
Tensor<[1,1024,256]>,
|ttnn.add|aten::add.Tensor|4| +|349|Tensor<[1,32,128,128]>,
Tensor<[1,32,128,128]>,
|ttnn.add|aten::convolution|4| +|350|Tensor<[1,32,16,16]>,
Tensor<[1,32,16,16]>,
|ttnn.add|aten::convolution|4| +|351|Tensor<[1,64,64,64]>,
Tensor<[1,64,64,64]>,
|ttnn.add|aten::convolution|4| +|352|Tensor<[1,64,16,16]>,
Tensor<[1,64,16,16]>,
|ttnn.add|aten::convolution|4| +|353|Tensor<[1,160,32,32]>,
Tensor<[1,160,32,32]>,
|ttnn.add|aten::convolution|4| +|354|Tensor<[1,160,16,16]>,
Tensor<[1,160,16,16]>,
|ttnn.add|aten::convolution|4| +|355|Tensor<[1,150,128,128]>,
Tensor<[1,150,128,128]>,
|ttnn.add|aten::convolution|4| +|356|Tensor<[1,16384,128]>,
Tensor<[1,16384,128]>,
|ttnn.add|aten::gelu|4| +|357|Tensor<[1,256,1024]>,
Tensor<[1,256,1024]>,
|ttnn.add|aten::gelu|4| +|358|Tensor<[1,1,7,7]>,
Tensor<[1,1,7,7]>,
|ttnn.add|aten::add.Tensor|4| +|359|Tensor<[1,7,4544]>,
Tensor<[1,7,4544]>,
|ttnn.add|aten::add.Tensor|4| +|360|Tensor<[1,71,7,64]>,
Tensor<[1,71,7,64]>,
|ttnn.add|aten::add.Tensor|5| +|361|Tensor<[1,1,7,64]>,
Tensor<[1,1,7,64]>,
|ttnn.add|aten::add.Tensor|5| +|362|Tensor<[1,71,7,7]>,
Tensor<[1,71,7,7]>,
|ttnn.add|aten::add.Tensor|4| +|363|Tensor<[1,7,18176]>,
Tensor<[1,7,18176]>,
|ttnn.add|aten::gelu|4| +|364|Tensor<[7,1]>,
Tensor<[7,1]>,
|ttnn.add|aten::triu|4| +|365|Tensor<[1,16,112,112]>,
Tensor<[1,16,112,112]>,
|ttnn.add|aten::add.Tensor|4| +|366|Tensor<[96]>,
Tensor<[96]>,
|ttnn.add|aten::add.Tensor|4| +|367|Tensor<[1,96,112,112]>,
Tensor<[1,96,112,112]>,
|ttnn.add|aten::add.Tensor|4| +|368|Tensor<[1,96,56,56]>,
Tensor<[1,96,56,56]>,
|ttnn.add|aten::add.Tensor|4| +|369|Tensor<[144]>,
Tensor<[144]>,
|ttnn.add|aten::add.Tensor|4| +|370|Tensor<[1,144,56,56]>,
Tensor<[1,144,56,56]>,
|ttnn.add|aten::add.Tensor|4| +|371|Tensor<[1,144,28,28]>,
Tensor<[1,144,28,28]>,
|ttnn.add|aten::add.Tensor|4| +|372|Tensor<[1,32,28,28]>,
Tensor<[1,32,28,28]>,
|ttnn.add|aten::add.Tensor|4| +|373|Tensor<[192]>,
Tensor<[192]>,
|ttnn.add|aten::add.Tensor|4| +|374|Tensor<[1,192,28,28]>,
Tensor<[1,192,28,28]>,
|ttnn.add|aten::add.Tensor|4| +|375|Tensor<[1,192,14,14]>,
Tensor<[1,192,14,14]>,
|ttnn.add|aten::add.Tensor|4| +|376|Tensor<[1,64,14,14]>,
Tensor<[1,64,14,14]>,
|ttnn.add|aten::add.Tensor|4| +|377|Tensor<[384]>,
Tensor<[384]>,
|ttnn.add|aten::add.Tensor|4| +|378|Tensor<[1,384,14,14]>,
Tensor<[1,384,14,14]>,
|ttnn.add|aten::add.Tensor|4| +|379|Tensor<[1,96,14,14]>,
Tensor<[1,96,14,14]>,
|ttnn.add|aten::add.Tensor|4| +|380|Tensor<[576]>,
Tensor<[576]>,
|ttnn.add|aten::add.Tensor|4| +|381|Tensor<[1,576,14,14]>,
Tensor<[1,576,14,14]>,
|ttnn.add|aten::add.Tensor|4| +|382|Tensor<[1,576,7,7]>,
Tensor<[1,576,7,7]>,
|ttnn.add|aten::add.Tensor|4| +|383|Tensor<[960]>,
Tensor<[960]>,
|ttnn.add|aten::add.Tensor|4| +|384|Tensor<[1,960,7,7]>,
Tensor<[1,960,7,7]>,
|ttnn.add|aten::add.Tensor|4| +|385|Tensor<[1,320,7,7]>,
Tensor<[1,320,7,7]>,
|ttnn.add|aten::add.Tensor|4| +|386|Tensor<[1,1280,7,7]>,
Tensor<[1,1280,7,7]>,
|ttnn.add|aten::add.Tensor|4| +|387|Tensor<[1,12,128]>,
Tensor<[1,12,128]>,
|ttnn.add|aten::add.Tensor|5| +|388|Tensor<[1,12,1]>,
Tensor<[1,12,1]>,
|ttnn.add|aten::add.Tensor|4| +|389|Tensor<[12,768]>,
Tensor<[12,768]>,
|ttnn.add|aten::add.Tensor|4| +|390|Tensor<[1,12,12,12]>,
Tensor<[1,12,12,12]>,
|ttnn.add|aten::add.Tensor|4| +|391|Tensor<[1,12,768]>,
Tensor<[1,12,768]>,
|ttnn.add|aten::add.Tensor|5| +|392|Tensor<[12,3072]>,
Tensor<[12,3072]>,
|ttnn.add|aten::add.Tensor|4| +|393|Tensor<[1,12,3072]>,
Tensor<[1,12,3072]>,
|ttnn.add|aten::add.Tensor|5| +|394|Tensor<[12,2]>,
Tensor<[12,2]>,
|ttnn.add|aten::add.Tensor|4| +|395|Tensor<[1,9,128]>,
Tensor<[1,9,128]>,
|ttnn.add|aten::add.Tensor|5| +|396|Tensor<[1,9,1]>,
Tensor<[1,9,1]>,
|ttnn.add|aten::add.Tensor|4| +|397|Tensor<[9,768]>,
Tensor<[9,768]>,
|ttnn.add|aten::add.Tensor|4| +|398|Tensor<[1,12,9,9]>,
Tensor<[1,12,9,9]>,
|ttnn.add|aten::add.Tensor|4| +|399|Tensor<[1,9,768]>,
Tensor<[1,9,768]>,
|ttnn.add|aten::add.Tensor|5| +|400|Tensor<[9,3072]>,
Tensor<[9,3072]>,
|ttnn.add|aten::add.Tensor|4| +|401|Tensor<[1,9,3072]>,
Tensor<[1,9,3072]>,
|ttnn.add|aten::add.Tensor|5| +|402|Tensor<[9,128]>,
Tensor<[9,128]>,
|ttnn.add|aten::add.Tensor|4| +|403|Tensor<[9,30000]>,
Tensor<[9,30000]>,
|ttnn.add|aten::add.Tensor|4| +|404|Tensor<[9,2048]>,
Tensor<[9,2048]>,
|ttnn.add|aten::add.Tensor|4| +|405|Tensor<[1,16,9,9]>,
Tensor<[1,16,9,9]>,
|ttnn.add|aten::add.Tensor|4| +|406|Tensor<[1,9,2048]>,
Tensor<[1,9,2048]>,
|ttnn.add|aten::add.Tensor|5| +|407|Tensor<[9,8192]>,
Tensor<[9,8192]>,
|ttnn.add|aten::add.Tensor|4| +|408|Tensor<[1,9,8192]>,
Tensor<[1,9,8192]>,
|ttnn.add|aten::add.Tensor|5| +|409|Tensor<[9,1024]>,
Tensor<[9,1024]>,
|ttnn.add|aten::add.Tensor|4| +|410|Tensor<[1,9,1024]>,
Tensor<[1,9,1024]>,
|ttnn.add|aten::add.Tensor|5| +|411|Tensor<[9,4096]>,
Tensor<[9,4096]>,
|ttnn.add|aten::add.Tensor|4| +|412|Tensor<[1,9,4096]>,
Tensor<[1,9,4096]>,
|ttnn.add|aten::add.Tensor|5| +|413|Tensor<[1,64,9,9]>,
Tensor<[1,64,9,9]>,
|ttnn.add|aten::add.Tensor|4| +|414|Tensor<[9,16384]>,
Tensor<[9,16384]>,
|ttnn.add|aten::add.Tensor|4| +|415|Tensor<[1,9,16384]>,
Tensor<[1,9,16384]>,
|ttnn.add|aten::add.Tensor|5| +|416|Tensor<[1,2]>,
Tensor<[1,2]>,
|ttnn.add|aten::add.Tensor|4| +|417|Tensor<[1,14,128]>,
Tensor<[1,14,128]>,
|ttnn.add|aten::add.Tensor|5| +|418|Tensor<[1,14,1]>,
Tensor<[1,14,1]>,
|ttnn.add|aten::add.Tensor|4| +|419|Tensor<[14,768]>,
Tensor<[14,768]>,
|ttnn.add|aten::add.Tensor|4| +|420|Tensor<[1,12,14,14]>,
Tensor<[1,12,14,14]>,
|ttnn.add|aten::add.Tensor|4| +|421|Tensor<[1,14,768]>,
Tensor<[1,14,768]>,
|ttnn.add|aten::add.Tensor|5| +|422|Tensor<[14,3072]>,
Tensor<[14,3072]>,
|ttnn.add|aten::add.Tensor|4| +|423|Tensor<[1,14,3072]>,
Tensor<[1,14,3072]>,
|ttnn.add|aten::add.Tensor|5| +|424|Tensor<[14,2]>,
Tensor<[14,2]>,
|ttnn.add|aten::add.Tensor|4| +|425|Tensor<[1,50,768]>,
Tensor<[1,50,768]>,
|ttnn.add|aten::add.Tensor|5| +|426|Tensor<[1,50,1]>,
Tensor<[1,50,1]>,
|ttnn.add|aten::add.Tensor|4| +|427|Tensor<[50,768]>,
Tensor<[50,768]>,
|ttnn.add|aten::add.Tensor|4| +|428|Tensor<[50,3072]>,
Tensor<[50,3072]>,
|ttnn.add|aten::add.Tensor|4| +|429|Tensor<[2,7,512]>,
Tensor<[2,7,512]>,
|ttnn.add|aten::add.Tensor|4| +|430|Tensor<[2,7,1]>,
Tensor<[2,7,1]>,
|ttnn.add|aten::add.Tensor|4| +|431|Tensor<[2,1,7,7]>,
Tensor<[2,1,7,7]>,
|ttnn.add|aten::add.Tensor|5| +|432|Tensor<[14,512]>,
Tensor<[14,512]>,
|ttnn.add|aten::add.Tensor|4| +|433|Tensor<[2,8,7,7]>,
Tensor<[2,8,7,7]>,
|ttnn.add|aten::add.Tensor|4| +|434|Tensor<[14,2048]>,
Tensor<[14,2048]>,
|ttnn.add|aten::add.Tensor|4| +|435|Tensor<[2]>,
Tensor<[2]>,
|ttnn.add|aten::arange|4| +|436|Tensor<[1,197,1024]>,
Tensor<[1,197,1024]>,
|ttnn.add|aten::add.Tensor|4| +|437|Tensor<[197,1024]>,
Tensor<[197,1024]>,
|ttnn.add|aten::add.Tensor|4| +|438|Tensor<[27]>,
Tensor<[27]>,
|ttnn.add|aten::add.Tensor|4| +|439|Tensor<[27,1]>,
Tensor<[27,1]>,
|ttnn.add|aten::add.Tensor|4| +|440|Tensor<[1,16,27,27]>,
Tensor<[1,16,27,27]>,
|ttnn.add|aten::add.Tensor|5| +|441|Tensor<[196,196]>,
Tensor<[196,196]>,
|ttnn.add|aten::add.Tensor|4| +|442|Tensor<[1,16,197,197]>,
Tensor<[1,16,197,197]>,
|ttnn.add|aten::add.Tensor|5| +|443|Tensor<[197,4096]>,
Tensor<[197,4096]>,
|ttnn.add|aten::add.Tensor|4| +|444|Tensor<[1,1024]>,
Tensor<[1,1024]>,
|ttnn.add|aten::add.Tensor|4| +|445|Tensor<[197]>,
Tensor<[197]>,
|ttnn.add|aten::arange|4| +|446|Tensor<[1,197,4096]>,
Tensor<[1,197,4096]>,
|ttnn.add|aten::gelu|4| +|447|Tensor<[1,12,27,27]>,
Tensor<[1,12,27,27]>,
|ttnn.add|aten::add.Tensor|5| +|448|Tensor<[1,12,197,197]>,
Tensor<[1,12,197,197]>,
|ttnn.add|aten::add.Tensor|5| +|449|Tensor<[1,64]>,
Tensor<[1,64]>,
|ttnn.add|aten::add.Tensor|4| +|450|Tensor<[1,12]>,
Tensor<[1,12]>,
|ttnn.add|aten::add.Tensor|4| +|451|Tensor<[1,784]>,
Tensor<[1,784]>,
|ttnn.add|aten::add.Tensor|4| diff --git a/docs/ops/stablehlo/stablehlo.and.md b/docs/ops/stablehlo/stablehlo.and.md new file mode 100644 index 00000000..10196b54 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.and.md @@ -0,0 +1,8 @@ + +### stablehlo.and::ttnn.and + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[19]>,
Tensor<[19]>,
|ttnn.and|aten::logical_and|5| +|1|Tensor<[197]>,
Tensor<[197]>,
|ttnn.and|aten::logical_and|5| diff --git a/docs/ops/stablehlo/stablehlo.broadcast_in_dim.md b/docs/ops/stablehlo/stablehlo.broadcast_in_dim.md new file mode 100644 index 00000000..1c4c9266 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.broadcast_in_dim.md @@ -0,0 +1,869 @@ + +### stablehlo.broadcast_in_dim + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,32,32,32]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|1|Tensor<[1,32,32,1]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|2|Scalar,
dims: []
||aten::_safe_softmax|4| +|3|Tensor<[1,1,32,32]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|4|Tensor<[1,1,1,32]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|5|Tensor<[1,32,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|6|Tensor<[32]>,
dims: [0]
||aten::arange|4| +|7|Tensor<[1,1,32]>,
dims: [0, 1, 2]
||aten::bmm|4| +|8|Tensor<[32,128,32]>,
dims: [0, 1, 2]
||aten::bmm|4| +|9|Tensor<[32,32,128]>,
dims: [0, 1, 2]
||aten::bmm|4| +|10|Tensor<[32]>,
dims: [1]
||aten::gt.Tensor|4| +|11|Tensor<[32,1]>,
dims: [0, 1]
||aten::gt.Tensor|4| +|12|Tensor<[1,32,32,128]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|13|Tensor<[1,32,128,32]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|14|Tensor<[1,32,128]>,
dims: [0, 1, 2]
||aten::mul.Tensor|4| +|15|Tensor<[1,32,4096]>,
dims: [0, 1, 2]
||aten::mul.Tensor|4| +|16|Tensor<[4096]>,
dims: [2]
||aten::mul.Tensor|4| +|17|Tensor<[1,1,32,128]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|18|Tensor<[1,32]>,
dims: [0, 1]
||aten::triu|4| +|19|Tensor<[32,32]>,
dims: [0, 1]
||aten::triu|4| +|20|Tensor<[1,12,7,7]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|21|Tensor<[1,12,7,1]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|22|Tensor<[7]>,
dims: [0]
||aten::add.Tensor|4| +|23|Tensor<[1,7,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|24|Tensor<[1,7,768]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|25|Tensor<[768]>,
dims: [2]
||aten::add.Tensor|4| +|26|Tensor<[7,2304]>,
dims: [0, 1]
||aten::add.Tensor|4| +|27|Tensor<[2304]>,
dims: [1]
||aten::add.Tensor|4| +|28|Tensor<[1,1,7,7]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|29|Tensor<[7,768]>,
dims: [0, 1]
||aten::add.Tensor|4| +|30|Tensor<[768]>,
dims: [1]
||aten::add.Tensor|4| +|31|Tensor<[7,3072]>,
dims: [0, 1]
||aten::add.Tensor|4| +|32|Tensor<[3072]>,
dims: [1]
||aten::add.Tensor|4| +|33|Tensor<[1,7,3072]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|34|Tensor<[1]>,
dims: [0]
||aten::arange|4| +|35|Tensor<[12,64,7]>,
dims: [0, 1, 2]
||aten::bmm|4| +|36|Tensor<[12,7,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|37|Tensor<[1,7]>,
dims: [0, 1]
||aten::eq.Scalar|4| +|38|Tensor<[1,1,1,7]>,
dims: [0, 1, 2, 3]
||aten::expand|4| +|39|Tensor<[7]>,
dims: [1]
||aten::lt.Tensor|4| +|40|Tensor<[7,1]>,
dims: [0, 1]
||aten::lt.Tensor|4| +|41|Tensor<[1,12,7,64]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|42|Tensor<[1,12,64,7]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|43|Tensor<[2304]>,
dims: [0]
||aten::mul.Tensor|4| +|44|Tensor<[768]>,
dims: [0]
||aten::mul.Tensor|4| +|45|Tensor<[3072]>,
dims: [0]
||aten::mul.Tensor|4| +|46|Tensor<[7,7]>,
dims: [0, 1]
||aten::where.self|4| +|47|Tensor<[1,32,112,112]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|48|Tensor<[32,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|49|Tensor<[64]>,
dims: [0]
||aten::add.Tensor|4| +|50|Tensor<[1,64,112,112]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|51|Tensor<[64,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|52|Tensor<[1,64,56,56]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|53|Tensor<[128]>,
dims: [0]
||aten::add.Tensor|4| +|54|Tensor<[1,128,56,56]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|55|Tensor<[128,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|56|Tensor<[1,128,28,28]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|57|Tensor<[256]>,
dims: [0]
||aten::add.Tensor|4| +|58|Tensor<[1,256,28,28]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|59|Tensor<[256,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|60|Tensor<[512]>,
dims: [0]
||aten::add.Tensor|4| +|61|Tensor<[1,512,28,28]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|62|Tensor<[512,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|63|Tensor<[1,19,28,28]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|64|Tensor<[19,1,1]>,
dims: [1, 2, 3]
||aten::convolution|4| +|65|Tensor<[1,38,28,28]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|66|Tensor<[38,1,1]>,
dims: [1, 2, 3]
||aten::convolution|4| +|67|Tensor<[256,512]>,
dims: [0, 1]
||aten::add.Tensor|4| +|68|Tensor<[512]>,
dims: [1]
||aten::add.Tensor|4| +|69|Tensor<[1,256,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|70|Tensor<[1,256,512]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|71|Tensor<[512]>,
dims: [2]
||aten::add.Tensor|4| +|72|Tensor<[256,256]>,
dims: [0, 1]
||aten::add.Tensor|4| +|73|Tensor<[256]>,
dims: [1]
||aten::add.Tensor|4| +|74|Tensor<[1,1000]>,
dims: [0, 1]
||aten::add.Tensor|4| +|75|Tensor<[1000]>,
dims: [1]
||aten::add.Tensor|4| +|76|Tensor<[1,1024,512]>,
dims: [0, 1, 2]
||aten::convolution|4| +|77|Tensor<[1024,1]>,
dims: [1, 2]
||aten::convolution|4| +|78|Tensor<[256,1]>,
dims: [1, 2]
||aten::convolution|4| +|79|Tensor<[1,512]>,
dims: [0, 1]
||aten::mean.dim|4| +|80|Tensor<[1000]>,
dims: [0]
||aten::mul.Tensor|4| +|81|Tensor<[8,920,920]>,
dims: [0, 1, 2]
||aten::_softmax|4| +|82|Tensor<[8,920,1]>,
dims: [0, 1, 2]
||aten::_softmax|4| +|83|Tensor<[8,100,100]>,
dims: [0, 1, 2]
||aten::_softmax|4| +|84|Tensor<[8,100,1]>,
dims: [0, 1, 2]
||aten::_softmax|4| +|85|Tensor<[8,100,920]>,
dims: [0, 1, 2]
||aten::_softmax|4| +|86|Tensor<[1,64,1,1]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|87|Tensor<[1,64,360,640]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|88|Tensor<[1,64,180,320]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|89|Tensor<[1,256,1,1]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|90|Tensor<[1,256,180,320]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|91|Tensor<[1,128,1,1]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|92|Tensor<[1,128,180,320]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|93|Tensor<[1,128,90,160]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|94|Tensor<[1,512,1,1]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|95|Tensor<[1,512,90,160]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|96|Tensor<[1,256,90,160]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|97|Tensor<[1,256,45,80]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|98|Tensor<[1,1024,1,1]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|99|Tensor<[1,1024,45,80]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|100|Tensor<[1,512,45,80]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|101|Tensor<[1,512,23,40]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|102|Tensor<[1,2048,1,1]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|103|Tensor<[1,2048,23,40]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|104|Tensor<[23]>,
dims: [0]
||aten::add.Tensor|4| +|105|Tensor<[40]>,
dims: [0]
||aten::add.Tensor|4| +|106|Tensor<[1,1,40]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|107|Tensor<[1,23,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|108|Tensor<[920,1,256]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|109|Tensor<[256]>,
dims: [2]
||aten::add.Tensor|4| +|110|Tensor<[920,256]>,
dims: [0, 1]
||aten::add.Tensor|4| +|111|Tensor<[920,1,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|112|Tensor<[920,2048]>,
dims: [0, 1]
||aten::add.Tensor|4| +|113|Tensor<[2048]>,
dims: [1]
||aten::add.Tensor|4| +|114|Tensor<[100,256]>,
dims: [0, 1]
||aten::add.Tensor|4| +|115|Tensor<[100,1,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|116|Tensor<[100,1,256]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|117|Tensor<[100,2048]>,
dims: [0, 1]
||aten::add.Tensor|4| +|118|Tensor<[6,1,100,92]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|119|Tensor<[92]>,
dims: [3]
||aten::add.Tensor|4| +|120|Tensor<[6,1,100,256]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|121|Tensor<[256]>,
dims: [3]
||aten::add.Tensor|4| +|122|Tensor<[6,1,100,4]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|123|Tensor<[4]>,
dims: [3]
||aten::add.Tensor|4| +|124|Tensor<[8,32,920]>,
dims: [0, 1, 2]
||aten::baddbmm|4| +|125|Tensor<[8,1,920]>,
dims: [0, 1, 2]
||aten::baddbmm|4| +|126|Tensor<[920,256,256]>,
dims: [0, 1, 2]
||aten::bmm|4| +|127|Tensor<[8,920,32]>,
dims: [0, 1, 2]
||aten::bmm|4| +|128|Tensor<[8,32,100]>,
dims: [0, 1, 2]
||aten::bmm|4| +|129|Tensor<[8,100,32]>,
dims: [0, 1, 2]
||aten::bmm|4| +|130|Tensor<[6,256,92]>,
dims: [0, 1, 2]
||aten::bmm|4| +|131|Tensor<[6,256,256]>,
dims: [0, 1, 2]
||aten::bmm|4| +|132|Tensor<[1,256,23,40]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|133|Tensor<[1,23,40]>,
dims: [0, 1, 2]
||aten::div.Tensor|4| +|134|Tensor<[1,23,40,1]>,
dims: [0, 1, 2, 3]
||aten::div.Tensor|4| +|135|Tensor<[128]>,
dims: [3]
||aten::div.Tensor|4| +|136|Tensor<[256,256]>,
dims: [1, 2]
||aten::expand|5| +|137|Tensor<[1,1,1,920]>,
dims: [0, 1, 2, 3]
||aten::expand|5| +|138|Tensor<[256,92]>,
dims: [2, 3]
||aten::expand|5| +|139|Tensor<[256,256]>,
dims: [2, 3]
||aten::expand|5| +|140|Tensor<[1,1,1,1]>,
dims: [0, 1, 2, 3]
||aten::index.Tensor|4| +|141|Tensor<[1,1,1]>,
dims: [1, 2, 3]
||aten::index.Tensor|4| +|142|Tensor<[23,1]>,
dims: [2, 3]
||aten::index.Tensor|4| +|143|Tensor<[40]>,
dims: [3]
||aten::index.Tensor|4| +|144|Tensor<[2048]>,
dims: [0]
||aten::mul.Tensor|4| +|145|Tensor<[1,920]>,
dims: [0, 1]
||aten::where.self|4| +|146|Tensor<[1,12,10,10]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|147|Tensor<[1,12,10,1]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|148|Tensor<[1,10]>,
dims: [0, 1]
||aten::add.Tensor|5| +|149|Tensor<[1,10,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|150|Tensor<[1,10,768]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|151|Tensor<[10,768]>,
dims: [0, 1]
||aten::add.Tensor|4| +|152|Tensor<[1,1,10,10]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|153|Tensor<[10,3072]>,
dims: [0, 1]
||aten::add.Tensor|4| +|154|Tensor<[10,250002]>,
dims: [0, 1]
||aten::add.Tensor|4| +|155|Tensor<[250002]>,
dims: [1]
||aten::add.Tensor|4| +|156|Tensor<[12,64,10]>,
dims: [0, 1, 2]
||aten::bmm|4| +|157|Tensor<[12,10,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|158|Tensor<[1,1,1,10]>,
dims: [0, 1, 2, 3]
||aten::expand|4| +|159|Tensor<[1,12,10,64]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|160|Tensor<[1,12,64,10]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|161|Tensor<[250002]>,
dims: [0]
||aten::mul.Tensor|4| +|162|Tensor<[1,8,4096,4096]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|163|Tensor<[1,8,4096,1]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|164|Tensor<[1,8,4096,9]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|165|Tensor<[1,8,1024,1024]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|166|Tensor<[1,8,1024,1]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|167|Tensor<[1,8,1024,9]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|168|Tensor<[1,8,256,256]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|169|Tensor<[1,8,256,1]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|170|Tensor<[1,8,256,9]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|171|Tensor<[1,8,64,64]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|172|Tensor<[1,8,64,1]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|173|Tensor<[1,8,64,9]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|174|Tensor<[1,1280]>,
dims: [0, 1]
||aten::add.Tensor|4| +|175|Tensor<[1280]>,
dims: [1]
||aten::add.Tensor|4| +|176|Tensor<[1,32,1,1]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|177|Tensor<[1,320,64,64]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|178|Tensor<[1,320,1,1]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|179|Tensor<[1,320]>,
dims: [0, 1]
||aten::add.Tensor|4| +|180|Tensor<[320]>,
dims: [1]
||aten::add.Tensor|4| +|181|Tensor<[1,4096,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|182|Tensor<[1,4096,320]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|183|Tensor<[320]>,
dims: [2]
||aten::add.Tensor|4| +|184|Tensor<[4096,320]>,
dims: [0, 1]
||aten::add.Tensor|4| +|185|Tensor<[4096,2560]>,
dims: [0, 1]
||aten::add.Tensor|4| +|186|Tensor<[2560]>,
dims: [1]
||aten::add.Tensor|4| +|187|Tensor<[1,320,32,32]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|188|Tensor<[1,640]>,
dims: [0, 1]
||aten::add.Tensor|4| +|189|Tensor<[640]>,
dims: [1]
||aten::add.Tensor|4| +|190|Tensor<[1,640,32,32]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|191|Tensor<[1,640,1,1]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|192|Tensor<[1,1024,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|193|Tensor<[1,1024,640]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|194|Tensor<[640]>,
dims: [2]
||aten::add.Tensor|4| +|195|Tensor<[1024,640]>,
dims: [0, 1]
||aten::add.Tensor|4| +|196|Tensor<[1024,5120]>,
dims: [0, 1]
||aten::add.Tensor|4| +|197|Tensor<[5120]>,
dims: [1]
||aten::add.Tensor|4| +|198|Tensor<[1,640,16,16]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|199|Tensor<[1,1280,16,16]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|200|Tensor<[1,1280,1,1]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|201|Tensor<[1,256,1280]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|202|Tensor<[1280]>,
dims: [2]
||aten::add.Tensor|4| +|203|Tensor<[256,1280]>,
dims: [0, 1]
||aten::add.Tensor|4| +|204|Tensor<[256,10240]>,
dims: [0, 1]
||aten::add.Tensor|4| +|205|Tensor<[10240]>,
dims: [1]
||aten::add.Tensor|4| +|206|Tensor<[1,1280,8,8]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|207|Tensor<[1,64,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|208|Tensor<[1,64,1280]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|209|Tensor<[64,1280]>,
dims: [0, 1]
||aten::add.Tensor|4| +|210|Tensor<[64,10240]>,
dims: [0, 1]
||aten::add.Tensor|4| +|211|Tensor<[1,2560,8,8]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|212|Tensor<[1,2560,1,1]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|213|Tensor<[16]>,
dims: [0]
||aten::add.Tensor|4| +|214|Tensor<[1,2560,16,16]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|215|Tensor<[1,1920,16,16]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|216|Tensor<[1,1920,1,1]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|217|Tensor<[1,1920,32,32]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|218|Tensor<[1,1280,32,32]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|219|Tensor<[1,960,32,32]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|220|Tensor<[1,960,1,1]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|221|Tensor<[1,960,64,64]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|222|Tensor<[1,640,64,64]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|223|Tensor<[160]>,
dims: [0]
||aten::arange.start|4| +|224|Tensor<[8,40,4096]>,
dims: [0, 1, 2]
||aten::bmm|4| +|225|Tensor<[8,4096,40]>,
dims: [0, 1, 2]
||aten::bmm|4| +|226|Tensor<[8,40,9]>,
dims: [0, 1, 2]
||aten::bmm|4| +|227|Tensor<[8,9,40]>,
dims: [0, 1, 2]
||aten::bmm|4| +|228|Tensor<[8,80,1024]>,
dims: [0, 1, 2]
||aten::bmm|4| +|229|Tensor<[8,1024,80]>,
dims: [0, 1, 2]
||aten::bmm|4| +|230|Tensor<[8,80,9]>,
dims: [0, 1, 2]
||aten::bmm|4| +|231|Tensor<[8,9,80]>,
dims: [0, 1, 2]
||aten::bmm|4| +|232|Tensor<[8,160,256]>,
dims: [0, 1, 2]
||aten::bmm|4| +|233|Tensor<[8,256,160]>,
dims: [0, 1, 2]
||aten::bmm|4| +|234|Tensor<[8,160,9]>,
dims: [0, 1, 2]
||aten::bmm|4| +|235|Tensor<[8,9,160]>,
dims: [0, 1, 2]
||aten::bmm|4| +|236|Tensor<[8,160,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|237|Tensor<[8,64,160]>,
dims: [0, 1, 2]
||aten::bmm|4| +|238|Tensor<[320,1,1]>,
dims: [1, 2, 3]
||aten::convolution|4| +|239|Tensor<[640,1,1]>,
dims: [1, 2, 3]
||aten::convolution|4| +|240|Tensor<[1280,1,1]>,
dims: [1, 2, 3]
||aten::convolution|4| +|241|Tensor<[1,4,64,64]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|242|Tensor<[4,1,1]>,
dims: [1, 2, 3]
||aten::convolution|4| +|243|Tensor<[1280]>,
dims: [0]
||aten::index.Tensor|4| +|244|Tensor<[16,1]>,
dims: [2, 3]
||aten::index.Tensor|4| +|245|Tensor<[16]>,
dims: [3]
||aten::index.Tensor|4| +|246|Tensor<[32,1]>,
dims: [2, 3]
||aten::index.Tensor|4| +|247|Tensor<[32]>,
dims: [3]
||aten::index.Tensor|4| +|248|Tensor<[640]>,
dims: [0]
||aten::index.Tensor|4| +|249|Tensor<[64,1]>,
dims: [2, 3]
||aten::index.Tensor|4| +|250|Tensor<[64]>,
dims: [3]
||aten::index.Tensor|4| +|251|Tensor<[1,8,4096,40]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|252|Tensor<[1,8,40,4096]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|253|Tensor<[1,8,40,9]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|254|Tensor<[1,8,1024,80]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|255|Tensor<[1,8,80,1024]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|256|Tensor<[1,8,80,9]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|257|Tensor<[1,8,256,160]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|258|Tensor<[1,8,160,256]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|259|Tensor<[1,8,160,9]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|260|Tensor<[1,8,64,160]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|261|Tensor<[1,8,160,64]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|262|Tensor<[1,1]>,
dims: [0, 1]
||aten::mul.Tensor|4| +|263|Tensor<[1,160]>,
dims: [0, 1]
||aten::mul.Tensor|4| +|264|Tensor<[1,32,10,4096]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|265|Tensor<[320]>,
dims: [0]
||aten::mul.Tensor|4| +|266|Tensor<[2560]>,
dims: [0]
||aten::mul.Tensor|4| +|267|Tensor<[1,32,10,1024]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|268|Tensor<[1,32,20,1024]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|269|Tensor<[5120]>,
dims: [0]
||aten::mul.Tensor|4| +|270|Tensor<[1,32,20,256]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|271|Tensor<[1,32,40,256]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|272|Tensor<[10240]>,
dims: [0]
||aten::mul.Tensor|4| +|273|Tensor<[1,32,40,64]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|274|Tensor<[1,32,80,64]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|275|Tensor<[1,32,80,256]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|276|Tensor<[1,32,60,256]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|277|Tensor<[1,32,60,1024]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|278|Tensor<[1,32,40,1024]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|279|Tensor<[1,32,30,1024]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|280|Tensor<[1,32,30,4096]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|281|Tensor<[1,32,20,4096]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|282|Tensor<[1,12,25,25]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|283|Tensor<[1,12,25,1]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|284|Tensor<[1,25,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|285|Tensor<[1,25,768]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|286|Tensor<[25,768]>,
dims: [0, 1]
||aten::add.Tensor|4| +|287|Tensor<[1,1,25,25]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|288|Tensor<[25,3072]>,
dims: [0, 1]
||aten::add.Tensor|4| +|289|Tensor<[25,2]>,
dims: [0, 1]
||aten::add.Tensor|4| +|290|Tensor<[2]>,
dims: [1]
||aten::add.Tensor|4| +|291|Tensor<[1]>,
dims: [1]
||aten::add.Tensor|4| +|292|Tensor<[12,64,25]>,
dims: [0, 1, 2]
||aten::bmm|4| +|293|Tensor<[12,25,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|294|Tensor<[1,1,1,25]>,
dims: [0, 1, 2, 3]
||aten::expand|4| +|295|Tensor<[1,12,25,64]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|296|Tensor<[1,12,64,25]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|297|Tensor<[2]>,
dims: [0]
||aten::mul.Tensor|4| +|298|Tensor<[1,3,1445,1445]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|299|Tensor<[1,3,1445,1]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|300|Tensor<[1,1445,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|301|Tensor<[1,1445,192]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|302|Tensor<[192]>,
dims: [2]
||aten::add.Tensor|4| +|303|Tensor<[1445,192]>,
dims: [0, 1]
||aten::add.Tensor|4| +|304|Tensor<[192]>,
dims: [1]
||aten::add.Tensor|4| +|305|Tensor<[1445,768]>,
dims: [0, 1]
||aten::add.Tensor|4| +|306|Tensor<[100,192]>,
dims: [0, 1]
||aten::add.Tensor|4| +|307|Tensor<[100,92]>,
dims: [0, 1]
||aten::add.Tensor|4| +|308|Tensor<[92]>,
dims: [1]
||aten::add.Tensor|4| +|309|Tensor<[100,4]>,
dims: [0, 1]
||aten::add.Tensor|4| +|310|Tensor<[4]>,
dims: [1]
||aten::add.Tensor|4| +|311|Tensor<[3,64,1445]>,
dims: [0, 1, 2]
||aten::bmm|4| +|312|Tensor<[3,1445,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|313|Tensor<[1,192,32,42]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|314|Tensor<[192,1,1]>,
dims: [1, 2, 3]
||aten::convolution|4| +|315|Tensor<[1,3,1445,64]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|316|Tensor<[1,3,64,1445]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|317|Tensor<[192]>,
dims: [0]
||aten::mul.Tensor|4| +|318|Tensor<[92]>,
dims: [0]
||aten::mul.Tensor|4| +|319|Tensor<[4]>,
dims: [0]
||aten::mul.Tensor|4| +|320|Tensor<[1,256,14,14]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|321|Tensor<[1,512,7,7]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|322|Tensor<[1,12,8,8]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|323|Tensor<[1,12,8,1]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|324|Tensor<[1,8,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|325|Tensor<[1,8,768]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|326|Tensor<[1,1,1,8]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|327|Tensor<[1,768]>,
dims: [0, 1]
||aten::add.Tensor|4| +|328|Tensor<[1,3]>,
dims: [0, 1]
||aten::add.Tensor|4| +|329|Tensor<[3]>,
dims: [1]
||aten::add.Tensor|4| +|330|Tensor<[12,64,8]>,
dims: [0, 1, 2]
||aten::bmm|4| +|331|Tensor<[12,8,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|332|Tensor<[1,768,8]>,
dims: [0, 1, 2]
||aten::convolution|4| +|333|Tensor<[768,1]>,
dims: [1, 2]
||aten::convolution|4| +|334|Tensor<[1,3072,8]>,
dims: [0, 1, 2]
||aten::convolution|4| +|335|Tensor<[3072,1]>,
dims: [1, 2]
||aten::convolution|4| +|336|Tensor<[3]>,
dims: [0]
||aten::mul.Tensor|4| +|337|Tensor<[1,8,256,2048]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|338|Tensor<[1,8,2048,256]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|339|Tensor<[1,8,2048,1]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|340|Tensor<[1,2048,768]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|341|Tensor<[2048,768]>,
dims: [1, 2]
||aten::add.Tensor|4| +|342|Tensor<[1,2048,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|343|Tensor<[2048,256]>,
dims: [0, 1]
||aten::add.Tensor|4| +|344|Tensor<[2048,1280]>,
dims: [0, 1]
||aten::add.Tensor|4| +|345|Tensor<[1,1,1,2048]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|346|Tensor<[256,768]>,
dims: [0, 1]
||aten::add.Tensor|4| +|347|Tensor<[2048,768]>,
dims: [0, 1]
||aten::add.Tensor|4| +|348|Tensor<[2048,262]>,
dims: [0, 1]
||aten::add.Tensor|4| +|349|Tensor<[262]>,
dims: [1]
||aten::add.Tensor|4| +|350|Tensor<[8,32,2048]>,
dims: [0, 1, 2]
||aten::bmm|4| +|351|Tensor<[8,2048,160]>,
dims: [0, 1, 2]
||aten::bmm|4| +|352|Tensor<[8,32,256]>,
dims: [0, 1, 2]
||aten::bmm|4| +|353|Tensor<[8,256,96]>,
dims: [0, 1, 2]
||aten::bmm|4| +|354|Tensor<[256,1280]>,
dims: [1, 2]
||aten::expand|5| +|355|Tensor<[1,256,56,56]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|356|Tensor<[1024]>,
dims: [0]
||aten::add.Tensor|4| +|357|Tensor<[1,1024,14,14]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|358|Tensor<[1024,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|359|Tensor<[1,512,14,14]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|360|Tensor<[1,2048,7,7]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|361|Tensor<[2048,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|362|Tensor<[1,12,201,201]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|363|Tensor<[1,12,201,1]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|364|Tensor<[12]>,
dims: [0]
||aten::add.Tensor|4| +|365|Tensor<[1,201,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|366|Tensor<[1,201,768]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|367|Tensor<[201,768]>,
dims: [0, 1]
||aten::add.Tensor|4| +|368|Tensor<[1,1,1,201]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|369|Tensor<[201,3072]>,
dims: [0, 1]
||aten::add.Tensor|4| +|370|Tensor<[1,1536]>,
dims: [0, 1]
||aten::add.Tensor|4| +|371|Tensor<[1536]>,
dims: [1]
||aten::add.Tensor|4| +|372|Tensor<[1,3129]>,
dims: [0, 1]
||aten::add.Tensor|4| +|373|Tensor<[3129]>,
dims: [1]
||aten::add.Tensor|4| +|374|Tensor<[12,64,201]>,
dims: [0, 1, 2]
||aten::bmm|4| +|375|Tensor<[12,201,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|376|Tensor<[1,768,12,16]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|377|Tensor<[768,1,1]>,
dims: [1, 2, 3]
||aten::convolution|4| +|378|Tensor<[12,1]>,
dims: [0, 1]
||aten::expand|4| +|379|Tensor<[1,16]>,
dims: [0, 1]
||aten::expand|4| +|380|Tensor<[12,1]>,
dims: [2, 3]
||aten::index.Tensor|4| +|381|Tensor<[1536]>,
dims: [0]
||aten::mul.Tensor|4| +|382|Tensor<[3129]>,
dims: [0]
||aten::mul.Tensor|4| +|383|Tensor<[1,192]>,
dims: [0, 1]
||aten::rsub.Scalar|4| +|384|Tensor<[1,128]>,
dims: [0, 1]
||aten::add.Tensor|4| +|385|Tensor<[128]>,
dims: [1]
||aten::add.Tensor|4| +|386|Tensor<[10]>,
dims: [1]
||aten::add.Tensor|4| +|387|Tensor<[1,32,26,26]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|388|Tensor<[1,64,24,24]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|389|Tensor<[10]>,
dims: [0]
||aten::mul.Tensor|4| +|390|Tensor<[16,19,19]>,
dims: [0, 1, 2]
||aten::_softmax|4| +|391|Tensor<[16,19,1]>,
dims: [0, 1, 2]
||aten::_softmax|4| +|392|Tensor<[19]>,
dims: [0]
||aten::add.Tensor|4| +|393|Tensor<[1,19]>,
dims: [0, 1]
||aten::add.Tensor|4| +|394|Tensor<[1,19,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|395|Tensor<[1,19,1024]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|396|Tensor<[1024]>,
dims: [2]
||aten::add.Tensor|4| +|397|Tensor<[19,1024]>,
dims: [0, 1]
||aten::add.Tensor|4| +|398|Tensor<[1024]>,
dims: [1]
||aten::add.Tensor|4| +|399|Tensor<[1,16,19,19]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|400|Tensor<[1,1,19,19]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|401|Tensor<[19,4096]>,
dims: [0, 1]
||aten::add.Tensor|4| +|402|Tensor<[4096]>,
dims: [1]
||aten::add.Tensor|4| +|403|Tensor<[16,64,19]>,
dims: [0, 1, 2]
||aten::bmm|4| +|404|Tensor<[16,19,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|405|Tensor<[1,1,1,19]>,
dims: [0, 1, 2, 3]
||aten::expand|4| +|406|Tensor<[19]>,
dims: [1]
||aten::lt.Tensor|4| +|407|Tensor<[19,1]>,
dims: [0, 1]
||aten::lt.Tensor|4| +|408|Tensor<[4096]>,
dims: [0]
||aten::mul.Tensor|4| +|409|Tensor<[19,256008]>,
dims: [0, 1]
||aten::sub.Tensor|4| +|410|Tensor<[19,19]>,
dims: [0, 1]
||aten::where.self|4| +|411|Tensor<[14]>,
dims: [0]
||aten::add.Tensor|4| +|412|Tensor<[1,14,56,56]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|413|Tensor<[14,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|414|Tensor<[24]>,
dims: [0]
||aten::add.Tensor|4| +|415|Tensor<[1,24,56,56]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|416|Tensor<[24,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|417|Tensor<[1,40,56,56]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|418|Tensor<[40,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|419|Tensor<[68]>,
dims: [0]
||aten::add.Tensor|4| +|420|Tensor<[1,68,56,56]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|421|Tensor<[68,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|422|Tensor<[1,16,28,28]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|423|Tensor<[16,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|424|Tensor<[28]>,
dims: [0]
||aten::add.Tensor|4| +|425|Tensor<[1,28,28,28]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|426|Tensor<[28,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|427|Tensor<[46]>,
dims: [0]
||aten::add.Tensor|4| +|428|Tensor<[1,46,28,28]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|429|Tensor<[46,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|430|Tensor<[78]>,
dims: [0]
||aten::add.Tensor|4| +|431|Tensor<[1,78,28,28]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|432|Tensor<[78,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|433|Tensor<[134]>,
dims: [0]
||aten::add.Tensor|4| +|434|Tensor<[1,134,28,28]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|435|Tensor<[134,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|436|Tensor<[20]>,
dims: [0]
||aten::add.Tensor|4| +|437|Tensor<[1,20,28,28]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|438|Tensor<[20,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|439|Tensor<[34]>,
dims: [0]
||aten::add.Tensor|4| +|440|Tensor<[1,34,28,28]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|441|Tensor<[34,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|442|Tensor<[58]>,
dims: [0]
||aten::add.Tensor|4| +|443|Tensor<[1,58,28,28]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|444|Tensor<[58,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|445|Tensor<[98]>,
dims: [0]
||aten::add.Tensor|4| +|446|Tensor<[1,98,28,28]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|447|Tensor<[98,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|448|Tensor<[168]>,
dims: [0]
||aten::add.Tensor|4| +|449|Tensor<[1,168,28,28]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|450|Tensor<[168,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|451|Tensor<[1,320,28,28]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|452|Tensor<[1,40,14,14]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|453|Tensor<[1,68,14,14]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|454|Tensor<[116]>,
dims: [0]
||aten::add.Tensor|4| +|455|Tensor<[1,116,14,14]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|456|Tensor<[116,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|457|Tensor<[196]>,
dims: [0]
||aten::add.Tensor|4| +|458|Tensor<[1,196,14,14]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|459|Tensor<[196,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|460|Tensor<[334]>,
dims: [0]
||aten::add.Tensor|4| +|461|Tensor<[1,334,14,14]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|462|Tensor<[334,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|463|Tensor<[1,640,14,14]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|464|Tensor<[1,160,7,7]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|465|Tensor<[160,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|466|Tensor<[272]>,
dims: [0]
||aten::add.Tensor|4| +|467|Tensor<[1,272,7,7]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|468|Tensor<[272,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|469|Tensor<[462]>,
dims: [0]
||aten::add.Tensor|4| +|470|Tensor<[1,462,7,7]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|471|Tensor<[462,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|472|Tensor<[1,1024,7,7]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|473|Tensor<[1,32,512,512]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|474|Tensor<[1,64,256,256]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|475|Tensor<[1,32,256,256]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|476|Tensor<[1,128,128,128]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|477|Tensor<[1,64,128,128]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|478|Tensor<[1,256,64,64]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|479|Tensor<[1,128,64,64]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|480|Tensor<[1,512,32,32]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|481|Tensor<[1,256,32,32]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|482|Tensor<[1,1024,16,16]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|483|Tensor<[1,512,16,16]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|484|Tensor<[1,256,16,16]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|485|Tensor<[1,128,32,32]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|486|Tensor<[1,255,16,16]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|487|Tensor<[255,1,1]>,
dims: [1, 2, 3]
||aten::convolution|4| +|488|Tensor<[1,255,32,32]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|489|Tensor<[1,255,64,64]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|490|Tensor<[1,1,256,256]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|491|Tensor<[1,4,14,14]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|492|Tensor<[1,16,14,14]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|493|Tensor<[1,1,28,28]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|494|Tensor<[1,16,32,32]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|495|Tensor<[1,16,32,1]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|496|Tensor<[1,32,1536]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|497|Tensor<[1536]>,
dims: [2]
||aten::add.Tensor|4| +|498|Tensor<[32,4608]>,
dims: [0, 1]
||aten::add.Tensor|4| +|499|Tensor<[4608]>,
dims: [1]
||aten::add.Tensor|4| +|500|Tensor<[32,1536]>,
dims: [0, 1]
||aten::add.Tensor|4| +|501|Tensor<[32,6144]>,
dims: [0, 1]
||aten::add.Tensor|4| +|502|Tensor<[6144]>,
dims: [1]
||aten::add.Tensor|4| +|503|Tensor<[1,32,6144]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|504|Tensor<[16,96,32]>,
dims: [0, 1, 2]
||aten::baddbmm|4| +|505|Tensor<[16,32,32]>,
dims: [0, 1, 2]
||aten::baddbmm|4| +|506|Tensor<[16,1,32]>,
dims: [0, 1, 2]
||aten::baddbmm|4| +|507|Tensor<[16,32,96]>,
dims: [0, 1, 2]
||aten::bmm|4| +|508|Tensor<[16,1]>,
dims: [1, 2]
||aten::mul.Tensor|4| +|509|Tensor<[4608]>,
dims: [0]
||aten::mul.Tensor|4| +|510|Tensor<[6144]>,
dims: [0]
||aten::mul.Tensor|4| +|511|Tensor<[1,12,16,16]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|512|Tensor<[1,12,16,1]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|513|Tensor<[1,16,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|514|Tensor<[1,16,768]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|515|Tensor<[16,768]>,
dims: [0, 1]
||aten::add.Tensor|4| +|516|Tensor<[1,1,16,16]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|517|Tensor<[16,3072]>,
dims: [0, 1]
||aten::add.Tensor|4| +|518|Tensor<[12,64,16]>,
dims: [0, 1, 2]
||aten::bmm|4| +|519|Tensor<[12,16,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|520|Tensor<[1,1,1,16]>,
dims: [0, 1, 2, 3]
||aten::expand|4| +|521|Tensor<[1,12,16,64]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|522|Tensor<[1,12,64,16]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|523|Tensor<[1,64,224,224]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|524|Tensor<[1,128,112,112]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|525|Tensor<[1,1,224,224]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|526|Tensor<[1,1,19200,300]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|527|Tensor<[1,1,19200,1]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|528|Tensor<[1,2,4800,300]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|529|Tensor<[1,2,4800,1]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|530|Tensor<[1,5,1200,300]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|531|Tensor<[1,5,1200,1]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|532|Tensor<[1,8,300,300]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|533|Tensor<[1,8,300,1]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|534|Tensor<[1,19200,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|535|Tensor<[1,19200,64]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|536|Tensor<[64]>,
dims: [2]
||aten::add.Tensor|4| +|537|Tensor<[19200,64]>,
dims: [0, 1]
||aten::add.Tensor|4| +|538|Tensor<[64]>,
dims: [1]
||aten::add.Tensor|4| +|539|Tensor<[1,300,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|540|Tensor<[1,300,64]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|541|Tensor<[300,64]>,
dims: [0, 1]
||aten::add.Tensor|4| +|542|Tensor<[19200,256]>,
dims: [0, 1]
||aten::add.Tensor|4| +|543|Tensor<[1,4800,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|544|Tensor<[1,4800,128]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|545|Tensor<[128]>,
dims: [2]
||aten::add.Tensor|4| +|546|Tensor<[4800,128]>,
dims: [0, 1]
||aten::add.Tensor|4| +|547|Tensor<[1,300,128]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|548|Tensor<[300,128]>,
dims: [0, 1]
||aten::add.Tensor|4| +|549|Tensor<[4800,512]>,
dims: [0, 1]
||aten::add.Tensor|4| +|550|Tensor<[1,1200,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|551|Tensor<[1,1200,320]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|552|Tensor<[1200,320]>,
dims: [0, 1]
||aten::add.Tensor|4| +|553|Tensor<[1,300,320]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|554|Tensor<[300,320]>,
dims: [0, 1]
||aten::add.Tensor|4| +|555|Tensor<[1200,1280]>,
dims: [0, 1]
||aten::add.Tensor|4| +|556|Tensor<[1,300,512]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|557|Tensor<[300,512]>,
dims: [0, 1]
||aten::add.Tensor|4| +|558|Tensor<[300,2048]>,
dims: [0, 1]
||aten::add.Tensor|4| +|559|Tensor<[30]>,
dims: [0]
||aten::add.Tensor|4| +|560|Tensor<[30,1]>,
dims: [0, 1]
||aten::add.Tensor|4| +|561|Tensor<[1,64,30,40]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|562|Tensor<[1,32,30,40]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|563|Tensor<[60]>,
dims: [0]
||aten::add.Tensor|4| +|564|Tensor<[60,1]>,
dims: [0, 1]
||aten::add.Tensor|4| +|565|Tensor<[80]>,
dims: [0]
||aten::add.Tensor|4| +|566|Tensor<[1,64,60,80]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|567|Tensor<[1,32,60,80]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|568|Tensor<[120]>,
dims: [0]
||aten::add.Tensor|4| +|569|Tensor<[120,1]>,
dims: [0, 1]
||aten::add.Tensor|4| +|570|Tensor<[1,64,120,160]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|571|Tensor<[1,32,120,160]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|572|Tensor<[240]>,
dims: [0]
||aten::add.Tensor|4| +|573|Tensor<[240,1]>,
dims: [0, 1]
||aten::add.Tensor|4| +|574|Tensor<[480]>,
dims: [0]
||aten::add.Tensor|4| +|575|Tensor<[480,1]>,
dims: [0, 1]
||aten::add.Tensor|4| +|576|Tensor<[1,64,300]>,
dims: [0, 1, 2]
||aten::bmm|4| +|577|Tensor<[1,256,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|578|Tensor<[2,64,300]>,
dims: [0, 1, 2]
||aten::bmm|4| +|579|Tensor<[2,300,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|580|Tensor<[1,512,128]>,
dims: [0, 1, 2]
||aten::bmm|4| +|581|Tensor<[5,64,300]>,
dims: [0, 1, 2]
||aten::bmm|4| +|582|Tensor<[5,300,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|583|Tensor<[1,1280,320]>,
dims: [0, 1, 2]
||aten::bmm|4| +|584|Tensor<[8,64,300]>,
dims: [0, 1, 2]
||aten::bmm|4| +|585|Tensor<[8,300,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|586|Tensor<[1,2048,512]>,
dims: [0, 1, 2]
||aten::bmm|4| +|587|Tensor<[1,64,15,20]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|588|Tensor<[1,256,120,160]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|589|Tensor<[1,128,60,80]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|590|Tensor<[1,128,15,20]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|591|Tensor<[1,512,60,80]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|592|Tensor<[1,320,30,40]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|593|Tensor<[1,320,15,20]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|594|Tensor<[1,1280,30,40]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|595|Tensor<[1,512,15,20]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|596|Tensor<[1,2048,15,20]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|597|Tensor<[1,2,30,40]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|598|Tensor<[2,1,1]>,
dims: [1, 2, 3]
||aten::convolution|4| +|599|Tensor<[1,2,60,80]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|600|Tensor<[1,2,120,160]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|601|Tensor<[1,64,480,640]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|602|Tensor<[1,1,480,640]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|603|Tensor<[256,64]>,
dims: [1, 2]
||aten::expand|5| +|604|Tensor<[512,128]>,
dims: [1, 2]
||aten::expand|5| +|605|Tensor<[1280,320]>,
dims: [1, 2]
||aten::expand|5| +|606|Tensor<[2048,512]>,
dims: [1, 2]
||aten::expand|5| +|607|Tensor<[30,1]>,
dims: [2, 3]
||aten::index.Tensor|4| +|608|Tensor<[60,1]>,
dims: [2, 3]
||aten::index.Tensor|4| +|609|Tensor<[80]>,
dims: [3]
||aten::index.Tensor|4| +|610|Tensor<[120,1]>,
dims: [2, 3]
||aten::index.Tensor|4| +|611|Tensor<[160]>,
dims: [3]
||aten::index.Tensor|4| +|612|Tensor<[240,1]>,
dims: [2, 3]
||aten::index.Tensor|4| +|613|Tensor<[320]>,
dims: [3]
||aten::index.Tensor|4| +|614|Tensor<[480,1]>,
dims: [2, 3]
||aten::index.Tensor|4| +|615|Tensor<[640]>,
dims: [3]
||aten::index.Tensor|4| +|616|Tensor<[1,1,30,40]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|617|Tensor<[1,1,60,80]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|618|Tensor<[1,1,120,160]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|619|Tensor<[1,64,240,320]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|620|Tensor<[1,12,197,197]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|621|Tensor<[1,12,197,1]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|622|Tensor<[1,197,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|623|Tensor<[1,197,768]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|624|Tensor<[197,768]>,
dims: [0, 1]
||aten::add.Tensor|4| +|625|Tensor<[197,3072]>,
dims: [0, 1]
||aten::add.Tensor|4| +|626|Tensor<[12,64,197]>,
dims: [0, 1, 2]
||aten::bmm|4| +|627|Tensor<[12,197,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|628|Tensor<[1,768,14,14]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|629|Tensor<[1,12,197,64]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|630|Tensor<[1,12,64,197]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|631|Tensor<[1,1,16384,256]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|632|Tensor<[1,1,16384,1]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|633|Tensor<[1,2,4096,256]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|634|Tensor<[1,2,4096,1]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|635|Tensor<[1,5,1024,256]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|636|Tensor<[1,5,1024,1]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|637|Tensor<[1,16384,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|638|Tensor<[1,16384,32]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|639|Tensor<[32]>,
dims: [2]
||aten::add.Tensor|4| +|640|Tensor<[16384,32]>,
dims: [0, 1]
||aten::add.Tensor|4| +|641|Tensor<[1,256,32]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|642|Tensor<[256,32]>,
dims: [0, 1]
||aten::add.Tensor|4| +|643|Tensor<[16384,128]>,
dims: [0, 1]
||aten::add.Tensor|4| +|644|Tensor<[1,4096,64]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|645|Tensor<[4096,64]>,
dims: [0, 1]
||aten::add.Tensor|4| +|646|Tensor<[256,64]>,
dims: [0, 1]
||aten::add.Tensor|4| +|647|Tensor<[4096,256]>,
dims: [0, 1]
||aten::add.Tensor|4| +|648|Tensor<[1,1024,160]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|649|Tensor<[160]>,
dims: [2]
||aten::add.Tensor|4| +|650|Tensor<[1024,160]>,
dims: [0, 1]
||aten::add.Tensor|4| +|651|Tensor<[160]>,
dims: [1]
||aten::add.Tensor|4| +|652|Tensor<[1,256,160]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|653|Tensor<[256,160]>,
dims: [0, 1]
||aten::add.Tensor|4| +|654|Tensor<[1,256,256]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|655|Tensor<[256,1024]>,
dims: [0, 1]
||aten::add.Tensor|4| +|656|Tensor<[1,16384,256]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|657|Tensor<[128,1]>,
dims: [0, 1]
||aten::add.Tensor|4| +|658|Tensor<[1,4096,256]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|659|Tensor<[1,1024,256]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|660|Tensor<[1,256,128,128]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|661|Tensor<[1,32,256]>,
dims: [0, 1, 2]
||aten::bmm|4| +|662|Tensor<[1,128,32]>,
dims: [0, 1, 2]
||aten::bmm|4| +|663|Tensor<[2,32,256]>,
dims: [0, 1, 2]
||aten::bmm|4| +|664|Tensor<[2,256,32]>,
dims: [0, 1, 2]
||aten::bmm|4| +|665|Tensor<[5,32,256]>,
dims: [0, 1, 2]
||aten::bmm|4| +|666|Tensor<[5,256,32]>,
dims: [0, 1, 2]
||aten::bmm|4| +|667|Tensor<[1,640,160]>,
dims: [0, 1, 2]
||aten::bmm|4| +|668|Tensor<[8,256,32]>,
dims: [0, 1, 2]
||aten::bmm|4| +|669|Tensor<[1,64,256]>,
dims: [0, 1, 2]
||aten::bmm|4| +|670|Tensor<[1,160,256]>,
dims: [0, 1, 2]
||aten::bmm|4| +|671|Tensor<[1,32,128,128]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|672|Tensor<[1,32,16,16]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|673|Tensor<[1,64,64,64]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|674|Tensor<[1,64,16,16]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|675|Tensor<[1,160,32,32]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|676|Tensor<[1,160,16,16]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|677|Tensor<[1,150,128,128]>,
dims: [0, 1, 2, 3]
||aten::convolution|4| +|678|Tensor<[150,1,1]>,
dims: [1, 2, 3]
||aten::convolution|4| +|679|Tensor<[128,32]>,
dims: [1, 2]
||aten::expand|5| +|680|Tensor<[640,160]>,
dims: [1, 2]
||aten::expand|5| +|681|Tensor<[1024,256]>,
dims: [1, 2]
||aten::expand|5| +|682|Tensor<[32,256]>,
dims: [1, 2]
||aten::expand|5| +|683|Tensor<[64,256]>,
dims: [1, 2]
||aten::expand|5| +|684|Tensor<[160,256]>,
dims: [1, 2]
||aten::expand|5| +|685|Tensor<[128,1]>,
dims: [2, 3]
||aten::index.Tensor|4| +|686|Tensor<[1,71,7,7]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|687|Tensor<[1,71,7,1]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|688|Tensor<[1,7,4544]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|689|Tensor<[4544]>,
dims: [2]
||aten::add.Tensor|4| +|690|Tensor<[1,1,7]>,
dims: [0, 1, 2]
||aten::bmm|4| +|691|Tensor<[71,64,7]>,
dims: [0, 1, 2]
||aten::bmm|4| +|692|Tensor<[71,7,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|693|Tensor<[1,1,64,7]>,
dims: [0, 1, 2, 3]
||aten::expand|5| +|694|Tensor<[1,1,7,64]>,
dims: [0, 1, 2, 3]
||aten::expand|5| +|695|Tensor<[7,1,1]>,
dims: [1, 2, 3]
||aten::index.Tensor|4| +|696|Tensor<[1,1]>,
dims: [2, 3]
||aten::index.Tensor|4| +|697|Tensor<[1,71,7,64]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|698|Tensor<[1,7,64]>,
dims: [0, 1, 2]
||aten::mul.Tensor|4| +|699|Tensor<[1,16,112,112]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|700|Tensor<[96]>,
dims: [0]
||aten::add.Tensor|4| +|701|Tensor<[1,96,112,112]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|702|Tensor<[96,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|703|Tensor<[1,96,56,56]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|704|Tensor<[144]>,
dims: [0]
||aten::add.Tensor|4| +|705|Tensor<[1,144,56,56]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|706|Tensor<[144,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|707|Tensor<[1,144,28,28]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|708|Tensor<[1,32,28,28]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|709|Tensor<[1,192,28,28]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|710|Tensor<[1,192,14,14]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|711|Tensor<[1,64,14,14]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|712|Tensor<[384]>,
dims: [0]
||aten::add.Tensor|4| +|713|Tensor<[1,384,14,14]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|714|Tensor<[384,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|715|Tensor<[1,96,14,14]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|716|Tensor<[576]>,
dims: [0]
||aten::add.Tensor|4| +|717|Tensor<[1,576,14,14]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|718|Tensor<[576,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|719|Tensor<[1,576,7,7]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|720|Tensor<[960]>,
dims: [0]
||aten::add.Tensor|4| +|721|Tensor<[1,960,7,7]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|722|Tensor<[960,1,1]>,
dims: [1, 2, 3]
||aten::add.Tensor|4| +|723|Tensor<[1,320,7,7]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|724|Tensor<[1,1280,7,7]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|725|Tensor<[1,12,12,12]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|726|Tensor<[1,12,12,1]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|727|Tensor<[1,12,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|728|Tensor<[1,12,128]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|729|Tensor<[12,768]>,
dims: [0, 1]
||aten::add.Tensor|4| +|730|Tensor<[1,1,12,12]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|731|Tensor<[1,12,768]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|732|Tensor<[12,3072]>,
dims: [0, 1]
||aten::add.Tensor|4| +|733|Tensor<[1,12,3072]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|734|Tensor<[12,2]>,
dims: [0, 1]
||aten::add.Tensor|4| +|735|Tensor<[12,64,12]>,
dims: [0, 1, 2]
||aten::bmm|4| +|736|Tensor<[12,12,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|737|Tensor<[1,1,1,12]>,
dims: [0, 1, 2, 3]
||aten::expand|4| +|738|Tensor<[1,12,12,64]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|739|Tensor<[1,12,64,12]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|740|Tensor<[1,12,9,9]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|741|Tensor<[1,12,9,1]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|742|Tensor<[1,9,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|743|Tensor<[1,9,128]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|744|Tensor<[9,768]>,
dims: [0, 1]
||aten::add.Tensor|4| +|745|Tensor<[1,1,9,9]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|746|Tensor<[1,9,768]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|747|Tensor<[9,3072]>,
dims: [0, 1]
||aten::add.Tensor|4| +|748|Tensor<[1,9,3072]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|749|Tensor<[9,128]>,
dims: [0, 1]
||aten::add.Tensor|4| +|750|Tensor<[9,30000]>,
dims: [0, 1]
||aten::add.Tensor|4| +|751|Tensor<[30000]>,
dims: [1]
||aten::add.Tensor|4| +|752|Tensor<[12,64,9]>,
dims: [0, 1, 2]
||aten::bmm|4| +|753|Tensor<[12,9,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|754|Tensor<[1,1,1,9]>,
dims: [0, 1, 2, 3]
||aten::expand|4| +|755|Tensor<[1,12,9,64]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|756|Tensor<[1,12,64,9]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|757|Tensor<[30000]>,
dims: [0]
||aten::mul.Tensor|4| +|758|Tensor<[1,16,9,9]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|759|Tensor<[1,16,9,1]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|760|Tensor<[9,2048]>,
dims: [0, 1]
||aten::add.Tensor|4| +|761|Tensor<[1,9,2048]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|762|Tensor<[2048]>,
dims: [2]
||aten::add.Tensor|4| +|763|Tensor<[9,8192]>,
dims: [0, 1]
||aten::add.Tensor|4| +|764|Tensor<[8192]>,
dims: [1]
||aten::add.Tensor|4| +|765|Tensor<[1,9,8192]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|766|Tensor<[16,128,9]>,
dims: [0, 1, 2]
||aten::bmm|4| +|767|Tensor<[16,9,128]>,
dims: [0, 1, 2]
||aten::bmm|4| +|768|Tensor<[1,16,9,128]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|769|Tensor<[1,16,128,9]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|770|Tensor<[8192]>,
dims: [0]
||aten::mul.Tensor|4| +|771|Tensor<[9,1024]>,
dims: [0, 1]
||aten::add.Tensor|4| +|772|Tensor<[1,9,1024]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|773|Tensor<[9,4096]>,
dims: [0, 1]
||aten::add.Tensor|4| +|774|Tensor<[1,9,4096]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|775|Tensor<[16,64,9]>,
dims: [0, 1, 2]
||aten::bmm|4| +|776|Tensor<[16,9,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|777|Tensor<[1,16,9,64]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|778|Tensor<[1,16,64,9]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|779|Tensor<[1,64,9,9]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|780|Tensor<[1,64,9,1]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|781|Tensor<[9,16384]>,
dims: [0, 1]
||aten::add.Tensor|4| +|782|Tensor<[16384]>,
dims: [1]
||aten::add.Tensor|4| +|783|Tensor<[1,9,16384]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|784|Tensor<[64,64,9]>,
dims: [0, 1, 2]
||aten::bmm|4| +|785|Tensor<[64,9,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|786|Tensor<[1,64,9,64]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|787|Tensor<[1,64,64,9]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|788|Tensor<[16384]>,
dims: [0]
||aten::mul.Tensor|4| +|789|Tensor<[1,2]>,
dims: [0, 1]
||aten::add.Tensor|4| +|790|Tensor<[1,12,14,14]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|791|Tensor<[1,12,14,1]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|792|Tensor<[1,14,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|793|Tensor<[1,14,128]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|794|Tensor<[14,768]>,
dims: [0, 1]
||aten::add.Tensor|4| +|795|Tensor<[1,1,14,14]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|796|Tensor<[1,14,768]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|797|Tensor<[14,3072]>,
dims: [0, 1]
||aten::add.Tensor|4| +|798|Tensor<[1,14,3072]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|799|Tensor<[14,2]>,
dims: [0, 1]
||aten::add.Tensor|4| +|800|Tensor<[12,64,14]>,
dims: [0, 1, 2]
||aten::bmm|4| +|801|Tensor<[12,14,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|802|Tensor<[1,1,1,14]>,
dims: [0, 1, 2, 3]
||aten::expand|4| +|803|Tensor<[1,12,14,64]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|804|Tensor<[1,12,64,14]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|805|Tensor<[1,12,50,50]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|806|Tensor<[1,12,50,1]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|807|Tensor<[2,8,7,7]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|808|Tensor<[2,8,7,1]>,
dims: [0, 1, 2, 3]
||aten::_safe_softmax|4| +|809|Tensor<[1,50,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|810|Tensor<[1,50,768]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|811|Tensor<[50,768]>,
dims: [0, 1]
||aten::add.Tensor|4| +|812|Tensor<[50,3072]>,
dims: [0, 1]
||aten::add.Tensor|4| +|813|Tensor<[2,7,512]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|814|Tensor<[1,7,512]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|815|Tensor<[2,7,1]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|816|Tensor<[14,512]>,
dims: [0, 1]
||aten::add.Tensor|4| +|817|Tensor<[2,1,7,7]>,
dims: [0, 1, 2, 3]
||aten::add.Tensor|4| +|818|Tensor<[14,2048]>,
dims: [0, 1]
||aten::add.Tensor|4| +|819|Tensor<[12,64,50]>,
dims: [0, 1, 2]
||aten::bmm|4| +|820|Tensor<[12,50,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|821|Tensor<[16,64,7]>,
dims: [0, 1, 2]
||aten::bmm|4| +|822|Tensor<[16,7,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|823|Tensor<[2,512]>,
dims: [0, 1]
||aten::div.Tensor|4| +|824|Tensor<[2,1]>,
dims: [0, 1]
||aten::div.Tensor|4| +|825|Tensor<[2,1,1,7]>,
dims: [0, 1, 2, 3]
||aten::expand|4| +|826|Tensor<[1,12,50,64]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|827|Tensor<[1,12,64,50]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|828|Tensor<[2,8,7,64]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|829|Tensor<[2,8,64,7]>,
dims: [0, 1, 2, 3]
||aten::mul.Scalar|4| +|830|Tensor<[1,50,3072]>,
dims: [0, 1, 2]
||aten::mul.Tensor|4| +|831|Tensor<[2,7,2048]>,
dims: [0, 1, 2]
||aten::mul.Tensor|4| +|832|Tensor<[1,16,197,197]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|833|Tensor<[1,16,197,1]>,
dims: [0, 1, 2, 3]
||aten::_softmax|4| +|834|Tensor<[1,197,1024]>,
dims: [0, 1, 2]
||aten::add.Tensor|4| +|835|Tensor<[197,1024]>,
dims: [0, 1]
||aten::add.Tensor|4| +|836|Tensor<[27]>,
dims: [0]
||aten::add.Tensor|4| +|837|Tensor<[27,1]>,
dims: [0, 1]
||aten::add.Tensor|4| +|838|Tensor<[196,196]>,
dims: [0, 1]
||aten::add.Tensor|4| +|839|Tensor<[197,4096]>,
dims: [0, 1]
||aten::add.Tensor|4| +|840|Tensor<[1,1024]>,
dims: [0, 1]
||aten::add.Tensor|4| +|841|Tensor<[197]>,
dims: [0]
||aten::arange|4| +|842|Tensor<[16,64,197]>,
dims: [0, 1, 2]
||aten::bmm|4| +|843|Tensor<[16,197,64]>,
dims: [0, 1, 2]
||aten::bmm|4| +|844|Tensor<[14,1]>,
dims: [0, 1]
||aten::expand|4| +|845|Tensor<[1,14]>,
dims: [0, 1]
||aten::expand|4| +|846|Tensor<[27,1]>,
dims: [2, 3]
||aten::index.Tensor|4| +|847|Tensor<[27]>,
dims: [3]
||aten::index.Tensor|4| +|848|Tensor<[1,16,27,27]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|849|Tensor<[2,196,1]>,
dims: [0, 1, 2]
||aten::sub.Tensor|4| +|850|Tensor<[2,1,196]>,
dims: [0, 1, 2]
||aten::sub.Tensor|4| +|851|Tensor<[1,197]>,
dims: [0, 1]
||aten::where.self|4| +|852|Tensor<[196,197]>,
dims: [0, 1]
||aten::where.self|4| +|853|Tensor<[197,1]>,
dims: [0, 1]
||aten::where.self|4| +|854|Tensor<[197,197]>,
dims: [0, 1]
||aten::where.self|4| +|855|Tensor<[12,1,1]>,
dims: [1, 2, 3]
||aten::index.Tensor|4| +|856|Tensor<[1,12,27,27]>,
dims: [0, 1, 2, 3]
||aten::mul.Tensor|4| +|857|Tensor<[1,64]>,
dims: [0, 1]
||aten::add.Tensor|4| +|858|Tensor<[1,12]>,
dims: [0, 1]
||aten::add.Tensor|4| +|859|Tensor<[12]>,
dims: [1]
||aten::add.Tensor|4| +|860|Tensor<[1,784]>,
dims: [0, 1]
||aten::add.Tensor|4| +|861|Tensor<[784]>,
dims: [1]
||aten::add.Tensor|4| +|862|Tensor<[784]>,
dims: [0]
||aten::mul.Tensor|4| diff --git a/docs/ops/stablehlo/stablehlo.ceil.md b/docs/ops/stablehlo/stablehlo.ceil.md new file mode 100644 index 00000000..785db80d --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.ceil.md @@ -0,0 +1,7 @@ + +### stablehlo.ceil::ttnn.ceil + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Scalar,
|ttnn.ceil|aten::arange|4| diff --git a/docs/ops/stablehlo/stablehlo.clamp.md b/docs/ops/stablehlo/stablehlo.clamp.md new file mode 100644 index 00000000..ae24c6be --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.clamp.md @@ -0,0 +1,53 @@ + +### stablehlo.clamp::ttnn.clamp + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,1024,512]>,
Tensor<[1,1024,512]>,
|ttnn.clamp|aten::gelu|4| +|1|Tensor<[1,256,256]>,
Tensor<[1,256,256]>,
|ttnn.clamp|aten::gelu|4| +|2|Tensor<[1,10,3072]>,
Tensor<[1,10,3072]>,
|ttnn.clamp|aten::gelu|4| +|3|Tensor<[1,10,768]>,
Tensor<[1,10,768]>,
|ttnn.clamp|aten::gelu|4| +|4|Tensor<[1,4096,1280]>,
Tensor<[1,4096,1280]>,
|ttnn.clamp|aten::gelu|4| +|5|Tensor<[1,1024,2560]>,
Tensor<[1,1024,2560]>,
|ttnn.clamp|aten::gelu|4| +|6|Tensor<[1,256,5120]>,
Tensor<[1,256,5120]>,
|ttnn.clamp|aten::gelu|4| +|7|Tensor<[1,64,5120]>,
Tensor<[1,64,5120]>,
|ttnn.clamp|aten::gelu|4| +|8|Tensor<[1,25,3072]>,
Tensor<[1,25,3072]>,
|ttnn.clamp|aten::gelu|4| +|9|Tensor<[1,1445,768]>,
Tensor<[1,1445,768]>,
|ttnn.clamp|aten::gelu|4| +|10|Tensor<[1,3072,8]>,
Tensor<[1,3072,8]>,
|ttnn.clamp|aten::gelu|4| +|11|Tensor<[1,256,1280]>,
Tensor<[1,256,1280]>,
|ttnn.clamp|aten::gelu|4| +|12|Tensor<[1,2048,768]>,
Tensor<[1,2048,768]>,
|ttnn.clamp|aten::gelu|4| +|13|Tensor<[1,201,3072]>,
Tensor<[1,201,3072]>,
|ttnn.clamp|aten::gelu|4| +|14|Tensor<[1,1536]>,
Tensor<[1,1536]>,
|ttnn.clamp|aten::gelu|4| +|15|Tensor<[1,19,4096]>,
Tensor<[1,19,4096]>,
|ttnn.clamp|aten::gelu|4| +|16|Tensor<[1,16,3072]>,
Tensor<[1,16,3072]>,
|ttnn.clamp|aten::gelu|4| +|17|Scalar,
Tensor<[30]>,
Scalar,
|ttnn.clamp|aten::clamp|4| +|18|Scalar,
Tensor<[30,1]>,
Scalar,
|ttnn.clamp|aten::clamp|4| +|19|Scalar,
Tensor<[40]>,
Scalar,
|ttnn.clamp|aten::clamp|4| +|20|Scalar,
Tensor<[60]>,
Scalar,
|ttnn.clamp|aten::clamp|4| +|21|Scalar,
Tensor<[60,1]>,
Scalar,
|ttnn.clamp|aten::clamp|4| +|22|Scalar,
Tensor<[80]>,
Scalar,
|ttnn.clamp|aten::clamp|4| +|23|Scalar,
Tensor<[120]>,
Scalar,
|ttnn.clamp|aten::clamp|4| +|24|Scalar,
Tensor<[120,1]>,
Scalar,
|ttnn.clamp|aten::clamp|4| +|25|Scalar,
Tensor<[160]>,
Scalar,
|ttnn.clamp|aten::clamp|4| +|26|Scalar,
Tensor<[240]>,
Scalar,
|ttnn.clamp|aten::clamp|4| +|27|Scalar,
Tensor<[240,1]>,
Scalar,
|ttnn.clamp|aten::clamp|4| +|28|Scalar,
Tensor<[320]>,
Scalar,
|ttnn.clamp|aten::clamp|4| +|29|Scalar,
Tensor<[480]>,
Scalar,
|ttnn.clamp|aten::clamp|4| +|30|Scalar,
Tensor<[480,1]>,
Scalar,
|ttnn.clamp|aten::clamp|4| +|31|Scalar,
Tensor<[640]>,
Scalar,
|ttnn.clamp|aten::clamp|4| +|32|Tensor<[1,19200,256]>,
Tensor<[1,19200,256]>,
|ttnn.clamp|aten::gelu|4| +|33|Tensor<[1,4800,512]>,
Tensor<[1,4800,512]>,
|ttnn.clamp|aten::gelu|4| +|34|Tensor<[1,1200,1280]>,
Tensor<[1,1200,1280]>,
|ttnn.clamp|aten::gelu|4| +|35|Tensor<[1,300,2048]>,
Tensor<[1,300,2048]>,
|ttnn.clamp|aten::gelu|4| +|36|Tensor<[1,197,3072]>,
Tensor<[1,197,3072]>,
|ttnn.clamp|aten::gelu|4| +|37|Scalar,
Tensor<[128]>,
Scalar,
|ttnn.clamp|aten::clamp|4| +|38|Scalar,
Tensor<[128,1]>,
Scalar,
|ttnn.clamp|aten::clamp|4| +|39|Tensor<[1,16384,128]>,
Tensor<[1,16384,128]>,
|ttnn.clamp|aten::gelu|4| +|40|Tensor<[1,4096,256]>,
Tensor<[1,4096,256]>,
|ttnn.clamp|aten::gelu|4| +|41|Tensor<[1,1024,640]>,
Tensor<[1,1024,640]>,
|ttnn.clamp|aten::gelu|4| +|42|Tensor<[1,256,1024]>,
Tensor<[1,256,1024]>,
|ttnn.clamp|aten::gelu|4| +|43|Tensor<[1,7,18176]>,
Tensor<[1,7,18176]>,
|ttnn.clamp|aten::gelu|4| +|44|Scalar,
Tensor<[27]>,
Scalar,
|ttnn.clamp|aten::clamp|4| +|45|Scalar,
Tensor<[27,1]>,
Scalar,
|ttnn.clamp|aten::clamp|4| +|46|Tensor<[1,197,4096]>,
Tensor<[1,197,4096]>,
|ttnn.clamp|aten::gelu|4| diff --git a/docs/ops/stablehlo/stablehlo.compare.md b/docs/ops/stablehlo/stablehlo.compare.md new file mode 100644 index 00000000..dca51bfd --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.compare.md @@ -0,0 +1,38 @@ + +### stablehlo.compare::ttnn.? + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,32,32,32]>,
Tensor<[1,32,32,32]>,
|ttnn.eq|aten::_safe_softmax|4| +|1|Tensor<[1,1,32,32]>,
Tensor<[1,1,32,32]>,
|ttnn.eq|aten::eq.Scalar|4| +|2|Tensor<[32,32]>,
Tensor<[32,32]>,
|ttnn.gt|aten::gt.Tensor|4| +|3|Tensor<[1,12,7,7]>,
Tensor<[1,12,7,7]>,
|ttnn.eq|aten::_safe_softmax|4| +|4|Tensor<[1,7]>,
Tensor<[1,7]>,
|ttnn.eq|aten::eq.Scalar|4| +|5|Tensor<[7,7]>,
Tensor<[7,7]>,
|ttnn.lt|aten::lt.Tensor|4| +|6|Tensor<[1,12,10,10]>,
Tensor<[1,12,10,10]>,
|ttnn.eq|aten::_safe_softmax|4| +|7|Tensor<[1,10]>,
Tensor<[1,10]>,
|ttnn.ne|aten::ne.Scalar|4| +|8|Tensor<[1,8,4096,4096]>,
Tensor<[1,8,4096,4096]>,
|ttnn.eq|aten::_safe_softmax|4| +|9|Tensor<[1,8,4096,9]>,
Tensor<[1,8,4096,9]>,
|ttnn.eq|aten::_safe_softmax|4| +|10|Tensor<[1,8,1024,1024]>,
Tensor<[1,8,1024,1024]>,
|ttnn.eq|aten::_safe_softmax|4| +|11|Tensor<[1,8,1024,9]>,
Tensor<[1,8,1024,9]>,
|ttnn.eq|aten::_safe_softmax|4| +|12|Tensor<[1,8,256,256]>,
Tensor<[1,8,256,256]>,
|ttnn.eq|aten::_safe_softmax|4| +|13|Tensor<[1,8,256,9]>,
Tensor<[1,8,256,9]>,
|ttnn.eq|aten::_safe_softmax|4| +|14|Tensor<[1,8,64,64]>,
Tensor<[1,8,64,64]>,
|ttnn.eq|aten::_safe_softmax|4| +|15|Tensor<[1,8,64,9]>,
Tensor<[1,8,64,9]>,
|ttnn.eq|aten::_safe_softmax|4| +|16|Tensor<[1,12,25,25]>,
Tensor<[1,12,25,25]>,
|ttnn.eq|aten::_safe_softmax|4| +|17|Tensor<[1,3,1445,1445]>,
Tensor<[1,3,1445,1445]>,
|ttnn.eq|aten::_safe_softmax|4| +|18|Tensor<[19]>,
Tensor<[19]>,
|ttnn.lt|aten::lt.Scalar|4| +|19|Tensor<[19,19]>,
Tensor<[19,19]>,
|ttnn.lt|aten::lt.Tensor|4| +|20|Tensor<[1,12,16,16]>,
Tensor<[1,12,16,16]>,
|ttnn.eq|aten::_safe_softmax|4| +|21|Tensor<[1,12,197,197]>,
Tensor<[1,12,197,197]>,
|ttnn.eq|aten::_safe_softmax|4| +|22|Tensor<[1,71,7,7]>,
Tensor<[1,71,7,7]>,
|ttnn.eq|aten::_safe_softmax|4| +|23|Tensor<[1,1,7,7]>,
Tensor<[1,1,7,7]>,
|ttnn.eq|aten::eq.Scalar|4| +|24|Tensor<[1,12,12,12]>,
Tensor<[1,12,12,12]>,
|ttnn.eq|aten::_safe_softmax|4| +|25|Tensor<[1,12,9,9]>,
Tensor<[1,12,9,9]>,
|ttnn.eq|aten::_safe_softmax|4| +|26|Tensor<[1,16,9,9]>,
Tensor<[1,16,9,9]>,
|ttnn.eq|aten::_safe_softmax|4| +|27|Tensor<[1,64,9,9]>,
Tensor<[1,64,9,9]>,
|ttnn.eq|aten::_safe_softmax|4| +|28|Tensor<[1,12,14,14]>,
Tensor<[1,12,14,14]>,
|ttnn.eq|aten::_safe_softmax|4| +|29|Tensor<[1,12,50,50]>,
Tensor<[1,12,50,50]>,
|ttnn.eq|aten::_safe_softmax|4| +|30|Tensor<[2,8,7,7]>,
Tensor<[2,8,7,7]>,
|ttnn.eq|aten::_safe_softmax|4| +|31|Tensor<[197]>,
Tensor<[197]>,
|ttnn.ge|aten::ge.Scalar|4| diff --git a/docs/ops/stablehlo/stablehlo.concatenate.md b/docs/ops/stablehlo/stablehlo.concatenate.md new file mode 100644 index 00000000..847011f3 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.concatenate.md @@ -0,0 +1,99 @@ + +### stablehlo.concatenate::ttnn.concat + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,32,64]>,
Tensor<[1,32,64]>,
dim: 2
|ttnn.concat|aten::cat|5| +|1|Tensor<[1,32,32,64]>,
Tensor<[1,32,32,64]>,
dim: 3
|ttnn.concat|aten::cat|5| +|2|Tensor<[1,1]>,
Tensor<[1,1]>,
dim: 1
|ttnn.concat|aten::index.Tensor|4| +|3|Tensor<[1,128,28,28]>,
Tensor<[1,19,28,28]>,
Tensor<[1,38,28,28]>,
dim: 1
|ttnn.concat|aten::cat|5| +|4|Tensor<[1,23,40,128]>,
Tensor<[1,23,40,128]>,
dim: 3
|ttnn.concat|aten::cat|5| +|5|Tensor<[1,1,23,40,1]>,
Tensor<[1,1,23,40,1]>,
Tensor<[1,1,23,40,1]>,
Tensor<[1,1,23,40,1]>,
dim: 4
|ttnn.concat|aten::index.Tensor|4| +|6|Tensor<[1,23,40,64,1]>,
Tensor<[1,23,40,64,1]>,
dim: 4
|ttnn.concat|aten::stack|5| +|7|Tensor<[1,100,1,256]>,
Tensor<[1,100,1,256]>,
Tensor<[1,100,1,256]>,
Tensor<[1,100,1,256]>,
Tensor<[1,100,1,256]>,
Tensor<[1,100,1,256]>,
dim: 0
|ttnn.concat|aten::stack|5| +|8|Tensor<[1,160]>,
Tensor<[1,160]>,
dim: 1
|ttnn.concat|aten::cat|5| +|9|Tensor<[1,1280,8,8]>,
Tensor<[1,1280,8,8]>,
dim: 1
|ttnn.concat|aten::cat|5| +|10|Tensor<[1,1280,16,16]>,
Tensor<[1,1280,16,16]>,
dim: 1
|ttnn.concat|aten::cat|5| +|11|Tensor<[1,1280,16,16]>,
Tensor<[1,640,16,16]>,
dim: 1
|ttnn.concat|aten::cat|5| +|12|Tensor<[1,1280,32,32]>,
Tensor<[1,640,32,32]>,
dim: 1
|ttnn.concat|aten::cat|5| +|13|Tensor<[1,640,32,32]>,
Tensor<[1,640,32,32]>,
dim: 1
|ttnn.concat|aten::cat|5| +|14|Tensor<[1,640,32,32]>,
Tensor<[1,320,32,32]>,
dim: 1
|ttnn.concat|aten::cat|5| +|15|Tensor<[1,640,64,64]>,
Tensor<[1,320,64,64]>,
dim: 1
|ttnn.concat|aten::cat|5| +|16|Tensor<[1,320,64,64]>,
Tensor<[1,320,64,64]>,
dim: 1
|ttnn.concat|aten::cat|5| +|17|Tensor<[1,1280,16,16,1]>,
Tensor<[1,1280,16,16,1]>,
Tensor<[1,1280,16,16,1]>,
Tensor<[1,1280,16,16,1]>,
dim: 4
|ttnn.concat|aten::index.Tensor|4| +|18|Tensor<[1,1280,32,32,1]>,
Tensor<[1,1280,32,32,1]>,
Tensor<[1,1280,32,32,1]>,
Tensor<[1,1280,32,32,1]>,
dim: 4
|ttnn.concat|aten::index.Tensor|4| +|19|Tensor<[1,640,64,64,1]>,
Tensor<[1,640,64,64,1]>,
Tensor<[1,640,64,64,1]>,
Tensor<[1,640,64,64,1]>,
dim: 4
|ttnn.concat|aten::index.Tensor|4| +|20|Tensor<[1,1,192]>,
Tensor<[1,1344,192]>,
Tensor<[1,100,192]>,
dim: 1
|ttnn.concat|aten::cat|5| +|21|Tensor<[1,8,768]>,
Tensor<[1,193,768]>,
dim: 1
|ttnn.concat|aten::cat|5| +|22|Tensor<[1,8]>,
Tensor<[1,193]>,
dim: 1
|ttnn.concat|aten::cat|4| +|23|Tensor<[1,1,12,16,1]>,
Tensor<[1,1,12,16,1]>,
Tensor<[1,1,12,16,1]>,
Tensor<[1,1,12,16,1]>,
dim: 4
|ttnn.concat|aten::index.Tensor|4| +|24|Tensor<[12,16,1]>,
Tensor<[12,16,1]>,
dim: 2
|ttnn.concat|aten::stack|4| +|25|Tensor<[19,1,1]>,
Tensor<[19,1,1]>,
dim: 2
|ttnn.concat|aten::gather|4| +|26|Tensor<[1,14,56,56]>,
Tensor<[1,64,56,56]>,
dim: 1
|ttnn.concat|aten::cat|5| +|27|Tensor<[1,14,56,56]>,
Tensor<[1,24,56,56]>,
Tensor<[1,64,56,56]>,
dim: 1
|ttnn.concat|aten::cat|5| +|28|Tensor<[1,14,56,56]>,
Tensor<[1,40,56,56]>,
dim: 1
|ttnn.concat|aten::cat|5| +|29|Tensor<[1,14,56,56]>,
Tensor<[1,24,56,56]>,
Tensor<[1,40,56,56]>,
Tensor<[1,64,56,56]>,
dim: 1
|ttnn.concat|aten::cat|5| +|30|Tensor<[1,14,56,56]>,
Tensor<[1,14,56,56]>,
Tensor<[1,14,56,56]>,
Tensor<[1,14,56,56]>,
Tensor<[1,68,56,56]>,
dim: 1
|ttnn.concat|aten::cat|5| +|31|Tensor<[1,16,28,28]>,
Tensor<[1,128,28,28]>,
dim: 1
|ttnn.concat|aten::cat|5| +|32|Tensor<[1,16,28,28]>,
Tensor<[1,28,28,28]>,
Tensor<[1,128,28,28]>,
dim: 1
|ttnn.concat|aten::cat|5| +|33|Tensor<[1,16,28,28]>,
Tensor<[1,46,28,28]>,
dim: 1
|ttnn.concat|aten::cat|5| +|34|Tensor<[1,16,28,28]>,
Tensor<[1,28,28,28]>,
Tensor<[1,46,28,28]>,
Tensor<[1,128,28,28]>,
dim: 1
|ttnn.concat|aten::cat|5| +|35|Tensor<[1,16,28,28]>,
Tensor<[1,78,28,28]>,
dim: 1
|ttnn.concat|aten::cat|5| +|36|Tensor<[1,16,28,28]>,
Tensor<[1,28,28,28]>,
Tensor<[1,78,28,28]>,
dim: 1
|ttnn.concat|aten::cat|5| +|37|Tensor<[1,16,28,28]>,
Tensor<[1,28,28,28]>,
Tensor<[1,46,28,28]>,
Tensor<[1,78,28,28]>,
Tensor<[1,128,28,28]>,
dim: 1
|ttnn.concat|aten::cat|5| +|38|Tensor<[1,16,28,28]>,
Tensor<[1,16,28,28]>,
Tensor<[1,16,28,28]>,
Tensor<[1,16,28,28]>,
Tensor<[1,16,28,28]>,
Tensor<[1,16,28,28]>,
Tensor<[1,16,28,28]>,
Tensor<[1,16,28,28]>,
Tensor<[1,134,28,28]>,
dim: 1
|ttnn.concat|aten::cat|5| +|39|Tensor<[1,20,28,28]>,
Tensor<[1,256,28,28]>,
dim: 1
|ttnn.concat|aten::cat|5| +|40|Tensor<[1,20,28,28]>,
Tensor<[1,34,28,28]>,
Tensor<[1,256,28,28]>,
dim: 1
|ttnn.concat|aten::cat|5| +|41|Tensor<[1,20,28,28]>,
Tensor<[1,58,28,28]>,
dim: 1
|ttnn.concat|aten::cat|5| +|42|Tensor<[1,20,28,28]>,
Tensor<[1,34,28,28]>,
Tensor<[1,58,28,28]>,
Tensor<[1,256,28,28]>,
dim: 1
|ttnn.concat|aten::cat|5| +|43|Tensor<[1,20,28,28]>,
Tensor<[1,98,28,28]>,
dim: 1
|ttnn.concat|aten::cat|5| +|44|Tensor<[1,20,28,28]>,
Tensor<[1,34,28,28]>,
Tensor<[1,98,28,28]>,
dim: 1
|ttnn.concat|aten::cat|5| +|45|Tensor<[1,20,28,28]>,
Tensor<[1,34,28,28]>,
Tensor<[1,58,28,28]>,
Tensor<[1,98,28,28]>,
Tensor<[1,256,28,28]>,
dim: 1
|ttnn.concat|aten::cat|5| +|46|Tensor<[1,20,28,28]>,
Tensor<[1,20,28,28]>,
Tensor<[1,20,28,28]>,
Tensor<[1,20,28,28]>,
Tensor<[1,20,28,28]>,
Tensor<[1,20,28,28]>,
Tensor<[1,20,28,28]>,
Tensor<[1,20,28,28]>,
Tensor<[1,168,28,28]>,
dim: 1
|ttnn.concat|aten::cat|5| +|47|Tensor<[1,40,14,14]>,
Tensor<[1,320,14,14]>,
dim: 1
|ttnn.concat|aten::cat|5| +|48|Tensor<[1,40,14,14]>,
Tensor<[1,68,14,14]>,
Tensor<[1,320,14,14]>,
dim: 1
|ttnn.concat|aten::cat|5| +|49|Tensor<[1,40,14,14]>,
Tensor<[1,116,14,14]>,
dim: 1
|ttnn.concat|aten::cat|5| +|50|Tensor<[1,40,14,14]>,
Tensor<[1,68,14,14]>,
Tensor<[1,116,14,14]>,
Tensor<[1,320,14,14]>,
dim: 1
|ttnn.concat|aten::cat|5| +|51|Tensor<[1,40,14,14]>,
Tensor<[1,196,14,14]>,
dim: 1
|ttnn.concat|aten::cat|5| +|52|Tensor<[1,40,14,14]>,
Tensor<[1,68,14,14]>,
Tensor<[1,196,14,14]>,
dim: 1
|ttnn.concat|aten::cat|5| +|53|Tensor<[1,40,14,14]>,
Tensor<[1,68,14,14]>,
Tensor<[1,116,14,14]>,
Tensor<[1,196,14,14]>,
Tensor<[1,320,14,14]>,
dim: 1
|ttnn.concat|aten::cat|5| +|54|Tensor<[1,40,14,14]>,
Tensor<[1,40,14,14]>,
Tensor<[1,40,14,14]>,
Tensor<[1,40,14,14]>,
Tensor<[1,40,14,14]>,
Tensor<[1,40,14,14]>,
Tensor<[1,40,14,14]>,
Tensor<[1,40,14,14]>,
Tensor<[1,334,14,14]>,
dim: 1
|ttnn.concat|aten::cat|5| +|55|Tensor<[1,160,7,7]>,
Tensor<[1,640,7,7]>,
dim: 1
|ttnn.concat|aten::cat|5| +|56|Tensor<[1,160,7,7]>,
Tensor<[1,272,7,7]>,
Tensor<[1,640,7,7]>,
dim: 1
|ttnn.concat|aten::cat|5| +|57|Tensor<[1,160,7,7]>,
Tensor<[1,160,7,7]>,
Tensor<[1,462,7,7]>,
dim: 1
|ttnn.concat|aten::cat|5| +|58|Tensor<[1,256,32,32]>,
Tensor<[1,512,32,32]>,
dim: 1
|ttnn.concat|aten::cat|5| +|59|Tensor<[1,128,64,64]>,
Tensor<[1,256,64,64]>,
dim: 1
|ttnn.concat|aten::cat|5| +|60|Tensor<[1,256,32,32,1]>,
Tensor<[1,256,32,32,1]>,
Tensor<[1,256,32,32,1]>,
Tensor<[1,256,32,32,1]>,
dim: 4
|ttnn.concat|aten::index.Tensor|4| +|61|Tensor<[1,128,64,64,1]>,
Tensor<[1,128,64,64,1]>,
Tensor<[1,128,64,64,1]>,
Tensor<[1,128,64,64,1]>,
dim: 4
|ttnn.concat|aten::index.Tensor|4| +|62|Tensor<[1,256,32,32]>,
Tensor<[1,256,32,32]>,
dim: 1
|ttnn.concat|aten::cat|5| +|63|Tensor<[1,128,64,64]>,
Tensor<[1,128,64,64]>,
dim: 1
|ttnn.concat|aten::cat|5| +|64|Tensor<[1,64,128,128]>,
Tensor<[1,64,128,128]>,
dim: 1
|ttnn.concat|aten::cat|5| +|65|Tensor<[1,32,256,256]>,
Tensor<[1,32,256,256]>,
dim: 1
|ttnn.concat|aten::cat|5| +|66|Tensor<[1,512,28,28]>,
Tensor<[1,512,28,28]>,
dim: 1
|ttnn.concat|aten::cat|5| +|67|Tensor<[1,256,56,56]>,
Tensor<[1,256,56,56]>,
dim: 1
|ttnn.concat|aten::cat|5| +|68|Tensor<[1,128,112,112]>,
Tensor<[1,128,112,112]>,
dim: 1
|ttnn.concat|aten::cat|5| +|69|Tensor<[1,64,224,224]>,
Tensor<[1,64,224,224]>,
dim: 1
|ttnn.concat|aten::cat|5| +|70|Tensor<[1,64,30,40]>,
Tensor<[1,64,30,40]>,
dim: 1
|ttnn.concat|aten::cat|5| +|71|Tensor<[1,64,60,80]>,
Tensor<[1,64,60,80]>,
dim: 1
|ttnn.concat|aten::cat|5| +|72|Tensor<[1,64,120,160]>,
Tensor<[1,64,120,160]>,
dim: 1
|ttnn.concat|aten::cat|5| +|73|Tensor<[1,64,30,40,1]>,
Tensor<[1,64,30,40,1]>,
Tensor<[1,64,30,40,1]>,
Tensor<[1,64,30,40,1]>,
dim: 4
|ttnn.concat|aten::index.Tensor|4| +|74|Tensor<[1,64,60,80,1]>,
Tensor<[1,64,60,80,1]>,
Tensor<[1,64,60,80,1]>,
Tensor<[1,64,60,80,1]>,
dim: 4
|ttnn.concat|aten::index.Tensor|4| +|75|Tensor<[1,64,120,160,1]>,
Tensor<[1,64,120,160,1]>,
Tensor<[1,64,120,160,1]>,
Tensor<[1,64,120,160,1]>,
dim: 4
|ttnn.concat|aten::index.Tensor|4| +|76|Tensor<[1,64,240,320,1]>,
Tensor<[1,64,240,320,1]>,
Tensor<[1,64,240,320,1]>,
Tensor<[1,64,240,320,1]>,
dim: 4
|ttnn.concat|aten::index.Tensor|4| +|77|Tensor<[1,64,480,640,1]>,
Tensor<[1,64,480,640,1]>,
Tensor<[1,64,480,640,1]>,
Tensor<[1,64,480,640,1]>,
dim: 4
|ttnn.concat|aten::index.Tensor|4| +|78|Tensor<[1,1,768]>,
Tensor<[1,196,768]>,
dim: 1
|ttnn.concat|aten::cat|5| +|79|Tensor<[1,256,128,128]>,
Tensor<[1,256,128,128]>,
Tensor<[1,256,128,128]>,
Tensor<[1,256,128,128]>,
dim: 1
|ttnn.concat|aten::cat|5| +|80|Tensor<[1,256,128,128,1]>,
Tensor<[1,256,128,128,1]>,
Tensor<[1,256,128,128,1]>,
Tensor<[1,256,128,128,1]>,
dim: 4
|ttnn.concat|aten::index.Tensor|4| +|81|Tensor<[1,7,32]>,
Tensor<[1,7,32]>,
dim: 2
|ttnn.concat|aten::cat|5| +|82|Tensor<[1,71,7,32]>,
Tensor<[1,71,7,32]>,
dim: 3
|ttnn.concat|aten::cat|5| +|83|Tensor<[1,1,7,32]>,
Tensor<[1,1,7,32]>,
dim: 3
|ttnn.concat|aten::cat|5| +|84|Tensor<[1,7,1,64,1]>,
Tensor<[1,7,1,64,1]>,
Tensor<[1,7,1,64,1]>,
Tensor<[1,7,1,64,1]>,
dim: 4
|ttnn.concat|aten::index.Tensor|4| +|85|Tensor<[1,1,768]>,
Tensor<[1,49,768]>,
dim: 1
|ttnn.concat|aten::cat|5| +|86|Tensor<[2,1]>,
Tensor<[2,1]>,
dim: 1
|ttnn.concat|aten::index.Tensor|4| +|87|Tensor<[1,1,1024]>,
Tensor<[1,196,1024]>,
dim: 1
|ttnn.concat|aten::cat|5| +|88|Tensor<[729,16]>,
Tensor<[3,16]>,
dim: 0
|ttnn.concat|aten::cat|5| +|89|Tensor<[1,16,27,27,1]>,
Tensor<[1,16,27,27,1]>,
Tensor<[1,16,27,27,1]>,
Tensor<[1,16,27,27,1]>,
dim: 4
|ttnn.concat|aten::index.Tensor|4| +|90|Tensor<[1,14,14]>,
Tensor<[1,14,14]>,
dim: 0
|ttnn.concat|aten::stack|4| +|91|Tensor<[729,12]>,
Tensor<[3,12]>,
dim: 0
|ttnn.concat|aten::cat|5| +|92|Tensor<[1,12,27,27,1]>,
Tensor<[1,12,27,27,1]>,
Tensor<[1,12,27,27,1]>,
Tensor<[1,12,27,27,1]>,
dim: 4
|ttnn.concat|aten::index.Tensor|4| diff --git a/docs/ops/stablehlo/stablehlo.constant.md b/docs/ops/stablehlo/stablehlo.constant.md new file mode 100644 index 00000000..e91c7cb3 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.constant.md @@ -0,0 +1,157 @@ + +### stablehlo.constant + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Scalar,
||aten::_safe_softmax|4| +|1|Tensor<[32]>,
||aten::arange|4| +|2|Tensor<[32,1]>,
||aten::triu|4| +|3|Tensor<[7]>,
||aten::add.Tensor|4| +|4|Tensor<[1]>,
||aten::arange|4| +|5|Tensor<[1,7]>,
||aten::eq.Scalar|4| +|6|Tensor<[64]>,
||aten::reciprocal|5| +|7|Tensor<[128]>,
||aten::reciprocal|5| +|8|Tensor<[256]>,
||aten::reciprocal|5| +|9|Tensor<[512]>,
||aten::reciprocal|5| +|10|Tensor<[1,32,112,112]>,
||aten::relu|4| +|11|Tensor<[1,64,112,112]>,
||aten::relu|4| +|12|Tensor<[1,64,56,56]>,
||aten::relu|4| +|13|Tensor<[1,128,56,56]>,
||aten::relu|4| +|14|Tensor<[1,128,28,28]>,
||aten::relu|4| +|15|Tensor<[1,256,28,28]>,
||aten::relu|4| +|16|Tensor<[1,512,28,28]>,
||aten::relu|4| +|17|Tensor<[1,1024,512]>,
||aten::gelu|4| +|18|Tensor<[1,256,256]>,
||aten::gelu|4| +|19|Tensor<[1,720,1280]>,
||aten::ones|4| +|20|Tensor<[1,64,360,640]>,
||aten::relu|4| +|21|Tensor<[1,64,180,320]>,
||aten::relu|4| +|22|Tensor<[1,256,180,320]>,
||aten::relu|4| +|23|Tensor<[1,128,180,320]>,
||aten::relu|4| +|24|Tensor<[1,128,90,160]>,
||aten::relu|4| +|25|Tensor<[1,512,90,160]>,
||aten::relu|4| +|26|Tensor<[1,256,90,160]>,
||aten::relu|4| +|27|Tensor<[1,256,45,80]>,
||aten::relu|4| +|28|Tensor<[1,1024,45,80]>,
||aten::relu|4| +|29|Tensor<[1,512,45,80]>,
||aten::relu|4| +|30|Tensor<[1,512,23,40]>,
||aten::relu|4| +|31|Tensor<[1,2048,23,40]>,
||aten::relu|4| +|32|Tensor<[920,1,2048]>,
||aten::relu|4| +|33|Tensor<[100,1,2048]>,
||aten::relu|4| +|34|Tensor<[6,1,100,256]>,
||aten::relu|4| +|35|Tensor<[1,1]>,
||aten::select_scatter|4| +|36|Tensor<[1,3,720,1280]>,
||aten::zeros|5| +|37|Tensor<[1,10]>,
||aten::add.Tensor|5| +|38|Tensor<[1,10,3072]>,
||aten::gelu|4| +|39|Tensor<[1,10,768]>,
||aten::gelu|4| +|40|Tensor<[1,4096,1280]>,
||aten::gelu|4| +|41|Tensor<[1,1024,2560]>,
||aten::gelu|4| +|42|Tensor<[1,256,5120]>,
||aten::gelu|4| +|43|Tensor<[1,64,5120]>,
||aten::gelu|4| +|44|Tensor<[1280]>,
||aten::index.Tensor|4| +|45|Tensor<[640]>,
||aten::index.Tensor|4| +|46|Tensor<[1,25,3072]>,
||aten::gelu|4| +|47|Tensor<[1,1445,768]>,
||aten::gelu|4| +|48|Tensor<[1,100,192]>,
||aten::relu|4| +|49|Tensor<[1,256,14,14]>,
||aten::relu|4| +|50|Tensor<[1,512,7,7]>,
||aten::relu|4| +|51|Tensor<[1,3072,8]>,
||aten::gelu|4| +|52|Tensor<[2048]>,
||aten::arange.start|4| +|53|Tensor<[1,256,1280]>,
||aten::gelu|4| +|54|Tensor<[1,2048,768]>,
||aten::gelu|4| +|55|Tensor<[1024]>,
||aten::reciprocal|5| +|56|Tensor<[1,256,56,56]>,
||aten::relu|4| +|57|Tensor<[1,1024,14,14]>,
||aten::relu|4| +|58|Tensor<[1,512,14,14]>,
||aten::relu|4| +|59|Tensor<[1,2048,7,7]>,
||aten::relu|4| +|60|Tensor<[1,193]>,
||aten::full_like|4| +|61|Tensor<[1,201,3072]>,
||aten::gelu|4| +|62|Tensor<[1,1536]>,
||aten::gelu|4| +|63|Tensor<[1,192]>,
||aten::rsub.Scalar|4| +|64|Tensor<[1,8]>,
||aten::zeros_like|4| +|65|Tensor<[1,32,26,26]>,
||aten::relu|4| +|66|Tensor<[1,64,24,24]>,
||aten::relu|4| +|67|Tensor<[1,128]>,
||aten::relu|4| +|68|Tensor<[19]>,
||aten::add.Tensor|4| +|69|Tensor<[1,19]>,
||aten::add.Tensor|4| +|70|Tensor<[1,19,4096]>,
||aten::gelu|4| +|71|Tensor<[14]>,
||aten::reciprocal|5| +|72|Tensor<[24]>,
||aten::reciprocal|5| +|73|Tensor<[40]>,
||aten::reciprocal|5| +|74|Tensor<[68]>,
||aten::reciprocal|5| +|75|Tensor<[16]>,
||aten::reciprocal|5| +|76|Tensor<[28]>,
||aten::reciprocal|5| +|77|Tensor<[46]>,
||aten::reciprocal|5| +|78|Tensor<[78]>,
||aten::reciprocal|5| +|79|Tensor<[134]>,
||aten::reciprocal|5| +|80|Tensor<[20]>,
||aten::reciprocal|5| +|81|Tensor<[34]>,
||aten::reciprocal|5| +|82|Tensor<[58]>,
||aten::reciprocal|5| +|83|Tensor<[98]>,
||aten::reciprocal|5| +|84|Tensor<[168]>,
||aten::reciprocal|5| +|85|Tensor<[320]>,
||aten::reciprocal|5| +|86|Tensor<[116]>,
||aten::reciprocal|5| +|87|Tensor<[196]>,
||aten::reciprocal|5| +|88|Tensor<[334]>,
||aten::reciprocal|5| +|89|Tensor<[160]>,
||aten::reciprocal|5| +|90|Tensor<[272]>,
||aten::reciprocal|5| +|91|Tensor<[462]>,
||aten::reciprocal|5| +|92|Tensor<[1,32,256,256]>,
||aten::relu|4| +|93|Tensor<[1,64,128,128]>,
||aten::relu|4| +|94|Tensor<[1,128,64,64]>,
||aten::relu|4| +|95|Tensor<[1,256,32,32]>,
||aten::relu|4| +|96|Tensor<[1,512,16,16]>,
||aten::relu|4| +|97|Tensor<[1,16,28,28]>,
||aten::relu|4| +|98|Tensor<[1,4,14,14]>,
||aten::relu|4| +|99|Tensor<[1,16,14,14]>,
||aten::relu|4| +|100|Tensor<[1,32]>,
||aten::sub.Tensor|4| +|101|Tensor<[1,16,3072]>,
||aten::gelu|4| +|102|Tensor<[1,64,224,224]>,
||aten::relu|4| +|103|Tensor<[1,128,112,112]>,
||aten::relu|4| +|104|Tensor<[30,1]>,
||aten::add.Tensor|4| +|105|Tensor<[60,1]>,
||aten::add.Tensor|4| +|106|Tensor<[80]>,
||aten::add.Tensor|4| +|107|Tensor<[120,1]>,
||aten::add.Tensor|4| +|108|Tensor<[240,1]>,
||aten::add.Tensor|4| +|109|Tensor<[480,1]>,
||aten::add.Tensor|4| +|110|Tensor<[30]>,
||aten::arange|4| +|111|Tensor<[60]>,
||aten::arange|4| +|112|Tensor<[120]>,
||aten::arange|4| +|113|Tensor<[240]>,
||aten::arange|4| +|114|Tensor<[480]>,
||aten::arange|4| +|115|Tensor<[1,19200,256]>,
||aten::gelu|4| +|116|Tensor<[1,4800,512]>,
||aten::gelu|4| +|117|Tensor<[1,1200,1280]>,
||aten::gelu|4| +|118|Tensor<[1,300,2048]>,
||aten::gelu|4| +|119|Tensor<[1,64,30,40]>,
||aten::relu|4| +|120|Tensor<[1,32,30,40]>,
||aten::relu|4| +|121|Tensor<[1,64,60,80]>,
||aten::relu|4| +|122|Tensor<[1,32,60,80]>,
||aten::relu|4| +|123|Tensor<[1,64,120,160]>,
||aten::relu|4| +|124|Tensor<[1,32,120,160]>,
||aten::relu|4| +|125|Tensor<[1,64,480,640]>,
||aten::relu|4| +|126|Tensor<[1,197,3072]>,
||aten::gelu|4| +|127|Tensor<[128,1]>,
||aten::add.Tensor|4| +|128|Tensor<[1,16384,128]>,
||aten::gelu|4| +|129|Tensor<[1,4096,256]>,
||aten::gelu|4| +|130|Tensor<[1,1024,640]>,
||aten::gelu|4| +|131|Tensor<[1,256,1024]>,
||aten::gelu|4| +|132|Tensor<[1,256,128,128]>,
||aten::relu|4| +|133|Tensor<[1,7,18176]>,
||aten::gelu|4| +|134|Tensor<[7,1]>,
||aten::triu|4| +|135|Tensor<[96]>,
||aten::reciprocal|5| +|136|Tensor<[144]>,
||aten::reciprocal|5| +|137|Tensor<[192]>,
||aten::reciprocal|5| +|138|Tensor<[384]>,
||aten::reciprocal|5| +|139|Tensor<[576]>,
||aten::reciprocal|5| +|140|Tensor<[960]>,
||aten::reciprocal|5| +|141|Tensor<[2]>,
||aten::arange|4| +|142|Tensor<[27,1]>,
||aten::add.Tensor|4| +|143|Tensor<[27]>,
||aten::add.Tensor|4| +|144|Tensor<[196,196]>,
||aten::add.Tensor|4| +|145|Tensor<[197]>,
||aten::arange|4| +|146|Tensor<[1,197,4096]>,
||aten::gelu|4| +|147|Tensor<[197,197]>,
||aten::zeros|4| +|148|Tensor<[12]>,
||aten::index.Tensor|4| +|149|Tensor<[1,64]>,
||aten::relu|4| +|150|Tensor<[1,12]>,
||aten::relu|4| diff --git a/docs/ops/stablehlo/stablehlo.convert.md b/docs/ops/stablehlo/stablehlo.convert.md new file mode 100644 index 00000000..58722c0a --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.convert.md @@ -0,0 +1,572 @@ + +### stablehlo.convert + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1]>,
||aten::_safe_softmax|4| +|1|Scalar,
||aten::_safe_softmax|4| +|2|Tensor<[1,1,1,32]>,
||aten::add.Tensor|4| +|3|Tensor<[1,1,32,32]>,
||aten::add.Tensor|4| +|4|Tensor<[1,32,4096]>,
||aten::embedding|4| +|5|Tensor<[32,32]>,
||aten::mul.Tensor|5| +|6|Tensor<[1,1,32]>,
||prims::convert_element_type|4| +|7|Tensor<[1,32,128]>,
||prims::convert_element_type|5| +|8|Tensor<[1,32,32,128]>,
||prims::convert_element_type|5| +|9|Tensor<[768]>,
||aten::add.Tensor|4| +|10|Tensor<[1,1,7,7]>,
||aten::add.Tensor|4| +|11|Tensor<[1,7,768]>,
||aten::embedding|4| +|12|Tensor<[7,7]>,
||prims::convert_element_type|5| +|13|Tensor<[2304]>,
||prims::convert_element_type|5| +|14|Tensor<[7,768]>,
||prims::convert_element_type|5| +|15|Tensor<[768,2304]>,
||prims::convert_element_type|5| +|16|Tensor<[7,2304]>,
||prims::convert_element_type|5| +|17|Tensor<[1,12,7,64]>,
||prims::convert_element_type|5| +|18|Tensor<[768,768]>,
||prims::convert_element_type|5| +|19|Tensor<[3072]>,
||prims::convert_element_type|5| +|20|Tensor<[768,3072]>,
||prims::convert_element_type|5| +|21|Tensor<[7,3072]>,
||prims::convert_element_type|5| +|22|Tensor<[3072,768]>,
||prims::convert_element_type|5| +|23|Tensor<[1,7]>,
||prims::convert_element_type|5| +|24|Tensor<[32,1,1]>,
||aten::add.Tensor|4| +|25|Tensor<[64,1,1]>,
||aten::add.Tensor|4| +|26|Tensor<[128,1,1]>,
||aten::add.Tensor|4| +|27|Tensor<[256,1,1]>,
||aten::add.Tensor|4| +|28|Tensor<[512,1,1]>,
||aten::add.Tensor|4| +|29|Tensor<[1,32,112,112]>,
||aten::sub.Tensor|4| +|30|Tensor<[1,64,112,112]>,
||aten::sub.Tensor|4| +|31|Tensor<[1,64,56,56]>,
||aten::sub.Tensor|4| +|32|Tensor<[1,128,56,56]>,
||aten::sub.Tensor|4| +|33|Tensor<[1,128,28,28]>,
||aten::sub.Tensor|4| +|34|Tensor<[1,256,28,28]>,
||aten::sub.Tensor|4| +|35|Tensor<[1,512,28,28]>,
||aten::sub.Tensor|4| +|36|Tensor<[32]>,
||prims::convert_element_type|5| +|37|Tensor<[64]>,
||prims::convert_element_type|5| +|38|Tensor<[128]>,
||prims::convert_element_type|5| +|39|Tensor<[256]>,
||prims::convert_element_type|5| +|40|Tensor<[512]>,
||prims::convert_element_type|5| +|41|Tensor<[1,1024,512]>,
||aten::gelu|4| +|42|Tensor<[1,256,256]>,
||aten::gelu|4| +|43|Tensor<[1,256,512]>,
||aten::sub.Tensor|4| +|44|Tensor<[256,768]>,
||prims::convert_element_type|5| +|45|Tensor<[768,512]>,
||prims::convert_element_type|5| +|46|Tensor<[256,512]>,
||prims::convert_element_type|5| +|47|Tensor<[512,256]>,
||prims::convert_element_type|5| +|48|Tensor<[256,256]>,
||prims::convert_element_type|5| +|49|Tensor<[1000]>,
||prims::convert_element_type|5| +|50|Tensor<[1,512]>,
||prims::convert_element_type|5| +|51|Tensor<[512,1000]>,
||prims::convert_element_type|5| +|52|Tensor<[1,1000]>,
||prims::convert_element_type|5| +|53|Tensor<[8,920,920]>,
||aten::_softmax|4| +|54|Tensor<[8,100,100]>,
||aten::_softmax|4| +|55|Tensor<[8,100,920]>,
||aten::_softmax|4| +|56|Tensor<[1,23,40]>,
||aten::cumsum|4| +|57|Tensor<[920,1,256]>,
||aten::sub.Tensor|4| +|58|Tensor<[100,1,256]>,
||aten::sub.Tensor|4| +|59|Tensor<[1,3,720,1280]>,
||aten::zeros|5| +|60|Tensor<[1,1,720,1280]>,
||prims::convert_element_type|5| +|61|Tensor<[23]>,
||prims::convert_element_type|4| +|62|Tensor<[40]>,
||prims::convert_element_type|4| +|63|Tensor<[1,1,23,40]>,
||prims::convert_element_type|5| +|64|Tensor<[1,256,23,40]>,
||prims::convert_element_type|5| +|65|Tensor<[920,256]>,
||prims::convert_element_type|5| +|66|Tensor<[2048]>,
||prims::convert_element_type|5| +|67|Tensor<[256,2048]>,
||prims::convert_element_type|5| +|68|Tensor<[920,2048]>,
||prims::convert_element_type|5| +|69|Tensor<[2048,256]>,
||prims::convert_element_type|5| +|70|Tensor<[100,256]>,
||prims::convert_element_type|5| +|71|Tensor<[100,2048]>,
||prims::convert_element_type|5| +|72|Tensor<[1,1,10,10]>,
||aten::add.Tensor|4| +|73|Tensor<[1,10]>,
||aten::cumsum|4| +|74|Tensor<[1,10,768]>,
||aten::embedding|4| +|75|Tensor<[1,10,3072]>,
||aten::gelu|4| +|76|Tensor<[10,768]>,
||prims::convert_element_type|5| +|77|Tensor<[1,12,10,64]>,
||prims::convert_element_type|5| +|78|Tensor<[10,3072]>,
||prims::convert_element_type|5| +|79|Tensor<[250002]>,
||prims::convert_element_type|5| +|80|Tensor<[768,250002]>,
||prims::convert_element_type|5| +|81|Tensor<[10,250002]>,
||prims::convert_element_type|5| +|82|Tensor<[1,320,1,1]>,
||aten::add.Tensor|4| +|83|Tensor<[320]>,
||aten::add.Tensor|4| +|84|Tensor<[640]>,
||aten::add.Tensor|4| +|85|Tensor<[1,640,1,1]>,
||aten::add.Tensor|4| +|86|Tensor<[1280]>,
||aten::add.Tensor|4| +|87|Tensor<[1,1280,1,1]>,
||aten::add.Tensor|4| +|88|Tensor<[1,2560,1,1]>,
||aten::add.Tensor|4| +|89|Tensor<[1,1920,1,1]>,
||aten::add.Tensor|4| +|90|Tensor<[1,960,1,1]>,
||aten::add.Tensor|4| +|91|Tensor<[1,4096,1280]>,
||aten::gelu|4| +|92|Tensor<[1,1024,2560]>,
||aten::gelu|4| +|93|Tensor<[1,256,5120]>,
||aten::gelu|4| +|94|Tensor<[1,64,5120]>,
||aten::gelu|4| +|95|Tensor<[1,32,10,4096]>,
||aten::sub.Tensor|4| +|96|Tensor<[1,4096,320]>,
||aten::sub.Tensor|4| +|97|Tensor<[1,32,10,1024]>,
||aten::sub.Tensor|4| +|98|Tensor<[1,32,20,1024]>,
||aten::sub.Tensor|4| +|99|Tensor<[1,1024,640]>,
||aten::sub.Tensor|4| +|100|Tensor<[1,32,20,256]>,
||aten::sub.Tensor|4| +|101|Tensor<[1,32,40,256]>,
||aten::sub.Tensor|4| +|102|Tensor<[1,256,1280]>,
||aten::sub.Tensor|4| +|103|Tensor<[1,32,40,64]>,
||aten::sub.Tensor|4| +|104|Tensor<[1,64,1280]>,
||aten::sub.Tensor|4| +|105|Tensor<[1,32,80,64]>,
||aten::sub.Tensor|4| +|106|Tensor<[1,32,80,256]>,
||aten::sub.Tensor|4| +|107|Tensor<[1,32,60,256]>,
||aten::sub.Tensor|4| +|108|Tensor<[1,32,60,1024]>,
||aten::sub.Tensor|4| +|109|Tensor<[1,32,40,1024]>,
||aten::sub.Tensor|4| +|110|Tensor<[1,32,30,1024]>,
||aten::sub.Tensor|4| +|111|Tensor<[1,32,30,4096]>,
||aten::sub.Tensor|4| +|112|Tensor<[1,32,20,4096]>,
||aten::sub.Tensor|4| +|113|Tensor<[1,1]>,
||prims::convert_element_type|4| +|114|Tensor<[1,320]>,
||prims::convert_element_type|5| +|115|Tensor<[320,1280]>,
||prims::convert_element_type|5| +|116|Tensor<[1,1280]>,
||prims::convert_element_type|5| +|117|Tensor<[1280,1280]>,
||prims::convert_element_type|5| +|118|Tensor<[1,320,64,64]>,
||prims::convert_element_type|5| +|119|Tensor<[1280,320]>,
||prims::convert_element_type|5| +|120|Tensor<[1,8,4096,40]>,
||prims::convert_element_type|5| +|121|Tensor<[4096,320]>,
||prims::convert_element_type|5| +|122|Tensor<[320,320]>,
||prims::convert_element_type|5| +|123|Tensor<[1,8,9,40]>,
||prims::convert_element_type|5| +|124|Tensor<[2560]>,
||prims::convert_element_type|5| +|125|Tensor<[320,2560]>,
||prims::convert_element_type|5| +|126|Tensor<[4096,2560]>,
||prims::convert_element_type|5| +|127|Tensor<[4096,1280]>,
||prims::convert_element_type|5| +|128|Tensor<[1,320,32,32]>,
||prims::convert_element_type|5| +|129|Tensor<[1280,640]>,
||prims::convert_element_type|5| +|130|Tensor<[1,640]>,
||prims::convert_element_type|5| +|131|Tensor<[1,640,32,32]>,
||prims::convert_element_type|5| +|132|Tensor<[1,8,1024,80]>,
||prims::convert_element_type|5| +|133|Tensor<[1024,640]>,
||prims::convert_element_type|5| +|134|Tensor<[640,640]>,
||prims::convert_element_type|5| +|135|Tensor<[1,8,9,80]>,
||prims::convert_element_type|5| +|136|Tensor<[5120]>,
||prims::convert_element_type|5| +|137|Tensor<[640,5120]>,
||prims::convert_element_type|5| +|138|Tensor<[1024,5120]>,
||prims::convert_element_type|5| +|139|Tensor<[1024,2560]>,
||prims::convert_element_type|5| +|140|Tensor<[2560,640]>,
||prims::convert_element_type|5| +|141|Tensor<[1,640,16,16]>,
||prims::convert_element_type|5| +|142|Tensor<[1,1280,16,16]>,
||prims::convert_element_type|5| +|143|Tensor<[1,8,256,160]>,
||prims::convert_element_type|5| +|144|Tensor<[256,1280]>,
||prims::convert_element_type|5| +|145|Tensor<[1,8,9,160]>,
||prims::convert_element_type|5| +|146|Tensor<[10240]>,
||prims::convert_element_type|5| +|147|Tensor<[1280,10240]>,
||prims::convert_element_type|5| +|148|Tensor<[256,10240]>,
||prims::convert_element_type|5| +|149|Tensor<[256,5120]>,
||prims::convert_element_type|5| +|150|Tensor<[5120,1280]>,
||prims::convert_element_type|5| +|151|Tensor<[1,1280,8,8]>,
||prims::convert_element_type|5| +|152|Tensor<[1,8,64,160]>,
||prims::convert_element_type|5| +|153|Tensor<[64,1280]>,
||prims::convert_element_type|5| +|154|Tensor<[64,10240]>,
||prims::convert_element_type|5| +|155|Tensor<[64,5120]>,
||prims::convert_element_type|5| +|156|Tensor<[1,2560,8,8]>,
||prims::convert_element_type|5| +|157|Tensor<[16]>,
||prims::convert_element_type|4| +|158|Tensor<[1,2560,16,16]>,
||prims::convert_element_type|5| +|159|Tensor<[1,1920,16,16]>,
||prims::convert_element_type|5| +|160|Tensor<[1,1280,32,32]>,
||prims::convert_element_type|5| +|161|Tensor<[1,1920,32,32]>,
||prims::convert_element_type|5| +|162|Tensor<[1,960,32,32]>,
||prims::convert_element_type|5| +|163|Tensor<[1,640,64,64]>,
||prims::convert_element_type|5| +|164|Tensor<[1,960,64,64]>,
||prims::convert_element_type|5| +|165|Tensor<[1,1,25,25]>,
||aten::add.Tensor|4| +|166|Tensor<[1,25,768]>,
||aten::embedding|4| +|167|Tensor<[1,25,3072]>,
||aten::gelu|4| +|168|Tensor<[25,768]>,
||prims::convert_element_type|5| +|169|Tensor<[1,12,25,64]>,
||prims::convert_element_type|5| +|170|Tensor<[25,3072]>,
||prims::convert_element_type|5| +|171|Tensor<[2]>,
||prims::convert_element_type|5| +|172|Tensor<[768,2]>,
||prims::convert_element_type|5| +|173|Tensor<[25,2]>,
||prims::convert_element_type|5| +|174|Tensor<[1,768]>,
||prims::convert_element_type|5| +|175|Tensor<[768,1]>,
||prims::convert_element_type|5| +|176|Tensor<[192]>,
||aten::add.Tensor|4| +|177|Tensor<[1,1445,768]>,
||aten::gelu|4| +|178|Tensor<[1,1445,192]>,
||aten::sub.Tensor|4| +|179|Tensor<[1445,192]>,
||prims::convert_element_type|5| +|180|Tensor<[192,192]>,
||prims::convert_element_type|5| +|181|Tensor<[1,3,1445,64]>,
||prims::convert_element_type|5| +|182|Tensor<[192,768]>,
||prims::convert_element_type|5| +|183|Tensor<[1445,768]>,
||prims::convert_element_type|5| +|184|Tensor<[768,192]>,
||prims::convert_element_type|5| +|185|Tensor<[100,192]>,
||prims::convert_element_type|5| +|186|Tensor<[92]>,
||prims::convert_element_type|5| +|187|Tensor<[192,92]>,
||prims::convert_element_type|5| +|188|Tensor<[100,92]>,
||prims::convert_element_type|5| +|189|Tensor<[4]>,
||prims::convert_element_type|5| +|190|Tensor<[192,4]>,
||prims::convert_element_type|5| +|191|Tensor<[100,4]>,
||prims::convert_element_type|5| +|192|Tensor<[1,256,14,14]>,
||aten::sub.Tensor|4| +|193|Tensor<[1,512,7,7]>,
||aten::sub.Tensor|4| +|194|Tensor<[1,12,8,8]>,
||aten::_softmax|4| +|195|Tensor<[1,8,768]>,
||aten::embedding|4| +|196|Tensor<[1,3072,8]>,
||aten::gelu|4| +|197|Tensor<[1,1,1,8]>,
||prims::convert_element_type|4| +|198|Tensor<[3]>,
||prims::convert_element_type|5| +|199|Tensor<[768,3]>,
||prims::convert_element_type|5| +|200|Tensor<[1,3]>,
||prims::convert_element_type|5| +|201|Tensor<[1,8,256,2048]>,
||aten::_softmax|4| +|202|Tensor<[1,8,256,256]>,
||aten::_softmax|4| +|203|Tensor<[1,8,2048,256]>,
||aten::_softmax|4| +|204|Tensor<[1,2048,768]>,
||aten::embedding|4| +|205|Tensor<[2048,768]>,
||aten::embedding|4| +|206|Tensor<[1,1,1,2048]>,
||prims::convert_element_type|4| +|207|Tensor<[1280,256]>,
||prims::convert_element_type|5| +|208|Tensor<[768,256]>,
||prims::convert_element_type|5| +|209|Tensor<[768,1280]>,
||prims::convert_element_type|5| +|210|Tensor<[2048,1280]>,
||prims::convert_element_type|5| +|211|Tensor<[1280,768]>,
||prims::convert_element_type|5| +|212|Tensor<[1024,1,1]>,
||aten::add.Tensor|4| +|213|Tensor<[2048,1,1]>,
||aten::add.Tensor|4| +|214|Tensor<[1,256,56,56]>,
||aten::sub.Tensor|4| +|215|Tensor<[1,1024,14,14]>,
||aten::sub.Tensor|4| +|216|Tensor<[1,512,14,14]>,
||aten::sub.Tensor|4| +|217|Tensor<[1,2048,7,7]>,
||aten::sub.Tensor|4| +|218|Tensor<[1024]>,
||prims::convert_element_type|5| +|219|Tensor<[1,2048]>,
||prims::convert_element_type|5| +|220|Tensor<[2048,1000]>,
||prims::convert_element_type|5| +|221|Tensor<[1,12,201,201]>,
||aten::_softmax|4| +|222|Tensor<[1,193,768]>,
||aten::embedding|4| +|223|Tensor<[1,201,3072]>,
||aten::gelu|4| +|224|Tensor<[1,1536]>,
||aten::gelu|4| +|225|Tensor<[1536]>,
||aten::mul.Tensor|4| +|226|Tensor<[1,201,768]>,
||aten::sub.Tensor|4| +|227|Tensor<[1,1,384,512]>,
||prims::convert_element_type|4| +|228|Tensor<[12]>,
||prims::convert_element_type|4| +|229|Tensor<[1,1,12,16]>,
||prims::convert_element_type|4| +|230|Tensor<[1,1,1,201]>,
||prims::convert_element_type|4| +|231|Tensor<[201,768]>,
||prims::convert_element_type|5| +|232|Tensor<[201,3072]>,
||prims::convert_element_type|5| +|233|Tensor<[768,1536]>,
||prims::convert_element_type|5| +|234|Tensor<[3129]>,
||prims::convert_element_type|5| +|235|Tensor<[1536,3129]>,
||prims::convert_element_type|5| +|236|Tensor<[1,3129]>,
||prims::convert_element_type|5| +|237|Tensor<[1,9216]>,
||prims::convert_element_type|5| +|238|Tensor<[9216,128]>,
||prims::convert_element_type|5| +|239|Tensor<[1,128]>,
||prims::convert_element_type|5| +|240|Tensor<[10]>,
||prims::convert_element_type|5| +|241|Tensor<[128,10]>,
||prims::convert_element_type|5| +|242|Tensor<[16,19,19]>,
||aten::_softmax|4| +|243|Tensor<[1,19,1024]>,
||aten::embedding|4| +|244|Tensor<[19]>,
||aten::floor_divide|4| +|245|Tensor<[1,19,4096]>,
||aten::gelu|4| +|246|Tensor<[19,1024]>,
||aten::index_select|4| +|247|Tensor<[19,19]>,
||prims::convert_element_type|5| +|248|Tensor<[1,1,19,19]>,
||prims::convert_element_type|4| +|249|Tensor<[1024,1024]>,
||prims::convert_element_type|5| +|250|Tensor<[4096]>,
||prims::convert_element_type|5| +|251|Tensor<[1024,4096]>,
||prims::convert_element_type|5| +|252|Tensor<[19,4096]>,
||prims::convert_element_type|5| +|253|Tensor<[4096,1024]>,
||prims::convert_element_type|5| +|254|Tensor<[19,256008]>,
||prims::convert_element_type|5| +|255|Tensor<[14,1,1]>,
||aten::add.Tensor|4| +|256|Tensor<[24,1,1]>,
||aten::add.Tensor|4| +|257|Tensor<[40,1,1]>,
||aten::add.Tensor|4| +|258|Tensor<[68,1,1]>,
||aten::add.Tensor|4| +|259|Tensor<[16,1,1]>,
||aten::add.Tensor|4| +|260|Tensor<[28,1,1]>,
||aten::add.Tensor|4| +|261|Tensor<[46,1,1]>,
||aten::add.Tensor|4| +|262|Tensor<[78,1,1]>,
||aten::add.Tensor|4| +|263|Tensor<[134,1,1]>,
||aten::add.Tensor|4| +|264|Tensor<[20,1,1]>,
||aten::add.Tensor|4| +|265|Tensor<[34,1,1]>,
||aten::add.Tensor|4| +|266|Tensor<[58,1,1]>,
||aten::add.Tensor|4| +|267|Tensor<[98,1,1]>,
||aten::add.Tensor|4| +|268|Tensor<[168,1,1]>,
||aten::add.Tensor|4| +|269|Tensor<[320,1,1]>,
||aten::add.Tensor|4| +|270|Tensor<[116,1,1]>,
||aten::add.Tensor|4| +|271|Tensor<[196,1,1]>,
||aten::add.Tensor|4| +|272|Tensor<[334,1,1]>,
||aten::add.Tensor|4| +|273|Tensor<[640,1,1]>,
||aten::add.Tensor|4| +|274|Tensor<[160,1,1]>,
||aten::add.Tensor|4| +|275|Tensor<[272,1,1]>,
||aten::add.Tensor|4| +|276|Tensor<[462,1,1]>,
||aten::add.Tensor|4| +|277|Tensor<[1,14,56,56]>,
||aten::sub.Tensor|4| +|278|Tensor<[1,24,56,56]>,
||aten::sub.Tensor|4| +|279|Tensor<[1,40,56,56]>,
||aten::sub.Tensor|4| +|280|Tensor<[1,68,56,56]>,
||aten::sub.Tensor|4| +|281|Tensor<[1,16,28,28]>,
||aten::sub.Tensor|4| +|282|Tensor<[1,28,28,28]>,
||aten::sub.Tensor|4| +|283|Tensor<[1,46,28,28]>,
||aten::sub.Tensor|4| +|284|Tensor<[1,78,28,28]>,
||aten::sub.Tensor|4| +|285|Tensor<[1,134,28,28]>,
||aten::sub.Tensor|4| +|286|Tensor<[1,20,28,28]>,
||aten::sub.Tensor|4| +|287|Tensor<[1,34,28,28]>,
||aten::sub.Tensor|4| +|288|Tensor<[1,58,28,28]>,
||aten::sub.Tensor|4| +|289|Tensor<[1,98,28,28]>,
||aten::sub.Tensor|4| +|290|Tensor<[1,168,28,28]>,
||aten::sub.Tensor|4| +|291|Tensor<[1,320,28,28]>,
||aten::sub.Tensor|4| +|292|Tensor<[1,40,14,14]>,
||aten::sub.Tensor|4| +|293|Tensor<[1,68,14,14]>,
||aten::sub.Tensor|4| +|294|Tensor<[1,116,14,14]>,
||aten::sub.Tensor|4| +|295|Tensor<[1,196,14,14]>,
||aten::sub.Tensor|4| +|296|Tensor<[1,334,14,14]>,
||aten::sub.Tensor|4| +|297|Tensor<[1,640,14,14]>,
||aten::sub.Tensor|4| +|298|Tensor<[1,160,7,7]>,
||aten::sub.Tensor|4| +|299|Tensor<[1,272,7,7]>,
||aten::sub.Tensor|4| +|300|Tensor<[1,462,7,7]>,
||aten::sub.Tensor|4| +|301|Tensor<[1,1024,7,7]>,
||aten::sub.Tensor|4| +|302|Tensor<[14]>,
||prims::convert_element_type|5| +|303|Tensor<[24]>,
||prims::convert_element_type|5| +|304|Tensor<[68]>,
||prims::convert_element_type|5| +|305|Tensor<[28]>,
||prims::convert_element_type|5| +|306|Tensor<[46]>,
||prims::convert_element_type|5| +|307|Tensor<[78]>,
||prims::convert_element_type|5| +|308|Tensor<[134]>,
||prims::convert_element_type|5| +|309|Tensor<[20]>,
||prims::convert_element_type|5| +|310|Tensor<[34]>,
||prims::convert_element_type|5| +|311|Tensor<[58]>,
||prims::convert_element_type|5| +|312|Tensor<[98]>,
||prims::convert_element_type|5| +|313|Tensor<[168]>,
||prims::convert_element_type|5| +|314|Tensor<[116]>,
||prims::convert_element_type|5| +|315|Tensor<[196]>,
||prims::convert_element_type|5| +|316|Tensor<[334]>,
||prims::convert_element_type|5| +|317|Tensor<[160]>,
||prims::convert_element_type|5| +|318|Tensor<[272]>,
||prims::convert_element_type|5| +|319|Tensor<[462]>,
||prims::convert_element_type|5| +|320|Tensor<[1,1024]>,
||prims::convert_element_type|5| +|321|Tensor<[1024,1000]>,
||prims::convert_element_type|5| +|322|Tensor<[1,32,512,512]>,
||aten::sub.Tensor|4| +|323|Tensor<[1,64,256,256]>,
||aten::sub.Tensor|4| +|324|Tensor<[1,32,256,256]>,
||aten::sub.Tensor|4| +|325|Tensor<[1,128,128,128]>,
||aten::sub.Tensor|4| +|326|Tensor<[1,64,128,128]>,
||aten::sub.Tensor|4| +|327|Tensor<[1,256,64,64]>,
||aten::sub.Tensor|4| +|328|Tensor<[1,128,64,64]>,
||aten::sub.Tensor|4| +|329|Tensor<[1,512,32,32]>,
||aten::sub.Tensor|4| +|330|Tensor<[1,256,32,32]>,
||aten::sub.Tensor|4| +|331|Tensor<[1,1024,16,16]>,
||aten::sub.Tensor|4| +|332|Tensor<[1,512,16,16]>,
||aten::sub.Tensor|4| +|333|Tensor<[1,256,16,16]>,
||aten::sub.Tensor|4| +|334|Tensor<[1,128,32,32]>,
||aten::sub.Tensor|4| +|335|Tensor<[1,32,1536]>,
||aten::embedding|4| +|336|Tensor<[16,1,32]>,
||prims::convert_element_type|5| +|337|Tensor<[4608]>,
||prims::convert_element_type|5| +|338|Tensor<[32,1536]>,
||prims::convert_element_type|5| +|339|Tensor<[1536,4608]>,
||prims::convert_element_type|5| +|340|Tensor<[32,4608]>,
||prims::convert_element_type|5| +|341|Tensor<[1,16,32,32]>,
||prims::convert_element_type|5| +|342|Tensor<[1536,1536]>,
||prims::convert_element_type|5| +|343|Tensor<[6144]>,
||prims::convert_element_type|5| +|344|Tensor<[1536,6144]>,
||prims::convert_element_type|5| +|345|Tensor<[32,6144]>,
||prims::convert_element_type|5| +|346|Tensor<[6144,1536]>,
||prims::convert_element_type|5| +|347|Tensor<[1,1,16,16]>,
||aten::add.Tensor|4| +|348|Tensor<[1,16,768]>,
||aten::embedding|4| +|349|Tensor<[1,16,3072]>,
||aten::gelu|4| +|350|Tensor<[16,768]>,
||prims::convert_element_type|5| +|351|Tensor<[1,12,16,64]>,
||prims::convert_element_type|5| +|352|Tensor<[16,3072]>,
||prims::convert_element_type|5| +|353|Tensor<[1,64,224,224]>,
||aten::sub.Tensor|4| +|354|Tensor<[1,128,112,112]>,
||aten::sub.Tensor|4| +|355|Tensor<[1,1,19200,300]>,
||aten::_softmax|4| +|356|Tensor<[1,2,4800,300]>,
||aten::_softmax|4| +|357|Tensor<[1,5,1200,300]>,
||aten::_softmax|4| +|358|Tensor<[1,8,300,300]>,
||aten::_softmax|4| +|359|Tensor<[1,19200,256]>,
||aten::gelu|4| +|360|Tensor<[1,4800,512]>,
||aten::gelu|4| +|361|Tensor<[1,1200,1280]>,
||aten::gelu|4| +|362|Tensor<[1,300,2048]>,
||aten::gelu|4| +|363|Tensor<[1,19200,64]>,
||aten::sub.Tensor|4| +|364|Tensor<[1,300,64]>,
||aten::sub.Tensor|4| +|365|Tensor<[1,4800,128]>,
||aten::sub.Tensor|4| +|366|Tensor<[1,300,128]>,
||aten::sub.Tensor|4| +|367|Tensor<[1,1200,320]>,
||aten::sub.Tensor|4| +|368|Tensor<[1,300,320]>,
||aten::sub.Tensor|4| +|369|Tensor<[1,300,512]>,
||aten::sub.Tensor|4| +|370|Tensor<[30,1]>,
||aten::sub.Tensor|4| +|371|Tensor<[1,64,30,40]>,
||aten::sub.Tensor|4| +|372|Tensor<[1,32,30,40]>,
||aten::sub.Tensor|4| +|373|Tensor<[80]>,
||aten::sub.Tensor|4| +|374|Tensor<[60,1]>,
||aten::sub.Tensor|4| +|375|Tensor<[1,64,60,80]>,
||aten::sub.Tensor|4| +|376|Tensor<[1,32,60,80]>,
||aten::sub.Tensor|4| +|377|Tensor<[120,1]>,
||aten::sub.Tensor|4| +|378|Tensor<[1,64,120,160]>,
||aten::sub.Tensor|4| +|379|Tensor<[1,32,120,160]>,
||aten::sub.Tensor|4| +|380|Tensor<[240,1]>,
||aten::sub.Tensor|4| +|381|Tensor<[480,1]>,
||aten::sub.Tensor|4| +|382|Tensor<[19200,64]>,
||prims::convert_element_type|5| +|383|Tensor<[64,64]>,
||prims::convert_element_type|5| +|384|Tensor<[300,64]>,
||prims::convert_element_type|5| +|385|Tensor<[64,256]>,
||prims::convert_element_type|5| +|386|Tensor<[19200,256]>,
||prims::convert_element_type|5| +|387|Tensor<[4800,128]>,
||prims::convert_element_type|5| +|388|Tensor<[128,128]>,
||prims::convert_element_type|5| +|389|Tensor<[300,128]>,
||prims::convert_element_type|5| +|390|Tensor<[128,512]>,
||prims::convert_element_type|5| +|391|Tensor<[4800,512]>,
||prims::convert_element_type|5| +|392|Tensor<[1200,320]>,
||prims::convert_element_type|5| +|393|Tensor<[300,320]>,
||prims::convert_element_type|5| +|394|Tensor<[1200,1280]>,
||prims::convert_element_type|5| +|395|Tensor<[300,512]>,
||prims::convert_element_type|5| +|396|Tensor<[512,512]>,
||prims::convert_element_type|5| +|397|Tensor<[512,2048]>,
||prims::convert_element_type|5| +|398|Tensor<[300,2048]>,
||prims::convert_element_type|5| +|399|Tensor<[1,64,15,20]>,
||prims::convert_element_type|5| +|400|Tensor<[30]>,
||prims::convert_element_type|4| +|401|Tensor<[60]>,
||prims::convert_element_type|4| +|402|Tensor<[120]>,
||prims::convert_element_type|4| +|403|Tensor<[240]>,
||prims::convert_element_type|4| +|404|Tensor<[1,64,240,320]>,
||prims::convert_element_type|5| +|405|Tensor<[480]>,
||prims::convert_element_type|4| +|406|Tensor<[1,64,480,640]>,
||prims::convert_element_type|5| +|407|Tensor<[1,197,3072]>,
||aten::gelu|4| +|408|Tensor<[1,197,768]>,
||aten::sub.Tensor|4| +|409|Tensor<[1,3,224,224]>,
||prims::convert_element_type|5| +|410|Tensor<[197,768]>,
||prims::convert_element_type|5| +|411|Tensor<[1,12,197,64]>,
||prims::convert_element_type|5| +|412|Tensor<[197,3072]>,
||prims::convert_element_type|5| +|413|Tensor<[768,1000]>,
||prims::convert_element_type|5| +|414|Tensor<[1,1,16384,256]>,
||aten::_softmax|4| +|415|Tensor<[1,2,4096,256]>,
||aten::_softmax|4| +|416|Tensor<[1,5,1024,256]>,
||aten::_softmax|4| +|417|Tensor<[1,16384,128]>,
||aten::gelu|4| +|418|Tensor<[1,4096,256]>,
||aten::gelu|4| +|419|Tensor<[1,256,1024]>,
||aten::gelu|4| +|420|Tensor<[1,16384,32]>,
||aten::sub.Tensor|4| +|421|Tensor<[1,256,32]>,
||aten::sub.Tensor|4| +|422|Tensor<[1,4096,64]>,
||aten::sub.Tensor|4| +|423|Tensor<[1,256,64]>,
||aten::sub.Tensor|4| +|424|Tensor<[1,1024,160]>,
||aten::sub.Tensor|4| +|425|Tensor<[1,256,160]>,
||aten::sub.Tensor|4| +|426|Tensor<[128,1]>,
||aten::sub.Tensor|4| +|427|Tensor<[1,256,128,128]>,
||aten::sub.Tensor|4| +|428|Tensor<[16384,32]>,
||prims::convert_element_type|5| +|429|Tensor<[256,32]>,
||prims::convert_element_type|5| +|430|Tensor<[32,128]>,
||prims::convert_element_type|5| +|431|Tensor<[16384,128]>,
||prims::convert_element_type|5| +|432|Tensor<[4096,64]>,
||prims::convert_element_type|5| +|433|Tensor<[256,64]>,
||prims::convert_element_type|5| +|434|Tensor<[4096,256]>,
||prims::convert_element_type|5| +|435|Tensor<[1024,160]>,
||prims::convert_element_type|5| +|436|Tensor<[160,160]>,
||prims::convert_element_type|5| +|437|Tensor<[256,160]>,
||prims::convert_element_type|5| +|438|Tensor<[160,640]>,
||prims::convert_element_type|5| +|439|Tensor<[256,1024]>,
||prims::convert_element_type|5| +|440|Tensor<[1,1,1,7]>,
||aten::add.Tensor|4| +|441|Tensor<[4544]>,
||aten::add.Tensor|4| +|442|Tensor<[1,7,4544]>,
||aten::embedding|4| +|443|Tensor<[1,7,18176]>,
||aten::gelu|4| +|444|Tensor<[1,1,7]>,
||prims::convert_element_type|4| +|445|Tensor<[1,7,64]>,
||prims::convert_element_type|5| +|446|Tensor<[1,71,7,64]>,
||prims::convert_element_type|5| +|447|Tensor<[1,1,7,64]>,
||prims::convert_element_type|5| +|448|Tensor<[96,1,1]>,
||aten::add.Tensor|4| +|449|Tensor<[144,1,1]>,
||aten::add.Tensor|4| +|450|Tensor<[192,1,1]>,
||aten::add.Tensor|4| +|451|Tensor<[384,1,1]>,
||aten::add.Tensor|4| +|452|Tensor<[576,1,1]>,
||aten::add.Tensor|4| +|453|Tensor<[960,1,1]>,
||aten::add.Tensor|4| +|454|Tensor<[1280,1,1]>,
||aten::add.Tensor|4| +|455|Tensor<[1,16,112,112]>,
||aten::sub.Tensor|4| +|456|Tensor<[1,96,112,112]>,
||aten::sub.Tensor|4| +|457|Tensor<[1,96,56,56]>,
||aten::sub.Tensor|4| +|458|Tensor<[1,144,56,56]>,
||aten::sub.Tensor|4| +|459|Tensor<[1,144,28,28]>,
||aten::sub.Tensor|4| +|460|Tensor<[1,32,28,28]>,
||aten::sub.Tensor|4| +|461|Tensor<[1,192,28,28]>,
||aten::sub.Tensor|4| +|462|Tensor<[1,192,14,14]>,
||aten::sub.Tensor|4| +|463|Tensor<[1,64,14,14]>,
||aten::sub.Tensor|4| +|464|Tensor<[1,384,14,14]>,
||aten::sub.Tensor|4| +|465|Tensor<[1,96,14,14]>,
||aten::sub.Tensor|4| +|466|Tensor<[1,576,14,14]>,
||aten::sub.Tensor|4| +|467|Tensor<[1,576,7,7]>,
||aten::sub.Tensor|4| +|468|Tensor<[1,960,7,7]>,
||aten::sub.Tensor|4| +|469|Tensor<[1,320,7,7]>,
||aten::sub.Tensor|4| +|470|Tensor<[1,1280,7,7]>,
||aten::sub.Tensor|4| +|471|Tensor<[96]>,
||prims::convert_element_type|5| +|472|Tensor<[144]>,
||prims::convert_element_type|5| +|473|Tensor<[384]>,
||prims::convert_element_type|5| +|474|Tensor<[576]>,
||prims::convert_element_type|5| +|475|Tensor<[960]>,
||prims::convert_element_type|5| +|476|Tensor<[1280,1000]>,
||prims::convert_element_type|5| +|477|Tensor<[1,1,12,12]>,
||aten::add.Tensor|4| +|478|Tensor<[1,12,128]>,
||aten::embedding|4| +|479|Tensor<[1,12,768]>,
||aten::sub.Tensor|4| +|480|Tensor<[12,128]>,
||prims::convert_element_type|5| +|481|Tensor<[128,768]>,
||prims::convert_element_type|5| +|482|Tensor<[12,768]>,
||prims::convert_element_type|5| +|483|Tensor<[1,12,12,64]>,
||prims::convert_element_type|5| +|484|Tensor<[12,3072]>,
||prims::convert_element_type|5| +|485|Tensor<[12,2]>,
||prims::convert_element_type|5| +|486|Tensor<[1,1,9,9]>,
||aten::add.Tensor|4| +|487|Tensor<[1,9,128]>,
||aten::embedding|4| +|488|Tensor<[1,9,768]>,
||aten::sub.Tensor|4| +|489|Tensor<[9,128]>,
||prims::convert_element_type|5| +|490|Tensor<[9,768]>,
||prims::convert_element_type|5| +|491|Tensor<[1,12,9,64]>,
||prims::convert_element_type|5| +|492|Tensor<[9,3072]>,
||prims::convert_element_type|5| +|493|Tensor<[768,128]>,
||prims::convert_element_type|5| +|494|Tensor<[30000]>,
||prims::convert_element_type|5| +|495|Tensor<[128,30000]>,
||prims::convert_element_type|5| +|496|Tensor<[9,30000]>,
||prims::convert_element_type|5| +|497|Tensor<[1,9,2048]>,
||aten::sub.Tensor|4| +|498|Tensor<[128,2048]>,
||prims::convert_element_type|5| +|499|Tensor<[9,2048]>,
||prims::convert_element_type|5| +|500|Tensor<[2048,2048]>,
||prims::convert_element_type|5| +|501|Tensor<[1,16,9,128]>,
||prims::convert_element_type|5| +|502|Tensor<[8192]>,
||prims::convert_element_type|5| +|503|Tensor<[2048,8192]>,
||prims::convert_element_type|5| +|504|Tensor<[9,8192]>,
||prims::convert_element_type|5| +|505|Tensor<[8192,2048]>,
||prims::convert_element_type|5| +|506|Tensor<[2048,128]>,
||prims::convert_element_type|5| +|507|Tensor<[1,9,1024]>,
||aten::sub.Tensor|4| +|508|Tensor<[128,1024]>,
||prims::convert_element_type|5| +|509|Tensor<[9,1024]>,
||prims::convert_element_type|5| +|510|Tensor<[1,16,9,64]>,
||prims::convert_element_type|5| +|511|Tensor<[9,4096]>,
||prims::convert_element_type|5| +|512|Tensor<[1024,128]>,
||prims::convert_element_type|5| +|513|Tensor<[1,9,4096]>,
||aten::sub.Tensor|4| +|514|Tensor<[128,4096]>,
||prims::convert_element_type|5| +|515|Tensor<[4096,4096]>,
||prims::convert_element_type|5| +|516|Tensor<[1,64,9,64]>,
||prims::convert_element_type|5| +|517|Tensor<[16384]>,
||prims::convert_element_type|5| +|518|Tensor<[4096,16384]>,
||prims::convert_element_type|5| +|519|Tensor<[9,16384]>,
||prims::convert_element_type|5| +|520|Tensor<[16384,4096]>,
||prims::convert_element_type|5| +|521|Tensor<[4096,128]>,
||prims::convert_element_type|5| +|522|Tensor<[1,2]>,
||prims::convert_element_type|5| +|523|Tensor<[1,1,14,14]>,
||aten::add.Tensor|4| +|524|Tensor<[1,14,128]>,
||aten::embedding|4| +|525|Tensor<[1,14,768]>,
||aten::sub.Tensor|4| +|526|Tensor<[14,128]>,
||prims::convert_element_type|5| +|527|Tensor<[14,768]>,
||prims::convert_element_type|5| +|528|Tensor<[1,12,14,64]>,
||prims::convert_element_type|5| +|529|Tensor<[14,3072]>,
||prims::convert_element_type|5| +|530|Tensor<[14,2]>,
||prims::convert_element_type|5| +|531|Tensor<[2,1,7,7]>,
||aten::add.Tensor|4| +|532|Tensor<[1,50,768]>,
||aten::embedding|4| +|533|Tensor<[2,7,512]>,
||aten::embedding|4| +|534|Tensor<[1,7,512]>,
||aten::embedding|4| +|535|Tensor<[50,768]>,
||prims::convert_element_type|5| +|536|Tensor<[1,12,50,64]>,
||prims::convert_element_type|5| +|537|Tensor<[50,3072]>,
||prims::convert_element_type|5| +|538|Tensor<[14,512]>,
||prims::convert_element_type|5| +|539|Tensor<[2,8,7,64]>,
||prims::convert_element_type|5| +|540|Tensor<[14,2048]>,
||prims::convert_element_type|5| +|541|Tensor<[2048,512]>,
||prims::convert_element_type|5| +|542|Tensor<[2,7]>,
||prims::convert_element_type|4| +|543|Tensor<[1,16,197,197]>,
||aten::_softmax|4| +|544|Tensor<[197]>,
||aten::floor_divide|4| +|545|Tensor<[1,197,4096]>,
||aten::gelu|4| +|546|Tensor<[1,197,1024]>,
||aten::sub.Tensor|4| +|547|Tensor<[27]>,
||aten::sub.Tensor|4| +|548|Tensor<[27,1]>,
||aten::sub.Tensor|4| +|549|Tensor<[197,1024]>,
||prims::convert_element_type|5| +|550|Tensor<[1,16,27,27]>,
||prims::convert_element_type|5| +|551|Tensor<[197,4096]>,
||prims::convert_element_type|5| +|552|Tensor<[1,12,197,197]>,
||aten::_softmax|4| +|553|Tensor<[1,12,27,27]>,
||prims::convert_element_type|5| +|554|Tensor<[1,784]>,
||prims::convert_element_type|5| +|555|Tensor<[784,128]>,
||prims::convert_element_type|5| +|556|Tensor<[128,64]>,
||prims::convert_element_type|5| +|557|Tensor<[1,64]>,
||prims::convert_element_type|5| +|558|Tensor<[64,12]>,
||prims::convert_element_type|5| +|559|Tensor<[1,12]>,
||prims::convert_element_type|5| +|560|Tensor<[12,3]>,
||prims::convert_element_type|5| +|561|Tensor<[3,12]>,
||prims::convert_element_type|5| +|562|Tensor<[12,64]>,
||prims::convert_element_type|5| +|563|Tensor<[64,128]>,
||prims::convert_element_type|5| +|564|Tensor<[784]>,
||prims::convert_element_type|5| +|565|Tensor<[128,784]>,
||prims::convert_element_type|5| diff --git a/docs/ops/stablehlo/stablehlo.convolution.md b/docs/ops/stablehlo/stablehlo.convolution.md new file mode 100644 index 00000000..df245bdc --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.convolution.md @@ -0,0 +1,311 @@ + +### stablehlo.convolution::ttnn.conv2d + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,3,224,224]>,
Tensor<[32,3,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|1|Tensor<[1,32,112,112]>,
Tensor<[32,1,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 32
|ttnn.conv2d|aten::convolution|5| +|2|Tensor<[1,32,112,112]>,
Tensor<[64,32,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|3|Tensor<[1,64,112,112]>,
Tensor<[64,1,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 64
|ttnn.conv2d|aten::convolution|5| +|4|Tensor<[1,64,56,56]>,
Tensor<[128,64,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|5|Tensor<[1,128,56,56]>,
Tensor<[128,1,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 128
|ttnn.conv2d|aten::convolution|5| +|6|Tensor<[1,128,56,56]>,
Tensor<[128,128,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|7|Tensor<[1,128,56,56]>,
Tensor<[128,1,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 128
|ttnn.conv2d|aten::convolution|5| +|8|Tensor<[1,128,28,28]>,
Tensor<[256,128,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|9|Tensor<[1,256,28,28]>,
Tensor<[256,1,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 256
|ttnn.conv2d|aten::convolution|5| +|10|Tensor<[1,256,28,28]>,
Tensor<[256,256,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|11|Tensor<[1,256,28,28]>,
Tensor<[512,256,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|12|Tensor<[1,512,28,28]>,
Tensor<[512,1,3,3]>,
stride: [1, 1]
pad: [[2, 2], [2, 2]]
rhs_dilate: [2, 2]
batch_group_count: 1
feature_group_count: 512
|ttnn.conv2d|aten::convolution|5| +|13|Tensor<[1,512,28,28]>,
Tensor<[512,512,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|14|Tensor<[1,512,28,28]>,
Tensor<[512,1,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 512
|ttnn.conv2d|aten::convolution|5| +|15|Tensor<[1,512,28,28]>,
Tensor<[128,512,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|16|Tensor<[1,128,28,28]>,
Tensor<[128,1,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 128
|ttnn.conv2d|aten::convolution|5| +|17|Tensor<[1,128,28,28]>,
Tensor<[128,128,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|18|Tensor<[1,128,28,28]>,
Tensor<[128,128,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|19|Tensor<[1,128,28,28]>,
Tensor<[512,128,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|20|Tensor<[1,512,28,28]>,
Tensor<[19,512,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|21|Tensor<[1,512,28,28]>,
Tensor<[38,512,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|22|Tensor<[1,185,28,28]>,
Tensor<[128,185,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|23|Tensor<[1,128,28,28]>,
Tensor<[128,128,3,3]>,
stride: [1, 1]
pad: [[2, 2], [2, 2]]
rhs_dilate: [2, 2]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|24|Tensor<[1,128,28,28]>,
Tensor<[19,128,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|25|Tensor<[1,128,28,28]>,
Tensor<[38,128,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|26|Tensor<[1,256,512]>,
Tensor<[1024,256,1]>,
stride: [1]
pad: [[0, 0]]
rhs_dilate: [1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|27|Tensor<[1,1024,512]>,
Tensor<[256,1024,1]>,
stride: [1]
pad: [[0, 0]]
rhs_dilate: [1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|28|Tensor<[1,3,720,1280]>,
Tensor<[64,3,7,7]>,
stride: [2, 2]
pad: [[3, 3], [3, 3]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|29|Tensor<[1,64,180,320]>,
Tensor<[64,64,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|30|Tensor<[1,64,180,320]>,
Tensor<[64,64,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|31|Tensor<[1,64,180,320]>,
Tensor<[256,64,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|32|Tensor<[1,256,180,320]>,
Tensor<[64,256,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|33|Tensor<[1,256,180,320]>,
Tensor<[128,256,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|34|Tensor<[1,128,180,320]>,
Tensor<[128,128,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|35|Tensor<[1,128,90,160]>,
Tensor<[512,128,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|36|Tensor<[1,256,180,320]>,
Tensor<[512,256,1,1]>,
stride: [2, 2]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|37|Tensor<[1,512,90,160]>,
Tensor<[128,512,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|38|Tensor<[1,128,90,160]>,
Tensor<[128,128,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|39|Tensor<[1,512,90,160]>,
Tensor<[256,512,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|40|Tensor<[1,256,90,160]>,
Tensor<[256,256,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|41|Tensor<[1,256,45,80]>,
Tensor<[1024,256,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|42|Tensor<[1,512,90,160]>,
Tensor<[1024,512,1,1]>,
stride: [2, 2]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|43|Tensor<[1,1024,45,80]>,
Tensor<[256,1024,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|44|Tensor<[1,256,45,80]>,
Tensor<[256,256,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|45|Tensor<[1,1024,45,80]>,
Tensor<[512,1024,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|46|Tensor<[1,512,45,80]>,
Tensor<[512,512,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|47|Tensor<[1,512,23,40]>,
Tensor<[2048,512,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|48|Tensor<[1,1024,45,80]>,
Tensor<[2048,1024,1,1]>,
stride: [2, 2]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|49|Tensor<[1,2048,23,40]>,
Tensor<[512,2048,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|50|Tensor<[1,512,23,40]>,
Tensor<[512,512,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|51|Tensor<[1,2048,23,40]>,
Tensor<[256,2048,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|52|Tensor<[1,4,64,64]>,
Tensor<[320,4,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|53|Tensor<[1,320,64,64]>,
Tensor<[320,320,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|54|Tensor<[1,320,64,64]>,
Tensor<[320,320,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|55|Tensor<[1,320,64,64]>,
Tensor<[320,320,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|56|Tensor<[1,320,32,32]>,
Tensor<[640,320,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|57|Tensor<[1,640,32,32]>,
Tensor<[640,640,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|58|Tensor<[1,320,32,32]>,
Tensor<[640,320,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|59|Tensor<[1,640,32,32]>,
Tensor<[640,640,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|60|Tensor<[1,640,32,32]>,
Tensor<[640,640,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|61|Tensor<[1,640,16,16]>,
Tensor<[1280,640,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|62|Tensor<[1,1280,16,16]>,
Tensor<[1280,1280,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|63|Tensor<[1,640,16,16]>,
Tensor<[1280,640,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|64|Tensor<[1,1280,16,16]>,
Tensor<[1280,1280,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|65|Tensor<[1,1280,16,16]>,
Tensor<[1280,1280,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|66|Tensor<[1,1280,8,8]>,
Tensor<[1280,1280,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|67|Tensor<[1,1280,8,8]>,
Tensor<[1280,1280,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|68|Tensor<[1,2560,8,8]>,
Tensor<[1280,2560,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|69|Tensor<[1,2560,8,8]>,
Tensor<[1280,2560,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|70|Tensor<[1,2560,16,16]>,
Tensor<[1280,2560,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|71|Tensor<[1,2560,16,16]>,
Tensor<[1280,2560,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|72|Tensor<[1,1920,16,16]>,
Tensor<[1280,1920,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|73|Tensor<[1,1920,16,16]>,
Tensor<[1280,1920,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|74|Tensor<[1,1280,32,32]>,
Tensor<[1280,1280,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|75|Tensor<[1,1920,32,32]>,
Tensor<[640,1920,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|76|Tensor<[1,1920,32,32]>,
Tensor<[640,1920,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|77|Tensor<[1,1280,32,32]>,
Tensor<[640,1280,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|78|Tensor<[1,1280,32,32]>,
Tensor<[640,1280,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|79|Tensor<[1,960,32,32]>,
Tensor<[640,960,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|80|Tensor<[1,960,32,32]>,
Tensor<[640,960,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|81|Tensor<[1,640,64,64]>,
Tensor<[640,640,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|82|Tensor<[1,960,64,64]>,
Tensor<[320,960,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|83|Tensor<[1,960,64,64]>,
Tensor<[320,960,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|84|Tensor<[1,640,64,64]>,
Tensor<[320,640,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|85|Tensor<[1,640,64,64]>,
Tensor<[320,640,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|86|Tensor<[1,320,64,64]>,
Tensor<[4,320,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|87|Tensor<[1,3,512,672]>,
Tensor<[192,3,16,16]>,
stride: [16, 16]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|88|Tensor<[1,3,224,224]>,
Tensor<[64,3,7,7]>,
stride: [2, 2]
pad: [[3, 3], [3, 3]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|89|Tensor<[1,64,56,56]>,
Tensor<[64,64,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|90|Tensor<[1,64,56,56]>,
Tensor<[128,64,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|91|Tensor<[1,64,56,56]>,
Tensor<[128,64,1,1]>,
stride: [2, 2]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|92|Tensor<[1,128,28,28]>,
Tensor<[256,128,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|93|Tensor<[1,256,14,14]>,
Tensor<[256,256,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|94|Tensor<[1,128,28,28]>,
Tensor<[256,128,1,1]>,
stride: [2, 2]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|95|Tensor<[1,256,14,14]>,
Tensor<[512,256,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|96|Tensor<[1,512,7,7]>,
Tensor<[512,512,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|97|Tensor<[1,256,14,14]>,
Tensor<[512,256,1,1]>,
stride: [2, 2]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|98|Tensor<[1,768,8]>,
Tensor<[768,192,1]>,
stride: [1]
pad: [[0, 0]]
rhs_dilate: [1]
batch_group_count: 1
feature_group_count: 4
|ttnn.conv2d|aten::convolution|4| +|99|Tensor<[1,768,8]>,
Tensor<[768,768,1]>,
stride: [1]
pad: [[0, 0]]
rhs_dilate: [1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|100|Tensor<[1,768,8]>,
Tensor<[3072,192,1]>,
stride: [1]
pad: [[0, 0]]
rhs_dilate: [1]
batch_group_count: 1
feature_group_count: 4
|ttnn.conv2d|aten::convolution|4| +|101|Tensor<[1,3072,8]>,
Tensor<[768,768,1]>,
stride: [1]
pad: [[0, 0]]
rhs_dilate: [1]
batch_group_count: 1
feature_group_count: 4
|ttnn.conv2d|aten::convolution|4| +|102|Tensor<[1,64,56,56]>,
Tensor<[64,64,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|103|Tensor<[1,64,56,56]>,
Tensor<[256,64,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|104|Tensor<[1,256,56,56]>,
Tensor<[64,256,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|105|Tensor<[1,256,56,56]>,
Tensor<[128,256,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|106|Tensor<[1,128,56,56]>,
Tensor<[128,128,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|107|Tensor<[1,256,56,56]>,
Tensor<[512,256,1,1]>,
stride: [2, 2]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|108|Tensor<[1,512,28,28]>,
Tensor<[256,512,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|109|Tensor<[1,256,28,28]>,
Tensor<[256,256,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|110|Tensor<[1,256,14,14]>,
Tensor<[1024,256,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|111|Tensor<[1,512,28,28]>,
Tensor<[1024,512,1,1]>,
stride: [2, 2]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|112|Tensor<[1,1024,14,14]>,
Tensor<[256,1024,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|113|Tensor<[1,1024,14,14]>,
Tensor<[512,1024,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|114|Tensor<[1,512,14,14]>,
Tensor<[512,512,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|115|Tensor<[1,512,7,7]>,
Tensor<[2048,512,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|116|Tensor<[1,1024,14,14]>,
Tensor<[2048,1024,1,1]>,
stride: [2, 2]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|117|Tensor<[1,2048,7,7]>,
Tensor<[512,2048,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|118|Tensor<[1,3,384,512]>,
Tensor<[768,3,32,32]>,
stride: [32, 32]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|119|Tensor<[1,1,28,28]>,
Tensor<[32,1,3,3]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|120|Tensor<[1,32,26,26]>,
Tensor<[64,32,3,3]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|121|Tensor<[1,32,112,112]>,
Tensor<[64,32,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|122|Tensor<[1,64,56,56]>,
Tensor<[14,64,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|123|Tensor<[1,78,56,56]>,
Tensor<[24,78,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|124|Tensor<[1,24,56,56]>,
Tensor<[14,24,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|125|Tensor<[1,102,56,56]>,
Tensor<[40,102,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|126|Tensor<[1,40,56,56]>,
Tensor<[14,40,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|127|Tensor<[1,54,56,56]>,
Tensor<[24,54,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|128|Tensor<[1,142,56,56]>,
Tensor<[68,142,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|129|Tensor<[1,124,56,56]>,
Tensor<[128,124,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|130|Tensor<[1,128,28,28]>,
Tensor<[16,128,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|131|Tensor<[1,144,28,28]>,
Tensor<[28,144,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|132|Tensor<[1,28,28,28]>,
Tensor<[16,28,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|133|Tensor<[1,172,28,28]>,
Tensor<[46,172,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|134|Tensor<[1,46,28,28]>,
Tensor<[16,46,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|135|Tensor<[1,62,28,28]>,
Tensor<[28,62,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|136|Tensor<[1,218,28,28]>,
Tensor<[78,218,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|137|Tensor<[1,78,28,28]>,
Tensor<[16,78,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|138|Tensor<[1,94,28,28]>,
Tensor<[28,94,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|139|Tensor<[1,122,28,28]>,
Tensor<[46,122,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|140|Tensor<[1,296,28,28]>,
Tensor<[134,296,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|141|Tensor<[1,262,28,28]>,
Tensor<[256,262,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|142|Tensor<[1,256,28,28]>,
Tensor<[20,256,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|143|Tensor<[1,276,28,28]>,
Tensor<[34,276,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|144|Tensor<[1,34,28,28]>,
Tensor<[20,34,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|145|Tensor<[1,310,28,28]>,
Tensor<[58,310,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|146|Tensor<[1,58,28,28]>,
Tensor<[20,58,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|147|Tensor<[1,78,28,28]>,
Tensor<[34,78,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|148|Tensor<[1,368,28,28]>,
Tensor<[98,368,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|149|Tensor<[1,98,28,28]>,
Tensor<[20,98,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|150|Tensor<[1,118,28,28]>,
Tensor<[34,118,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|151|Tensor<[1,152,28,28]>,
Tensor<[58,152,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|152|Tensor<[1,466,28,28]>,
Tensor<[168,466,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|153|Tensor<[1,328,28,28]>,
Tensor<[320,328,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|154|Tensor<[1,320,14,14]>,
Tensor<[40,320,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|155|Tensor<[1,360,14,14]>,
Tensor<[68,360,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|156|Tensor<[1,68,14,14]>,
Tensor<[40,68,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|157|Tensor<[1,428,14,14]>,
Tensor<[116,428,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|158|Tensor<[1,116,14,14]>,
Tensor<[40,116,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|159|Tensor<[1,156,14,14]>,
Tensor<[68,156,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|160|Tensor<[1,544,14,14]>,
Tensor<[196,544,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|161|Tensor<[1,196,14,14]>,
Tensor<[40,196,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|162|Tensor<[1,236,14,14]>,
Tensor<[68,236,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|163|Tensor<[1,304,14,14]>,
Tensor<[116,304,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|164|Tensor<[1,740,14,14]>,
Tensor<[334,740,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|165|Tensor<[1,654,14,14]>,
Tensor<[640,654,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|166|Tensor<[1,640,7,7]>,
Tensor<[160,640,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|167|Tensor<[1,800,7,7]>,
Tensor<[272,800,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|168|Tensor<[1,272,7,7]>,
Tensor<[160,272,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|169|Tensor<[1,1072,7,7]>,
Tensor<[462,1072,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|170|Tensor<[1,782,7,7]>,
Tensor<[1024,782,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|171|Tensor<[1,3,512,512]>,
Tensor<[32,3,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|172|Tensor<[1,32,512,512]>,
Tensor<[64,32,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|173|Tensor<[1,64,256,256]>,
Tensor<[32,64,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|174|Tensor<[1,32,256,256]>,
Tensor<[64,32,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|175|Tensor<[1,64,256,256]>,
Tensor<[128,64,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|176|Tensor<[1,128,128,128]>,
Tensor<[64,128,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|177|Tensor<[1,64,128,128]>,
Tensor<[128,64,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|178|Tensor<[1,128,128,128]>,
Tensor<[256,128,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|179|Tensor<[1,256,64,64]>,
Tensor<[128,256,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|180|Tensor<[1,128,64,64]>,
Tensor<[256,128,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|181|Tensor<[1,256,64,64]>,
Tensor<[512,256,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|182|Tensor<[1,512,32,32]>,
Tensor<[256,512,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|183|Tensor<[1,256,32,32]>,
Tensor<[512,256,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|184|Tensor<[1,512,32,32]>,
Tensor<[1024,512,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|185|Tensor<[1,1024,16,16]>,
Tensor<[512,1024,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|186|Tensor<[1,512,16,16]>,
Tensor<[1024,512,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|187|Tensor<[1,1024,16,16]>,
Tensor<[255,1024,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|188|Tensor<[1,512,16,16]>,
Tensor<[256,512,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|189|Tensor<[1,768,32,32]>,
Tensor<[256,768,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|190|Tensor<[1,512,32,32]>,
Tensor<[255,512,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|191|Tensor<[1,256,32,32]>,
Tensor<[128,256,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|192|Tensor<[1,384,64,64]>,
Tensor<[128,384,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|193|Tensor<[1,256,64,64]>,
Tensor<[255,256,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|194|Tensor<[1,3,256,256]>,
Tensor<[32,3,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|195|Tensor<[1,32,256,256]>,
Tensor<[32,32,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|196|Tensor<[1,32,128,128]>,
Tensor<[64,32,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|197|Tensor<[1,64,128,128]>,
Tensor<[64,64,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|198|Tensor<[1,64,64,64]>,
Tensor<[128,64,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|199|Tensor<[1,128,64,64]>,
Tensor<[128,128,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|200|Tensor<[1,128,32,32]>,
Tensor<[256,128,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|201|Tensor<[1,256,32,32]>,
Tensor<[256,256,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|202|Tensor<[1,256,16,16]>,
Tensor<[512,256,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|203|Tensor<[1,512,16,16]>,
Tensor<[512,512,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|204|Tensor<[1,512,16,16]>,
Tensor<[2,2,256,512]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
lhs_dilate: [2, 2]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|205|Tensor<[1,512,32,32]>,
Tensor<[256,512,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|206|Tensor<[1,256,32,32]>,
Tensor<[2,2,128,256]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
lhs_dilate: [2, 2]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|207|Tensor<[1,256,64,64]>,
Tensor<[128,256,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|208|Tensor<[1,128,64,64]>,
Tensor<[2,2,64,128]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
lhs_dilate: [2, 2]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|209|Tensor<[1,128,128,128]>,
Tensor<[64,128,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|210|Tensor<[1,64,128,128]>,
Tensor<[2,2,32,64]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
lhs_dilate: [2, 2]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|211|Tensor<[1,64,256,256]>,
Tensor<[32,64,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|212|Tensor<[1,32,256,256]>,
Tensor<[1,32,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|213|Tensor<[1,1,28,28]>,
Tensor<[16,1,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|214|Tensor<[1,16,14,14]>,
Tensor<[4,16,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|215|Tensor<[1,4,7,7]>,
Tensor<[2,2,16,4]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
lhs_dilate: [2, 2]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|216|Tensor<[1,16,14,14]>,
Tensor<[2,2,1,16]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
lhs_dilate: [2, 2]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|217|Tensor<[1,3,224,224]>,
Tensor<[64,3,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|218|Tensor<[1,64,224,224]>,
Tensor<[64,64,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|219|Tensor<[1,64,112,112]>,
Tensor<[128,64,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|220|Tensor<[1,128,112,112]>,
Tensor<[128,128,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|221|Tensor<[1,128,56,56]>,
Tensor<[256,128,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|222|Tensor<[1,256,56,56]>,
Tensor<[256,256,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|223|Tensor<[1,256,28,28]>,
Tensor<[512,256,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|224|Tensor<[1,512,28,28]>,
Tensor<[512,512,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|225|Tensor<[1,512,14,14]>,
Tensor<[1024,512,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|226|Tensor<[1,1024,14,14]>,
Tensor<[1024,1024,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|227|Tensor<[1,1024,14,14]>,
Tensor<[2,2,512,1024]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
lhs_dilate: [2, 2]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|228|Tensor<[1,1024,28,28]>,
Tensor<[512,1024,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|229|Tensor<[1,512,28,28]>,
Tensor<[2,2,256,512]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
lhs_dilate: [2, 2]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|230|Tensor<[1,512,56,56]>,
Tensor<[256,512,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|231|Tensor<[1,256,56,56]>,
Tensor<[2,2,128,256]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
lhs_dilate: [2, 2]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|232|Tensor<[1,256,112,112]>,
Tensor<[128,256,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|233|Tensor<[1,128,112,112]>,
Tensor<[2,2,64,128]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
lhs_dilate: [2, 2]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|234|Tensor<[1,128,224,224]>,
Tensor<[64,128,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|235|Tensor<[1,64,224,224]>,
Tensor<[1,64,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|236|Tensor<[1,3,480,640]>,
Tensor<[64,3,7,7]>,
stride: [4, 4]
pad: [[3, 3], [3, 3]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|237|Tensor<[1,64,120,160]>,
Tensor<[64,64,8,8]>,
stride: [8, 8]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|238|Tensor<[1,256,120,160]>,
Tensor<[256,1,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 256
|ttnn.conv2d|aten::convolution|4| +|239|Tensor<[1,64,120,160]>,
Tensor<[128,64,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|240|Tensor<[1,128,60,80]>,
Tensor<[128,128,4,4]>,
stride: [4, 4]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|241|Tensor<[1,512,60,80]>,
Tensor<[512,1,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 512
|ttnn.conv2d|aten::convolution|4| +|242|Tensor<[1,128,60,80]>,
Tensor<[320,128,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|243|Tensor<[1,320,30,40]>,
Tensor<[320,320,2,2]>,
stride: [2, 2]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|244|Tensor<[1,1280,30,40]>,
Tensor<[1280,1,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1280
|ttnn.conv2d|aten::convolution|4| +|245|Tensor<[1,320,30,40]>,
Tensor<[512,320,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|246|Tensor<[1,2048,15,20]>,
Tensor<[2048,1,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 2048
|ttnn.conv2d|aten::convolution|4| +|247|Tensor<[1,512,15,20]>,
Tensor<[64,512,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|248|Tensor<[1,320,30,40]>,
Tensor<[64,320,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|249|Tensor<[1,128,30,40]>,
Tensor<[64,128,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|250|Tensor<[1,64,30,40]>,
Tensor<[32,64,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|251|Tensor<[1,32,30,40]>,
Tensor<[2,32,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|252|Tensor<[1,128,60,80]>,
Tensor<[64,128,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|253|Tensor<[1,128,60,80]>,
Tensor<[64,128,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|254|Tensor<[1,64,60,80]>,
Tensor<[32,64,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|255|Tensor<[1,32,60,80]>,
Tensor<[2,32,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|256|Tensor<[1,128,120,160]>,
Tensor<[64,128,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|257|Tensor<[1,64,120,160]>,
Tensor<[32,64,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|258|Tensor<[1,32,120,160]>,
Tensor<[2,32,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|259|Tensor<[1,64,480,640]>,
Tensor<[64,64,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|260|Tensor<[1,64,480,640]>,
Tensor<[1,64,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|261|Tensor<[1,3,224,224]>,
Tensor<[768,3,16,16]>,
stride: [16, 16]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|262|Tensor<[1,3,512,512]>,
Tensor<[32,3,7,7]>,
stride: [4, 4]
pad: [[3, 3], [3, 3]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|263|Tensor<[1,32,128,128]>,
Tensor<[32,32,8,8]>,
stride: [8, 8]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|264|Tensor<[1,128,128,128]>,
Tensor<[128,1,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 128
|ttnn.conv2d|aten::convolution|4| +|265|Tensor<[1,32,128,128]>,
Tensor<[64,32,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|266|Tensor<[1,64,64,64]>,
Tensor<[64,64,4,4]>,
stride: [4, 4]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|267|Tensor<[1,256,64,64]>,
Tensor<[256,1,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 256
|ttnn.conv2d|aten::convolution|4| +|268|Tensor<[1,64,64,64]>,
Tensor<[160,64,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|269|Tensor<[1,160,32,32]>,
Tensor<[160,160,2,2]>,
stride: [2, 2]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|270|Tensor<[1,640,32,32]>,
Tensor<[640,1,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 640
|ttnn.conv2d|aten::convolution|4| +|271|Tensor<[1,160,32,32]>,
Tensor<[256,160,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|272|Tensor<[1,1024,16,16]>,
Tensor<[1024,1,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1024
|ttnn.conv2d|aten::convolution|4| +|273|Tensor<[1,1024,128,128]>,
Tensor<[256,1024,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|274|Tensor<[1,256,128,128]>,
Tensor<[150,256,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| +|275|Tensor<[1,32,112,112]>,
Tensor<[16,32,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|276|Tensor<[1,16,112,112]>,
Tensor<[96,16,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|277|Tensor<[1,96,112,112]>,
Tensor<[96,1,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 96
|ttnn.conv2d|aten::convolution|5| +|278|Tensor<[1,96,56,56]>,
Tensor<[24,96,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|279|Tensor<[1,24,56,56]>,
Tensor<[144,24,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|280|Tensor<[1,144,56,56]>,
Tensor<[144,1,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 144
|ttnn.conv2d|aten::convolution|5| +|281|Tensor<[1,144,56,56]>,
Tensor<[24,144,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|282|Tensor<[1,144,56,56]>,
Tensor<[144,1,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 144
|ttnn.conv2d|aten::convolution|5| +|283|Tensor<[1,144,28,28]>,
Tensor<[32,144,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|284|Tensor<[1,32,28,28]>,
Tensor<[192,32,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|285|Tensor<[1,192,28,28]>,
Tensor<[192,1,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 192
|ttnn.conv2d|aten::convolution|5| +|286|Tensor<[1,192,28,28]>,
Tensor<[32,192,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|287|Tensor<[1,192,28,28]>,
Tensor<[192,1,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 192
|ttnn.conv2d|aten::convolution|5| +|288|Tensor<[1,192,14,14]>,
Tensor<[64,192,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|289|Tensor<[1,64,14,14]>,
Tensor<[384,64,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|290|Tensor<[1,384,14,14]>,
Tensor<[384,1,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 384
|ttnn.conv2d|aten::convolution|5| +|291|Tensor<[1,384,14,14]>,
Tensor<[64,384,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|292|Tensor<[1,384,14,14]>,
Tensor<[96,384,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|293|Tensor<[1,96,14,14]>,
Tensor<[576,96,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|294|Tensor<[1,576,14,14]>,
Tensor<[576,1,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 576
|ttnn.conv2d|aten::convolution|5| +|295|Tensor<[1,576,14,14]>,
Tensor<[96,576,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|296|Tensor<[1,576,14,14]>,
Tensor<[576,1,3,3]>,
stride: [2, 2]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 576
|ttnn.conv2d|aten::convolution|5| +|297|Tensor<[1,576,7,7]>,
Tensor<[160,576,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|298|Tensor<[1,160,7,7]>,
Tensor<[960,160,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|299|Tensor<[1,960,7,7]>,
Tensor<[960,1,3,3]>,
stride: [1, 1]
pad: [[1, 1], [1, 1]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 960
|ttnn.conv2d|aten::convolution|5| +|300|Tensor<[1,960,7,7]>,
Tensor<[160,960,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|301|Tensor<[1,960,7,7]>,
Tensor<[320,960,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|302|Tensor<[1,320,7,7]>,
Tensor<[1280,320,1,1]>,
stride: [1, 1]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|303|Tensor<[1,3,224,224]>,
Tensor<[768,3,32,32]>,
stride: [32, 32]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|5| +|304|Tensor<[1,3,224,224]>,
Tensor<[1024,3,16,16]>,
stride: [16, 16]
pad: [[0, 0], [0, 0]]
rhs_dilate: [1, 1]
batch_group_count: 1
feature_group_count: 1
|ttnn.conv2d|aten::convolution|4| diff --git a/docs/ops/stablehlo/stablehlo.cosine.md b/docs/ops/stablehlo/stablehlo.cosine.md new file mode 100644 index 00000000..e6b9e395 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.cosine.md @@ -0,0 +1,10 @@ + +### stablehlo.cosine::ttnn.cos + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,32,128]>,
|ttnn.cos|aten::cos|4| +|1|Tensor<[1,23,40,64]>,
|ttnn.cos|aten::cos|4| +|2|Tensor<[1,160]>,
|ttnn.cos|aten::cos|4| +|3|Tensor<[1,7,64]>,
|ttnn.cos|aten::cos|4| diff --git a/docs/ops/stablehlo/stablehlo.divide.md b/docs/ops/stablehlo/stablehlo.divide.md new file mode 100644 index 00000000..6484abac --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.divide.md @@ -0,0 +1,130 @@ + +### stablehlo.divide::ttnn.div + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,32,32,32]>,
Tensor<[1,32,32,32]>,
|ttnn.div|aten::_safe_softmax|4| +|1|Scalar,
Scalar,
|ttnn.div|aten::arange|4| +|2|Tensor<[1,32,1]>,
Tensor<[1,32,1]>,
|ttnn.div|aten::mean.dim|4| +|3|Tensor<[1,12,7,7]>,
Tensor<[1,12,7,7]>,
|ttnn.div|aten::_safe_softmax|4| +|4|Tensor<[32]>,
Tensor<[32]>,
|ttnn.div|aten::reciprocal|5| +|5|Tensor<[64]>,
Tensor<[64]>,
|ttnn.div|aten::reciprocal|5| +|6|Tensor<[128]>,
Tensor<[128]>,
|ttnn.div|aten::reciprocal|5| +|7|Tensor<[256]>,
Tensor<[256]>,
|ttnn.div|aten::reciprocal|5| +|8|Tensor<[512]>,
Tensor<[512]>,
|ttnn.div|aten::reciprocal|5| +|9|Tensor<[1,1024,512]>,
Tensor<[1,1024,512]>,
|ttnn.div|aten::gelu|4| +|10|Tensor<[1,256,256]>,
Tensor<[1,256,256]>,
|ttnn.div|aten::gelu|4| +|11|Tensor<[1,512]>,
Tensor<[1,512]>,
|ttnn.div|aten::mean.dim|4| +|12|Tensor<[8,920,920]>,
Tensor<[8,920,920]>,
|ttnn.div|aten::_softmax|4| +|13|Tensor<[8,100,100]>,
Tensor<[8,100,100]>,
|ttnn.div|aten::_softmax|4| +|14|Tensor<[8,100,920]>,
Tensor<[8,100,920]>,
|ttnn.div|aten::_softmax|4| +|15|Tensor<[1,23,40]>,
Tensor<[1,23,40]>,
|ttnn.div|aten::div.Tensor|4| +|16|Tensor<[1,23,40,128]>,
Tensor<[1,23,40,128]>,
|ttnn.div|aten::div.Tensor|4| +|17|Tensor<[1,12,10,10]>,
Tensor<[1,12,10,10]>,
|ttnn.div|aten::_safe_softmax|4| +|18|Tensor<[1,10,3072]>,
Tensor<[1,10,3072]>,
|ttnn.div|aten::gelu|4| +|19|Tensor<[1,10,768]>,
Tensor<[1,10,768]>,
|ttnn.div|aten::gelu|4| +|20|Tensor<[1,8,4096,4096]>,
Tensor<[1,8,4096,4096]>,
|ttnn.div|aten::_safe_softmax|4| +|21|Tensor<[1,8,4096,9]>,
Tensor<[1,8,4096,9]>,
|ttnn.div|aten::_safe_softmax|4| +|22|Tensor<[1,8,1024,1024]>,
Tensor<[1,8,1024,1024]>,
|ttnn.div|aten::_safe_softmax|4| +|23|Tensor<[1,8,1024,9]>,
Tensor<[1,8,1024,9]>,
|ttnn.div|aten::_safe_softmax|4| +|24|Tensor<[1,8,256,256]>,
Tensor<[1,8,256,256]>,
|ttnn.div|aten::_safe_softmax|4| +|25|Tensor<[1,8,256,9]>,
Tensor<[1,8,256,9]>,
|ttnn.div|aten::_safe_softmax|4| +|26|Tensor<[1,8,64,64]>,
Tensor<[1,8,64,64]>,
|ttnn.div|aten::_safe_softmax|4| +|27|Tensor<[1,8,64,9]>,
Tensor<[1,8,64,9]>,
|ttnn.div|aten::_safe_softmax|4| +|28|Tensor<[160]>,
Tensor<[160]>,
|ttnn.div|aten::div.Tensor|4| +|29|Tensor<[1,320,64,64]>,
Tensor<[1,320,64,64]>,
|ttnn.div|aten::div.Tensor|4| +|30|Tensor<[1,4096,320]>,
Tensor<[1,4096,320]>,
|ttnn.div|aten::div.Tensor|4| +|31|Tensor<[1,640,32,32]>,
Tensor<[1,640,32,32]>,
|ttnn.div|aten::div.Tensor|4| +|32|Tensor<[1,1024,640]>,
Tensor<[1,1024,640]>,
|ttnn.div|aten::div.Tensor|4| +|33|Tensor<[1,1280,16,16]>,
Tensor<[1,1280,16,16]>,
|ttnn.div|aten::div.Tensor|4| +|34|Tensor<[1,256,1280]>,
Tensor<[1,256,1280]>,
|ttnn.div|aten::div.Tensor|4| +|35|Tensor<[1,1280,8,8]>,
Tensor<[1,1280,8,8]>,
|ttnn.div|aten::div.Tensor|4| +|36|Tensor<[1,64,1280]>,
Tensor<[1,64,1280]>,
|ttnn.div|aten::div.Tensor|4| +|37|Tensor<[1,4096,1280]>,
Tensor<[1,4096,1280]>,
|ttnn.div|aten::gelu|4| +|38|Tensor<[1,1024,2560]>,
Tensor<[1,1024,2560]>,
|ttnn.div|aten::gelu|4| +|39|Tensor<[1,256,5120]>,
Tensor<[1,256,5120]>,
|ttnn.div|aten::gelu|4| +|40|Tensor<[1,64,5120]>,
Tensor<[1,64,5120]>,
|ttnn.div|aten::gelu|4| +|41|Tensor<[1,12,25,25]>,
Tensor<[1,12,25,25]>,
|ttnn.div|aten::_safe_softmax|4| +|42|Tensor<[1,25,3072]>,
Tensor<[1,25,3072]>,
|ttnn.div|aten::gelu|4| +|43|Tensor<[1,3,1445,1445]>,
Tensor<[1,3,1445,1445]>,
|ttnn.div|aten::_safe_softmax|4| +|44|Tensor<[1,1445,768]>,
Tensor<[1,1445,768]>,
|ttnn.div|aten::gelu|4| +|45|Tensor<[1,512,1,1]>,
Tensor<[1,512,1,1]>,
|ttnn.div|aten::mean.dim|4| +|46|Tensor<[1,12,8,8]>,
Tensor<[1,12,8,8]>,
|ttnn.div|aten::_softmax|4| +|47|Tensor<[1,3072,8]>,
Tensor<[1,3072,8]>,
|ttnn.div|aten::gelu|4| +|48|Tensor<[1,8,256,2048]>,
Tensor<[1,8,256,2048]>,
|ttnn.div|aten::_softmax|4| +|49|Tensor<[1,8,2048,256]>,
Tensor<[1,8,2048,256]>,
|ttnn.div|aten::_softmax|4| +|50|Tensor<[1,2048,768]>,
Tensor<[1,2048,768]>,
|ttnn.div|aten::gelu|4| +|51|Tensor<[1,2048,1,1]>,
Tensor<[1,2048,1,1]>,
|ttnn.div|aten::mean.dim|4| +|52|Tensor<[1024]>,
Tensor<[1024]>,
|ttnn.div|aten::reciprocal|5| +|53|Tensor<[2048]>,
Tensor<[2048]>,
|ttnn.div|aten::reciprocal|5| +|54|Tensor<[1,12,201,201]>,
Tensor<[1,12,201,201]>,
|ttnn.div|aten::_softmax|4| +|55|Tensor<[1,201,3072]>,
Tensor<[1,201,3072]>,
|ttnn.div|aten::gelu|4| +|56|Tensor<[1,1536]>,
Tensor<[1,1536]>,
|ttnn.div|aten::gelu|4| +|57|Tensor<[16,19,19]>,
Tensor<[16,19,19]>,
|ttnn.div|aten::_softmax|4| +|58|Tensor<[19]>,
Tensor<[19]>,
|ttnn.div|aten::floor_divide|4| +|59|Tensor<[1,19,4096]>,
Tensor<[1,19,4096]>,
|ttnn.div|aten::gelu|4| +|60|Tensor<[1,1024,1,1]>,
Tensor<[1,1024,1,1]>,
|ttnn.div|aten::mean.dim|4| +|61|Tensor<[14]>,
Tensor<[14]>,
|ttnn.div|aten::reciprocal|5| +|62|Tensor<[24]>,
Tensor<[24]>,
|ttnn.div|aten::reciprocal|5| +|63|Tensor<[40]>,
Tensor<[40]>,
|ttnn.div|aten::reciprocal|5| +|64|Tensor<[68]>,
Tensor<[68]>,
|ttnn.div|aten::reciprocal|5| +|65|Tensor<[16]>,
Tensor<[16]>,
|ttnn.div|aten::reciprocal|5| +|66|Tensor<[28]>,
Tensor<[28]>,
|ttnn.div|aten::reciprocal|5| +|67|Tensor<[46]>,
Tensor<[46]>,
|ttnn.div|aten::reciprocal|5| +|68|Tensor<[78]>,
Tensor<[78]>,
|ttnn.div|aten::reciprocal|5| +|69|Tensor<[134]>,
Tensor<[134]>,
|ttnn.div|aten::reciprocal|5| +|70|Tensor<[20]>,
Tensor<[20]>,
|ttnn.div|aten::reciprocal|5| +|71|Tensor<[34]>,
Tensor<[34]>,
|ttnn.div|aten::reciprocal|5| +|72|Tensor<[58]>,
Tensor<[58]>,
|ttnn.div|aten::reciprocal|5| +|73|Tensor<[98]>,
Tensor<[98]>,
|ttnn.div|aten::reciprocal|5| +|74|Tensor<[168]>,
Tensor<[168]>,
|ttnn.div|aten::reciprocal|5| +|75|Tensor<[320]>,
Tensor<[320]>,
|ttnn.div|aten::reciprocal|5| +|76|Tensor<[116]>,
Tensor<[116]>,
|ttnn.div|aten::reciprocal|5| +|77|Tensor<[196]>,
Tensor<[196]>,
|ttnn.div|aten::reciprocal|5| +|78|Tensor<[334]>,
Tensor<[334]>,
|ttnn.div|aten::reciprocal|5| +|79|Tensor<[640]>,
Tensor<[640]>,
|ttnn.div|aten::reciprocal|5| +|80|Tensor<[272]>,
Tensor<[272]>,
|ttnn.div|aten::reciprocal|5| +|81|Tensor<[462]>,
Tensor<[462]>,
|ttnn.div|aten::reciprocal|5| +|82|Tensor<[1,16,32,32]>,
Tensor<[1,16,32,32]>,
|ttnn.div|aten::_softmax|4| +|83|Tensor<[1,12,16,16]>,
Tensor<[1,12,16,16]>,
|ttnn.div|aten::_safe_softmax|4| +|84|Tensor<[1,16,3072]>,
Tensor<[1,16,3072]>,
|ttnn.div|aten::gelu|4| +|85|Tensor<[1,1,19200,300]>,
Tensor<[1,1,19200,300]>,
|ttnn.div|aten::_softmax|4| +|86|Tensor<[1,2,4800,300]>,
Tensor<[1,2,4800,300]>,
|ttnn.div|aten::_softmax|4| +|87|Tensor<[1,5,1200,300]>,
Tensor<[1,5,1200,300]>,
|ttnn.div|aten::_softmax|4| +|88|Tensor<[1,8,300,300]>,
Tensor<[1,8,300,300]>,
|ttnn.div|aten::_softmax|4| +|89|Tensor<[1,19200,256]>,
Tensor<[1,19200,256]>,
|ttnn.div|aten::gelu|4| +|90|Tensor<[1,4800,512]>,
Tensor<[1,4800,512]>,
|ttnn.div|aten::gelu|4| +|91|Tensor<[1,1200,1280]>,
Tensor<[1,1200,1280]>,
|ttnn.div|aten::gelu|4| +|92|Tensor<[1,300,2048]>,
Tensor<[1,300,2048]>,
|ttnn.div|aten::gelu|4| +|93|Tensor<[1,12,197,197]>,
Tensor<[1,12,197,197]>,
|ttnn.div|aten::_safe_softmax|4| +|94|Tensor<[1,197,3072]>,
Tensor<[1,197,3072]>,
|ttnn.div|aten::gelu|4| +|95|Tensor<[1,1,16384,256]>,
Tensor<[1,1,16384,256]>,
|ttnn.div|aten::_softmax|4| +|96|Tensor<[1,2,4096,256]>,
Tensor<[1,2,4096,256]>,
|ttnn.div|aten::_softmax|4| +|97|Tensor<[1,5,1024,256]>,
Tensor<[1,5,1024,256]>,
|ttnn.div|aten::_softmax|4| +|98|Tensor<[1,16384,128]>,
Tensor<[1,16384,128]>,
|ttnn.div|aten::gelu|4| +|99|Tensor<[1,4096,256]>,
Tensor<[1,4096,256]>,
|ttnn.div|aten::gelu|4| +|100|Tensor<[1,256,1024]>,
Tensor<[1,256,1024]>,
|ttnn.div|aten::gelu|4| +|101|Tensor<[1,71,7,7]>,
Tensor<[1,71,7,7]>,
|ttnn.div|aten::_safe_softmax|4| +|102|Tensor<[1,7,18176]>,
Tensor<[1,7,18176]>,
|ttnn.div|aten::gelu|4| +|103|Tensor<[1,1280,1,1]>,
Tensor<[1,1280,1,1]>,
|ttnn.div|aten::mean.dim|4| +|104|Tensor<[96]>,
Tensor<[96]>,
|ttnn.div|aten::reciprocal|5| +|105|Tensor<[144]>,
Tensor<[144]>,
|ttnn.div|aten::reciprocal|5| +|106|Tensor<[192]>,
Tensor<[192]>,
|ttnn.div|aten::reciprocal|5| +|107|Tensor<[384]>,
Tensor<[384]>,
|ttnn.div|aten::reciprocal|5| +|108|Tensor<[576]>,
Tensor<[576]>,
|ttnn.div|aten::reciprocal|5| +|109|Tensor<[960]>,
Tensor<[960]>,
|ttnn.div|aten::reciprocal|5| +|110|Tensor<[1280]>,
Tensor<[1280]>,
|ttnn.div|aten::reciprocal|5| +|111|Tensor<[1,12,12,12]>,
Tensor<[1,12,12,12]>,
|ttnn.div|aten::_safe_softmax|4| +|112|Tensor<[1,12,9,9]>,
Tensor<[1,12,9,9]>,
|ttnn.div|aten::_safe_softmax|4| +|113|Tensor<[1,16,9,9]>,
Tensor<[1,16,9,9]>,
|ttnn.div|aten::_safe_softmax|4| +|114|Tensor<[1,64,9,9]>,
Tensor<[1,64,9,9]>,
|ttnn.div|aten::_safe_softmax|4| +|115|Tensor<[1,12,14,14]>,
Tensor<[1,12,14,14]>,
|ttnn.div|aten::_safe_softmax|4| +|116|Tensor<[1,12,50,50]>,
Tensor<[1,12,50,50]>,
|ttnn.div|aten::_safe_softmax|4| +|117|Tensor<[2,8,7,7]>,
Tensor<[2,8,7,7]>,
|ttnn.div|aten::_safe_softmax|4| +|118|Tensor<[2,512]>,
Tensor<[2,512]>,
|ttnn.div|aten::div.Tensor|4| +|119|Tensor<[1,16,197,197]>,
Tensor<[1,16,197,197]>,
|ttnn.div|aten::_softmax|4| +|120|Tensor<[197]>,
Tensor<[197]>,
|ttnn.div|aten::floor_divide|4| +|121|Tensor<[1,197,4096]>,
Tensor<[1,197,4096]>,
|ttnn.div|aten::gelu|4| +|122|Tensor<[1,1024]>,
Tensor<[1,1024]>,
|ttnn.div|aten::mean.dim|4| +|123|Tensor<[1,768]>,
Tensor<[1,768]>,
|ttnn.div|aten::mean.dim|4| diff --git a/docs/ops/stablehlo/stablehlo.dot_general.md b/docs/ops/stablehlo/stablehlo.dot_general.md new file mode 100644 index 00000000..6da6c1df --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.dot_general.md @@ -0,0 +1,274 @@ + +### stablehlo.dot_general::ttnn.matmul + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,64,1]>,
Tensor<[1,1,32]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|1|Tensor<[32,32,128]>,
Tensor<[32,128,32]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|2|Tensor<[32,32,32]>,
Tensor<[32,32,128]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|3|Tensor<[32,4096]>,
Tensor<[4096,4096]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|4|Tensor<[32,4096]>,
Tensor<[4096,11008]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|5|Tensor<[32,11008]>,
Tensor<[11008,4096]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|6|Tensor<[32,4096]>,
Tensor<[4096,32000]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|7|Tensor<[12,7,64]>,
Tensor<[12,64,7]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|8|Tensor<[12,7,7]>,
Tensor<[12,7,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|9|Tensor<[7,768]>,
Tensor<[768,2304]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|10|Tensor<[7,768]>,
Tensor<[768,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|11|Tensor<[7,768]>,
Tensor<[768,3072]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|12|Tensor<[7,3072]>,
Tensor<[3072,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|13|Tensor<[7,768]>,
Tensor<[768,2]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|14|Tensor<[256,768]>,
Tensor<[768,512]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|15|Tensor<[256,512]>,
Tensor<[512,256]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|16|Tensor<[256,256]>,
Tensor<[256,512]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|17|Tensor<[1,512]>,
Tensor<[512,1000]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|18|Tensor<[8,920,32]>,
Tensor<[8,32,920]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::baddbmm|4| +|19|Tensor<[8,100,32]>,
Tensor<[8,32,920]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::baddbmm|4| +|20|Tensor<[920,1,256]>,
Tensor<[920,256,256]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|21|Tensor<[8,920,920]>,
Tensor<[8,920,32]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|22|Tensor<[8,100,32]>,
Tensor<[8,32,100]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|23|Tensor<[8,100,100]>,
Tensor<[8,100,32]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|24|Tensor<[8,100,920]>,
Tensor<[8,920,32]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|25|Tensor<[6,100,256]>,
Tensor<[6,256,92]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|26|Tensor<[6,100,256]>,
Tensor<[6,256,256]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|27|Tensor<[920,256]>,
Tensor<[256,256]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|28|Tensor<[920,256]>,
Tensor<[256,2048]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|29|Tensor<[920,2048]>,
Tensor<[2048,256]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|30|Tensor<[100,256]>,
Tensor<[256,256]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|31|Tensor<[100,256]>,
Tensor<[256,2048]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|32|Tensor<[100,2048]>,
Tensor<[2048,256]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|33|Tensor<[600,256]>,
Tensor<[256,256]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|34|Tensor<[600,256]>,
Tensor<[256,4]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|35|Tensor<[12,10,64]>,
Tensor<[12,64,10]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|36|Tensor<[12,10,10]>,
Tensor<[12,10,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|37|Tensor<[10,768]>,
Tensor<[768,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|38|Tensor<[10,768]>,
Tensor<[768,3072]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|39|Tensor<[10,3072]>,
Tensor<[3072,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|40|Tensor<[10,768]>,
Tensor<[768,250002]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|41|Tensor<[8,4096,40]>,
Tensor<[8,40,4096]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|42|Tensor<[8,4096,4096]>,
Tensor<[8,4096,40]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|43|Tensor<[8,4096,40]>,
Tensor<[8,40,9]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|44|Tensor<[8,4096,9]>,
Tensor<[8,9,40]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|45|Tensor<[8,1024,80]>,
Tensor<[8,80,1024]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|46|Tensor<[8,1024,1024]>,
Tensor<[8,1024,80]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|47|Tensor<[8,1024,80]>,
Tensor<[8,80,9]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|48|Tensor<[8,1024,9]>,
Tensor<[8,9,80]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|49|Tensor<[8,256,160]>,
Tensor<[8,160,256]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|50|Tensor<[8,256,256]>,
Tensor<[8,256,160]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|51|Tensor<[8,256,160]>,
Tensor<[8,160,9]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|52|Tensor<[8,256,9]>,
Tensor<[8,9,160]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|53|Tensor<[8,64,160]>,
Tensor<[8,160,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|54|Tensor<[8,64,64]>,
Tensor<[8,64,160]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|55|Tensor<[8,64,160]>,
Tensor<[8,160,9]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|56|Tensor<[8,64,9]>,
Tensor<[8,9,160]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|57|Tensor<[1,320]>,
Tensor<[320,1280]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|58|Tensor<[1,1280]>,
Tensor<[1280,1280]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|59|Tensor<[1,1280]>,
Tensor<[1280,320]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|60|Tensor<[4096,320]>,
Tensor<[320,320]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|61|Tensor<[9,768]>,
Tensor<[768,320]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|62|Tensor<[4096,320]>,
Tensor<[320,2560]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|63|Tensor<[4096,1280]>,
Tensor<[1280,320]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|64|Tensor<[1,1280]>,
Tensor<[1280,640]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|65|Tensor<[1024,640]>,
Tensor<[640,640]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|66|Tensor<[9,768]>,
Tensor<[768,640]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|67|Tensor<[1024,640]>,
Tensor<[640,5120]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|68|Tensor<[1024,2560]>,
Tensor<[2560,640]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|69|Tensor<[256,1280]>,
Tensor<[1280,1280]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|70|Tensor<[9,768]>,
Tensor<[768,1280]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|71|Tensor<[256,1280]>,
Tensor<[1280,10240]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|72|Tensor<[256,5120]>,
Tensor<[5120,1280]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|73|Tensor<[64,1280]>,
Tensor<[1280,1280]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|74|Tensor<[64,1280]>,
Tensor<[1280,10240]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|75|Tensor<[64,5120]>,
Tensor<[5120,1280]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|76|Tensor<[12,25,64]>,
Tensor<[12,64,25]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|77|Tensor<[12,25,25]>,
Tensor<[12,25,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|78|Tensor<[25,768]>,
Tensor<[768,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|79|Tensor<[25,768]>,
Tensor<[768,3072]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|80|Tensor<[25,3072]>,
Tensor<[3072,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|81|Tensor<[25,768]>,
Tensor<[768,2]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|82|Tensor<[1,768]>,
Tensor<[768,1]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|83|Tensor<[3,1445,64]>,
Tensor<[3,64,1445]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|84|Tensor<[3,1445,1445]>,
Tensor<[3,1445,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|85|Tensor<[1445,192]>,
Tensor<[192,192]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|86|Tensor<[1445,192]>,
Tensor<[192,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|87|Tensor<[1445,768]>,
Tensor<[768,192]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|88|Tensor<[100,192]>,
Tensor<[192,192]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|89|Tensor<[100,192]>,
Tensor<[192,92]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|90|Tensor<[100,192]>,
Tensor<[192,4]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|91|Tensor<[12,8,64]>,
Tensor<[12,64,8]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|92|Tensor<[12,8,8]>,
Tensor<[12,8,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|93|Tensor<[1,768]>,
Tensor<[768,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|94|Tensor<[1,768]>,
Tensor<[768,3]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|95|Tensor<[8,256,32]>,
Tensor<[8,32,2048]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|96|Tensor<[8,256,2048]>,
Tensor<[8,2048,160]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|97|Tensor<[8,256,32]>,
Tensor<[8,32,256]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|98|Tensor<[8,2048,32]>,
Tensor<[8,32,256]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|99|Tensor<[8,2048,256]>,
Tensor<[8,256,96]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|100|Tensor<[256,1280]>,
Tensor<[1280,256]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|101|Tensor<[2048,768]>,
Tensor<[768,256]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|102|Tensor<[2048,768]>,
Tensor<[768,1280]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|103|Tensor<[256,1280]>,
Tensor<[1280,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|104|Tensor<[2048,768]>,
Tensor<[768,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|105|Tensor<[2048,768]>,
Tensor<[768,262]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|106|Tensor<[1,2048]>,
Tensor<[2048,1000]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|107|Tensor<[12,201,64]>,
Tensor<[12,64,201]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|108|Tensor<[12,201,201]>,
Tensor<[12,201,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|109|Tensor<[201,768]>,
Tensor<[768,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|110|Tensor<[201,768]>,
Tensor<[768,3072]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|111|Tensor<[201,3072]>,
Tensor<[3072,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|112|Tensor<[1,768]>,
Tensor<[768,1536]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|113|Tensor<[1,1536]>,
Tensor<[1536,3129]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|114|Tensor<[1,9216]>,
Tensor<[9216,128]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|115|Tensor<[1,128]>,
Tensor<[128,10]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|116|Tensor<[16,19,64]>,
Tensor<[16,64,19]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|117|Tensor<[16,19,19]>,
Tensor<[16,19,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|118|Tensor<[19,1024]>,
Tensor<[1024,1024]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|119|Tensor<[19,1024]>,
Tensor<[1024,4096]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|120|Tensor<[19,4096]>,
Tensor<[4096,1024]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|121|Tensor<[19,1024]>,
Tensor<[1024,256008]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|122|Tensor<[1,1024]>,
Tensor<[1024,1000]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|123|Tensor<[16,32,96]>,
Tensor<[16,96,32]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::baddbmm|4| +|124|Tensor<[16,32,32]>,
Tensor<[16,32,96]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|125|Tensor<[32,1536]>,
Tensor<[1536,4608]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|126|Tensor<[32,1536]>,
Tensor<[1536,1536]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|127|Tensor<[32,1536]>,
Tensor<[1536,6144]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|128|Tensor<[32,6144]>,
Tensor<[6144,1536]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|129|Tensor<[32,1536]>,
Tensor<[1536,250880]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|130|Tensor<[12,16,64]>,
Tensor<[12,64,16]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|131|Tensor<[12,16,16]>,
Tensor<[12,16,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|132|Tensor<[16,768]>,
Tensor<[768,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|133|Tensor<[16,768]>,
Tensor<[768,3072]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|134|Tensor<[16,3072]>,
Tensor<[3072,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|135|Tensor<[1,19200,64]>,
Tensor<[1,64,300]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|136|Tensor<[1,19200,300]>,
Tensor<[1,300,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|137|Tensor<[1,19200,256]>,
Tensor<[1,256,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|138|Tensor<[2,4800,64]>,
Tensor<[2,64,300]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|139|Tensor<[2,4800,300]>,
Tensor<[2,300,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|140|Tensor<[1,4800,512]>,
Tensor<[1,512,128]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|141|Tensor<[5,1200,64]>,
Tensor<[5,64,300]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|142|Tensor<[5,1200,300]>,
Tensor<[5,300,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|143|Tensor<[1,1200,1280]>,
Tensor<[1,1280,320]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|144|Tensor<[8,300,64]>,
Tensor<[8,64,300]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|145|Tensor<[8,300,300]>,
Tensor<[8,300,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|146|Tensor<[1,300,2048]>,
Tensor<[1,2048,512]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|147|Tensor<[19200,64]>,
Tensor<[64,64]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|148|Tensor<[300,64]>,
Tensor<[64,64]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|149|Tensor<[19200,64]>,
Tensor<[64,256]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|150|Tensor<[4800,128]>,
Tensor<[128,128]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|151|Tensor<[300,128]>,
Tensor<[128,128]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|152|Tensor<[4800,128]>,
Tensor<[128,512]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|153|Tensor<[1200,320]>,
Tensor<[320,320]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|154|Tensor<[300,320]>,
Tensor<[320,320]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|155|Tensor<[1200,320]>,
Tensor<[320,1280]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|156|Tensor<[300,512]>,
Tensor<[512,512]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|157|Tensor<[300,512]>,
Tensor<[512,2048]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|158|Tensor<[12,197,64]>,
Tensor<[12,64,197]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|159|Tensor<[12,197,197]>,
Tensor<[12,197,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|160|Tensor<[197,768]>,
Tensor<[768,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|161|Tensor<[197,768]>,
Tensor<[768,3072]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|162|Tensor<[197,3072]>,
Tensor<[3072,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|163|Tensor<[1,768]>,
Tensor<[768,1000]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|164|Tensor<[1,16384,32]>,
Tensor<[1,32,256]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|165|Tensor<[1,16384,256]>,
Tensor<[1,256,32]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|166|Tensor<[1,16384,128]>,
Tensor<[1,128,32]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|167|Tensor<[2,4096,32]>,
Tensor<[2,32,256]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|168|Tensor<[2,4096,256]>,
Tensor<[2,256,32]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|169|Tensor<[1,4096,256]>,
Tensor<[1,256,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|170|Tensor<[5,1024,32]>,
Tensor<[5,32,256]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|171|Tensor<[5,1024,256]>,
Tensor<[5,256,32]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|172|Tensor<[1,1024,640]>,
Tensor<[1,640,160]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|173|Tensor<[8,256,256]>,
Tensor<[8,256,32]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|174|Tensor<[1,256,1024]>,
Tensor<[1,1024,256]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|175|Tensor<[1,4096,64]>,
Tensor<[1,64,256]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|176|Tensor<[1,1024,160]>,
Tensor<[1,160,256]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|177|Tensor<[1,256,256]>,
Tensor<[1,256,256]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|178|Tensor<[16384,32]>,
Tensor<[32,32]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|179|Tensor<[256,32]>,
Tensor<[32,32]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|180|Tensor<[16384,32]>,
Tensor<[32,128]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|181|Tensor<[4096,64]>,
Tensor<[64,64]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|182|Tensor<[256,64]>,
Tensor<[64,64]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|183|Tensor<[4096,64]>,
Tensor<[64,256]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|184|Tensor<[1024,160]>,
Tensor<[160,160]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|185|Tensor<[256,160]>,
Tensor<[160,160]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|186|Tensor<[1024,160]>,
Tensor<[160,640]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|187|Tensor<[256,256]>,
Tensor<[256,256]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|188|Tensor<[256,256]>,
Tensor<[256,1024]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|189|Tensor<[1,32,1]>,
Tensor<[1,1,7]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|190|Tensor<[71,7,64]>,
Tensor<[71,64,7]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|191|Tensor<[71,7,7]>,
Tensor<[71,7,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|192|Tensor<[7,4544]>,
Tensor<[4544,4672]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|193|Tensor<[7,4544]>,
Tensor<[4544,4544]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|194|Tensor<[7,4544]>,
Tensor<[4544,18176]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|195|Tensor<[7,18176]>,
Tensor<[18176,4544]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|196|Tensor<[7,4544]>,
Tensor<[4544,65024]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|197|Tensor<[1,1280]>,
Tensor<[1280,1000]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|198|Tensor<[12,12,64]>,
Tensor<[12,64,12]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|199|Tensor<[12,12,12]>,
Tensor<[12,12,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|200|Tensor<[12,128]>,
Tensor<[128,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|201|Tensor<[12,768]>,
Tensor<[768,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|202|Tensor<[12,768]>,
Tensor<[768,3072]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|203|Tensor<[12,3072]>,
Tensor<[3072,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|204|Tensor<[12,768]>,
Tensor<[768,2]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|205|Tensor<[12,9,64]>,
Tensor<[12,64,9]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|206|Tensor<[12,9,9]>,
Tensor<[12,9,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|207|Tensor<[9,128]>,
Tensor<[128,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|208|Tensor<[9,768]>,
Tensor<[768,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|209|Tensor<[9,768]>,
Tensor<[768,3072]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|210|Tensor<[9,3072]>,
Tensor<[3072,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|211|Tensor<[9,768]>,
Tensor<[768,128]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|212|Tensor<[9,128]>,
Tensor<[128,30000]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|213|Tensor<[16,9,128]>,
Tensor<[16,128,9]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|214|Tensor<[16,9,9]>,
Tensor<[16,9,128]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|215|Tensor<[9,128]>,
Tensor<[128,2048]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|216|Tensor<[9,2048]>,
Tensor<[2048,2048]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|217|Tensor<[9,2048]>,
Tensor<[2048,8192]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|218|Tensor<[9,8192]>,
Tensor<[8192,2048]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|219|Tensor<[9,2048]>,
Tensor<[2048,128]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|220|Tensor<[16,9,64]>,
Tensor<[16,64,9]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|221|Tensor<[16,9,9]>,
Tensor<[16,9,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|222|Tensor<[9,128]>,
Tensor<[128,1024]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|223|Tensor<[9,1024]>,
Tensor<[1024,1024]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|224|Tensor<[9,1024]>,
Tensor<[1024,4096]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|225|Tensor<[9,4096]>,
Tensor<[4096,1024]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|226|Tensor<[9,1024]>,
Tensor<[1024,128]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|227|Tensor<[64,9,64]>,
Tensor<[64,64,9]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|228|Tensor<[64,9,9]>,
Tensor<[64,9,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|229|Tensor<[9,128]>,
Tensor<[128,4096]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|230|Tensor<[9,4096]>,
Tensor<[4096,4096]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|231|Tensor<[9,4096]>,
Tensor<[4096,16384]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|232|Tensor<[9,16384]>,
Tensor<[16384,4096]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|233|Tensor<[9,4096]>,
Tensor<[4096,128]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|234|Tensor<[1,768]>,
Tensor<[768,2]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|235|Tensor<[12,14,64]>,
Tensor<[12,64,14]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|236|Tensor<[12,14,14]>,
Tensor<[12,14,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|237|Tensor<[14,128]>,
Tensor<[128,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|238|Tensor<[14,768]>,
Tensor<[768,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|239|Tensor<[14,768]>,
Tensor<[768,3072]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|240|Tensor<[14,3072]>,
Tensor<[3072,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|241|Tensor<[14,768]>,
Tensor<[768,2]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|242|Tensor<[12,50,64]>,
Tensor<[12,64,50]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|243|Tensor<[12,50,50]>,
Tensor<[12,50,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|244|Tensor<[16,7,64]>,
Tensor<[16,64,7]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|245|Tensor<[16,7,7]>,
Tensor<[16,7,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|246|Tensor<[50,768]>,
Tensor<[768,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|247|Tensor<[50,768]>,
Tensor<[768,3072]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|248|Tensor<[50,3072]>,
Tensor<[3072,768]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|249|Tensor<[14,512]>,
Tensor<[512,512]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|250|Tensor<[14,512]>,
Tensor<[512,2048]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|251|Tensor<[14,2048]>,
Tensor<[2048,512]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|252|Tensor<[1,768]>,
Tensor<[768,512]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|253|Tensor<[2,512]>,
Tensor<[512,512]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|254|Tensor<[2,512]>,
Tensor<[512,1]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|255|Tensor<[16,197,64]>,
Tensor<[16,64,197]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|256|Tensor<[16,197,197]>,
Tensor<[16,197,64]>,
batching_dims: [0] x [0]
contracting_dims: [2] x [1]
|ttnn.matmul|aten::bmm|4| +|257|Tensor<[197,1024]>,
Tensor<[1024,1024]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|258|Tensor<[197,1024]>,
Tensor<[1024,4096]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|259|Tensor<[197,4096]>,
Tensor<[4096,1024]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|260|Tensor<[1,784]>,
Tensor<[784,128]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|261|Tensor<[1,128]>,
Tensor<[128,64]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|262|Tensor<[1,64]>,
Tensor<[64,12]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|263|Tensor<[1,12]>,
Tensor<[12,3]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|264|Tensor<[1,3]>,
Tensor<[3,12]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|265|Tensor<[1,12]>,
Tensor<[12,64]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|266|Tensor<[1,64]>,
Tensor<[64,128]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| +|267|Tensor<[1,128]>,
Tensor<[128,784]>,
contracting_dims: [1] x [0]
|ttnn.matmul|aten::mm|5| diff --git a/docs/ops/stablehlo/stablehlo.dynamic_iota.md b/docs/ops/stablehlo/stablehlo.dynamic_iota.md new file mode 100644 index 00000000..b5387b42 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.dynamic_iota.md @@ -0,0 +1,7 @@ + +### stablehlo.dynamic_iota::ttnn.arange + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1]>,
dim: 0
|ttnn.arange|aten::arange|4| diff --git a/docs/ops/stablehlo/stablehlo.exponential.md b/docs/ops/stablehlo/stablehlo.exponential.md new file mode 100644 index 00000000..495785fe --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.exponential.md @@ -0,0 +1,51 @@ + +### stablehlo.exponential::ttnn.exp + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,32,32,32]>,
|ttnn.exp|aten::_safe_softmax|4| +|1|Tensor<[1,12,7,7]>,
|ttnn.exp|aten::_safe_softmax|4| +|2|Tensor<[1,128,28,28]>,
|ttnn.exp|aten::elu|4| +|3|Tensor<[8,920,920]>,
|ttnn.exp|aten::_softmax|4| +|4|Tensor<[8,100,100]>,
|ttnn.exp|aten::_softmax|4| +|5|Tensor<[8,100,920]>,
|ttnn.exp|aten::_softmax|4| +|6|Tensor<[1,12,10,10]>,
|ttnn.exp|aten::_safe_softmax|4| +|7|Tensor<[1,8,4096,4096]>,
|ttnn.exp|aten::_safe_softmax|4| +|8|Tensor<[1,8,4096,9]>,
|ttnn.exp|aten::_safe_softmax|4| +|9|Tensor<[1,8,1024,1024]>,
|ttnn.exp|aten::_safe_softmax|4| +|10|Tensor<[1,8,1024,9]>,
|ttnn.exp|aten::_safe_softmax|4| +|11|Tensor<[1,8,256,256]>,
|ttnn.exp|aten::_safe_softmax|4| +|12|Tensor<[1,8,256,9]>,
|ttnn.exp|aten::_safe_softmax|4| +|13|Tensor<[1,8,64,64]>,
|ttnn.exp|aten::_safe_softmax|4| +|14|Tensor<[1,8,64,9]>,
|ttnn.exp|aten::_safe_softmax|4| +|15|Tensor<[160]>,
|ttnn.exp|aten::exp|5| +|16|Tensor<[1,12,25,25]>,
|ttnn.exp|aten::_safe_softmax|4| +|17|Tensor<[1,3,1445,1445]>,
|ttnn.exp|aten::_safe_softmax|4| +|18|Tensor<[1,12,8,8]>,
|ttnn.exp|aten::_softmax|4| +|19|Tensor<[1,8,256,2048]>,
|ttnn.exp|aten::_softmax|4| +|20|Tensor<[1,8,2048,256]>,
|ttnn.exp|aten::_softmax|4| +|21|Tensor<[1,12,201,201]>,
|ttnn.exp|aten::_softmax|4| +|22|Tensor<[1,10]>,
|ttnn.exp|aten::exp|5| +|23|Tensor<[16,19,19]>,
|ttnn.exp|aten::_softmax|4| +|24|Tensor<[19,256008]>,
|ttnn.exp|aten::exp|5| +|25|Tensor<[1,16,32,32]>,
|ttnn.exp|aten::_softmax|4| +|26|Tensor<[1,12,16,16]>,
|ttnn.exp|aten::_safe_softmax|4| +|27|Tensor<[1,1,19200,300]>,
|ttnn.exp|aten::_softmax|4| +|28|Tensor<[1,2,4800,300]>,
|ttnn.exp|aten::_softmax|4| +|29|Tensor<[1,5,1200,300]>,
|ttnn.exp|aten::_softmax|4| +|30|Tensor<[1,8,300,300]>,
|ttnn.exp|aten::_softmax|4| +|31|Tensor<[1,12,197,197]>,
|ttnn.exp|aten::_safe_softmax|4| +|32|Tensor<[1,1,16384,256]>,
|ttnn.exp|aten::_softmax|4| +|33|Tensor<[1,2,4096,256]>,
|ttnn.exp|aten::_softmax|4| +|34|Tensor<[1,5,1024,256]>,
|ttnn.exp|aten::_softmax|4| +|35|Tensor<[1,71,7,7]>,
|ttnn.exp|aten::_safe_softmax|4| +|36|Tensor<[1,12,12,12]>,
|ttnn.exp|aten::_safe_softmax|4| +|37|Tensor<[1,12,9,9]>,
|ttnn.exp|aten::_safe_softmax|4| +|38|Tensor<[1,16,9,9]>,
|ttnn.exp|aten::_safe_softmax|4| +|39|Tensor<[1,64,9,9]>,
|ttnn.exp|aten::_safe_softmax|4| +|40|Tensor<[1,12,14,14]>,
|ttnn.exp|aten::_safe_softmax|4| +|41|Tensor<[1,12,50,50]>,
|ttnn.exp|aten::_safe_softmax|4| +|42|Tensor<[2,8,7,7]>,
|ttnn.exp|aten::_safe_softmax|4| +|43|Scalar,
|ttnn.exp|aten::exp|5| +|44|Tensor<[1,16,197,197]>,
|ttnn.exp|aten::_softmax|4| diff --git a/docs/ops/stablehlo/stablehlo.floor.md b/docs/ops/stablehlo/stablehlo.floor.md new file mode 100644 index 00000000..52dc40dc --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.floor.md @@ -0,0 +1,9 @@ + +### stablehlo.floor::ttnn.floor + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[128]>,
|ttnn.floor|aten::floor_divide|4| +|1|Tensor<[19]>,
|ttnn.floor|aten::floor_divide|4| +|2|Tensor<[197]>,
|ttnn.floor|aten::floor_divide|4| diff --git a/docs/ops/stablehlo/stablehlo.gather.md b/docs/ops/stablehlo/stablehlo.gather.md new file mode 100644 index 00000000..d48dac79 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.gather.md @@ -0,0 +1,65 @@ + +### stablehlo.gather::ttnn.embedding + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[32000,4096]>,
Tensor<[1,32]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|1|Tensor<[50257,768]>,
Tensor<[1,7]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|2|Tensor<[1024,768]>,
Tensor<[1,7]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|3|Tensor<[1,7,2]>,
Tensor<[1,2]>,
offset_dims: [1]
collapsed_slice_dims: [0, 1]
start_index_map: [0, 1]
index_vector_dim: 1
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|4|Tensor<[1,1,720,1280]>,
Tensor<[1,1,23,40,4]>,
collapsed_slice_dims: [0, 1, 2, 3]
start_index_map: [0, 1, 2, 3]
index_vector_dim: 4
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|5|Tensor<[250002,768]>,
Tensor<[1,10]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|6|Tensor<[1,768]>,
Tensor<[1,10]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|7|Tensor<[514,768]>,
Tensor<[1,10]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|8|Tensor<[1,1280,8,8]>,
Tensor<[1,1280,16,16,4]>,
collapsed_slice_dims: [0, 1, 2, 3]
start_index_map: [0, 1, 2, 3]
index_vector_dim: 4
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|9|Tensor<[1,1280,16,16]>,
Tensor<[1,1280,32,32,4]>,
collapsed_slice_dims: [0, 1, 2, 3]
start_index_map: [0, 1, 2, 3]
index_vector_dim: 4
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|10|Tensor<[1,640,32,32]>,
Tensor<[1,640,64,64,4]>,
collapsed_slice_dims: [0, 1, 2, 3]
start_index_map: [0, 1, 2, 3]
index_vector_dim: 4
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|11|Tensor<[30522,768]>,
Tensor<[1,25]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|12|Tensor<[2,768]>,
Tensor<[1,25]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|13|Tensor<[512,768]>,
Tensor<[1,25]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|14|Tensor<[30528,768]>,
Tensor<[1,8]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|15|Tensor<[512,768]>,
Tensor<[1,8]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|16|Tensor<[2,768]>,
Tensor<[1,8]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|17|Tensor<[262,768]>,
Tensor<[1,2048]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|18|Tensor<[2048,768]>,
Tensor<[2048]>,
offset_dims: [1]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 1
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|19|Tensor<[30522,768]>,
Tensor<[1,8]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|20|Tensor<[40,768]>,
Tensor<[1,8]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|21|Tensor<[2,768]>,
Tensor<[1,193]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|22|Tensor<[1,1,384,512]>,
Tensor<[1,1,12,16,4]>,
collapsed_slice_dims: [0, 1, 2, 3]
start_index_map: [0, 1, 2, 3]
index_vector_dim: 4
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|23|Tensor<[256008,1024]>,
Tensor<[1,19]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|24|Tensor<[19,256008]>,
Tensor<[19,1,2]>,
collapsed_slice_dims: [0, 1]
start_index_map: [0, 1]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::gather|4| +|25|Tensor<[2050,1024]>,
Tensor<[19]>,
offset_dims: [1]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 1
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index_select|4| +|26|Tensor<[1,256,16,16]>,
Tensor<[1,256,32,32,4]>,
collapsed_slice_dims: [0, 1, 2, 3]
start_index_map: [0, 1, 2, 3]
index_vector_dim: 4
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|27|Tensor<[1,128,32,32]>,
Tensor<[1,128,64,64,4]>,
collapsed_slice_dims: [0, 1, 2, 3]
start_index_map: [0, 1, 2, 3]
index_vector_dim: 4
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|28|Tensor<[250880,1536]>,
Tensor<[1,32]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|29|Tensor<[30522,768]>,
Tensor<[1,16]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|30|Tensor<[512,768]>,
Tensor<[1,16]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|31|Tensor<[1,64,15,20]>,
Tensor<[1,64,30,40,4]>,
collapsed_slice_dims: [0, 1, 2, 3]
start_index_map: [0, 1, 2, 3]
index_vector_dim: 4
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|32|Tensor<[1,64,30,40]>,
Tensor<[1,64,60,80,4]>,
collapsed_slice_dims: [0, 1, 2, 3]
start_index_map: [0, 1, 2, 3]
index_vector_dim: 4
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|33|Tensor<[1,64,60,80]>,
Tensor<[1,64,120,160,4]>,
collapsed_slice_dims: [0, 1, 2, 3]
start_index_map: [0, 1, 2, 3]
index_vector_dim: 4
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|34|Tensor<[1,64,120,160]>,
Tensor<[1,64,240,320,4]>,
collapsed_slice_dims: [0, 1, 2, 3]
start_index_map: [0, 1, 2, 3]
index_vector_dim: 4
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|35|Tensor<[1,64,240,320]>,
Tensor<[1,64,480,640,4]>,
collapsed_slice_dims: [0, 1, 2, 3]
start_index_map: [0, 1, 2, 3]
index_vector_dim: 4
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|36|Tensor<[1,256,128,128]>,
Tensor<[1,256,128,128,4]>,
collapsed_slice_dims: [0, 1, 2, 3]
start_index_map: [0, 1, 2, 3]
index_vector_dim: 4
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|37|Tensor<[1,256,64,64]>,
Tensor<[1,256,128,128,4]>,
collapsed_slice_dims: [0, 1, 2, 3]
start_index_map: [0, 1, 2, 3]
index_vector_dim: 4
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|38|Tensor<[1,256,32,32]>,
Tensor<[1,256,128,128,4]>,
collapsed_slice_dims: [0, 1, 2, 3]
start_index_map: [0, 1, 2, 3]
index_vector_dim: 4
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|39|Tensor<[1,256,16,16]>,
Tensor<[1,256,128,128,4]>,
collapsed_slice_dims: [0, 1, 2, 3]
start_index_map: [0, 1, 2, 3]
index_vector_dim: 4
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|40|Tensor<[65024,4544]>,
Tensor<[1,7]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|41|Tensor<[1,7,73,64]>,
Tensor<[1,7,1,64,4]>,
collapsed_slice_dims: [0, 1, 2, 3]
start_index_map: [0, 1, 2, 3]
index_vector_dim: 4
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|42|Tensor<[30000,128]>,
Tensor<[1,12]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|43|Tensor<[2,128]>,
Tensor<[1,12]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|44|Tensor<[512,128]>,
Tensor<[1,12]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|45|Tensor<[30000,128]>,
Tensor<[1,9]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|46|Tensor<[2,128]>,
Tensor<[1,9]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|47|Tensor<[512,128]>,
Tensor<[1,9]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|48|Tensor<[30000,128]>,
Tensor<[1,14]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|49|Tensor<[2,128]>,
Tensor<[1,14]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|50|Tensor<[512,128]>,
Tensor<[1,14]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|51|Tensor<[50,768]>,
Tensor<[1,50]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|52|Tensor<[49408,512]>,
Tensor<[2,7]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|53|Tensor<[77,512]>,
Tensor<[1,7]>,
offset_dims: [2]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 2
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::embedding|4| +|54|Tensor<[2,7,512]>,
Tensor<[2,2]>,
offset_dims: [1]
collapsed_slice_dims: [0, 1]
start_index_map: [0, 1]
index_vector_dim: 1
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|55|Tensor<[1,16,27,27]>,
Tensor<[1,16,27,27,4]>,
collapsed_slice_dims: [0, 1, 2, 3]
start_index_map: [0, 1, 2, 3]
index_vector_dim: 4
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|56|Tensor<[732,16]>,
Tensor<[38809,1]>,
offset_dims: [1]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 1
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|57|Tensor<[1,12,27,27]>,
Tensor<[1,12,27,27,4]>,
collapsed_slice_dims: [0, 1, 2, 3]
start_index_map: [0, 1, 2, 3]
index_vector_dim: 4
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| +|58|Tensor<[732,12]>,
Tensor<[38809,1]>,
offset_dims: [1]
collapsed_slice_dims: [0]
start_index_map: [0]
index_vector_dim: 1
indices_are_sorted: false
slice_sizes: array|ttnn.embedding|aten::index.Tensor|4| diff --git a/docs/ops/stablehlo/stablehlo.iota.md b/docs/ops/stablehlo/stablehlo.iota.md new file mode 100644 index 00000000..a72485f6 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.iota.md @@ -0,0 +1,7 @@ + +### stablehlo.iota::ttnn.arange + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[19,1,1]>,
Tensor<[19,1,1]>,
dim: 0
|ttnn.arange|aten::gather|4| diff --git a/docs/ops/stablehlo/stablehlo.log.md b/docs/ops/stablehlo/stablehlo.log.md new file mode 100644 index 00000000..253b82fb --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.log.md @@ -0,0 +1,8 @@ + +### stablehlo.log::ttnn.log + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,1]>,
|ttnn.log|aten::log|4| +|1|Tensor<[19,1]>,
|ttnn.log|aten::log|4| diff --git a/docs/ops/stablehlo/stablehlo.logistic.md b/docs/ops/stablehlo/stablehlo.logistic.md new file mode 100644 index 00000000..2f0a9e78 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.logistic.md @@ -0,0 +1,31 @@ + +### stablehlo.logistic::ttnn.sigmoig + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,32,11008]>,
|ttnn.sigmoig|aten::silu|4| +|1|Tensor<[6,1,100,4]>,
|ttnn.sigmoig|aten::sigmoid|4| +|2|Tensor<[1,1280]>,
|ttnn.sigmoig|aten::silu|4| +|3|Tensor<[1,320,64,64]>,
|ttnn.sigmoig|aten::silu|4| +|4|Tensor<[1,320,32,32]>,
|ttnn.sigmoig|aten::silu|4| +|5|Tensor<[1,640,32,32]>,
|ttnn.sigmoig|aten::silu|4| +|6|Tensor<[1,640,16,16]>,
|ttnn.sigmoig|aten::silu|4| +|7|Tensor<[1,1280,16,16]>,
|ttnn.sigmoig|aten::silu|4| +|8|Tensor<[1,1280,8,8]>,
|ttnn.sigmoig|aten::silu|4| +|9|Tensor<[1,2560,8,8]>,
|ttnn.sigmoig|aten::silu|4| +|10|Tensor<[1,2560,16,16]>,
|ttnn.sigmoig|aten::silu|4| +|11|Tensor<[1,1920,16,16]>,
|ttnn.sigmoig|aten::silu|4| +|12|Tensor<[1,1920,32,32]>,
|ttnn.sigmoig|aten::silu|4| +|13|Tensor<[1,1280,32,32]>,
|ttnn.sigmoig|aten::silu|4| +|14|Tensor<[1,960,32,32]>,
|ttnn.sigmoig|aten::silu|4| +|15|Tensor<[1,960,64,64]>,
|ttnn.sigmoig|aten::silu|4| +|16|Tensor<[1,640,64,64]>,
|ttnn.sigmoig|aten::silu|4| +|17|Tensor<[1,100,4]>,
|ttnn.sigmoig|aten::sigmoid|4| +|18|Tensor<[1,1,256,256]>,
|ttnn.sigmoig|aten::sigmoid|4| +|19|Tensor<[1,2,30,40]>,
|ttnn.sigmoig|aten::sigmoid|4| +|20|Tensor<[1,2,60,80]>,
|ttnn.sigmoig|aten::sigmoid|4| +|21|Tensor<[1,2,120,160]>,
|ttnn.sigmoig|aten::sigmoid|4| +|22|Tensor<[1,1,480,640]>,
|ttnn.sigmoig|aten::sigmoid|4| +|23|Tensor<[1,50,3072]>,
|ttnn.sigmoig|aten::sigmoid|4| +|24|Tensor<[2,7,2048]>,
|ttnn.sigmoig|aten::sigmoid|4| diff --git a/docs/ops/stablehlo/stablehlo.maximum.md b/docs/ops/stablehlo/stablehlo.maximum.md new file mode 100644 index 00000000..96ddbd37 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.maximum.md @@ -0,0 +1,102 @@ + +### stablehlo.maximum::ttnn.maximum + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,128,28,28]>,
Tensor<[1,128,28,28]>,
|ttnn.maximum|aten::elu|4| +|1|Tensor<[1,32,112,112]>,
Tensor<[1,32,112,112]>,
|ttnn.maximum|aten::relu|4| +|2|Tensor<[1,64,112,112]>,
Tensor<[1,64,112,112]>,
|ttnn.maximum|aten::relu|4| +|3|Tensor<[1,64,56,56]>,
Tensor<[1,64,56,56]>,
|ttnn.maximum|aten::relu|4| +|4|Tensor<[1,128,56,56]>,
Tensor<[1,128,56,56]>,
|ttnn.maximum|aten::relu|4| +|5|Tensor<[1,256,28,28]>,
Tensor<[1,256,28,28]>,
|ttnn.maximum|aten::relu|4| +|6|Tensor<[1,512,28,28]>,
Tensor<[1,512,28,28]>,
|ttnn.maximum|aten::relu|4| +|7|Tensor<[1,64,360,640]>,
Tensor<[1,64,360,640]>,
|ttnn.maximum|aten::relu|4| +|8|Tensor<[1,64,180,320]>,
Tensor<[1,64,180,320]>,
|ttnn.maximum|aten::relu|4| +|9|Tensor<[1,256,180,320]>,
Tensor<[1,256,180,320]>,
|ttnn.maximum|aten::relu|4| +|10|Tensor<[1,128,180,320]>,
Tensor<[1,128,180,320]>,
|ttnn.maximum|aten::relu|4| +|11|Tensor<[1,128,90,160]>,
Tensor<[1,128,90,160]>,
|ttnn.maximum|aten::relu|4| +|12|Tensor<[1,512,90,160]>,
Tensor<[1,512,90,160]>,
|ttnn.maximum|aten::relu|4| +|13|Tensor<[1,256,90,160]>,
Tensor<[1,256,90,160]>,
|ttnn.maximum|aten::relu|4| +|14|Tensor<[1,256,45,80]>,
Tensor<[1,256,45,80]>,
|ttnn.maximum|aten::relu|4| +|15|Tensor<[1,1024,45,80]>,
Tensor<[1,1024,45,80]>,
|ttnn.maximum|aten::relu|4| +|16|Tensor<[1,512,45,80]>,
Tensor<[1,512,45,80]>,
|ttnn.maximum|aten::relu|4| +|17|Tensor<[1,512,23,40]>,
Tensor<[1,512,23,40]>,
|ttnn.maximum|aten::relu|4| +|18|Tensor<[1,2048,23,40]>,
Tensor<[1,2048,23,40]>,
|ttnn.maximum|aten::relu|4| +|19|Tensor<[920,1,2048]>,
Tensor<[920,1,2048]>,
|ttnn.maximum|aten::relu|4| +|20|Tensor<[100,1,2048]>,
Tensor<[100,1,2048]>,
|ttnn.maximum|aten::relu|4| +|21|Tensor<[6,1,100,256]>,
Tensor<[6,1,100,256]>,
|ttnn.maximum|aten::relu|4| +|22|Tensor<[1,100,192]>,
Tensor<[1,100,192]>,
|ttnn.maximum|aten::relu|4| +|23|Tensor<[1,256,14,14]>,
Tensor<[1,256,14,14]>,
|ttnn.maximum|aten::relu|4| +|24|Tensor<[1,512,7,7]>,
Tensor<[1,512,7,7]>,
|ttnn.maximum|aten::relu|4| +|25|Tensor<[1,256,56,56]>,
Tensor<[1,256,56,56]>,
|ttnn.maximum|aten::relu|4| +|26|Tensor<[1,1024,14,14]>,
Tensor<[1,1024,14,14]>,
|ttnn.maximum|aten::relu|4| +|27|Tensor<[1,512,14,14]>,
Tensor<[1,512,14,14]>,
|ttnn.maximum|aten::relu|4| +|28|Tensor<[1,2048,7,7]>,
Tensor<[1,2048,7,7]>,
|ttnn.maximum|aten::relu|4| +|29|Tensor<[1,32,26,26]>,
Tensor<[1,32,26,26]>,
|ttnn.maximum|aten::relu|4| +|30|Tensor<[1,64,24,24]>,
Tensor<[1,64,24,24]>,
|ttnn.maximum|aten::relu|4| +|31|Tensor<[1,128]>,
Tensor<[1,128]>,
|ttnn.maximum|aten::relu|4| +|32|Tensor<[1,16,19,19]>,
Tensor<[1,16,19,19]>,
|ttnn.maximum|aten::maximum|4| +|33|Tensor<[1,14,56,56]>,
Tensor<[1,14,56,56]>,
|ttnn.maximum|aten::hardtanh|4| +|34|Tensor<[1,24,56,56]>,
Tensor<[1,24,56,56]>,
|ttnn.maximum|aten::hardtanh|4| +|35|Tensor<[1,40,56,56]>,
Tensor<[1,40,56,56]>,
|ttnn.maximum|aten::hardtanh|4| +|36|Tensor<[1,68,56,56]>,
Tensor<[1,68,56,56]>,
|ttnn.maximum|aten::hardtanh|4| +|37|Tensor<[1,16,28,28]>,
Tensor<[1,16,28,28]>,
|ttnn.maximum|aten::hardtanh|4| +|38|Tensor<[1,28,28,28]>,
Tensor<[1,28,28,28]>,
|ttnn.maximum|aten::hardtanh|4| +|39|Tensor<[1,46,28,28]>,
Tensor<[1,46,28,28]>,
|ttnn.maximum|aten::hardtanh|4| +|40|Tensor<[1,78,28,28]>,
Tensor<[1,78,28,28]>,
|ttnn.maximum|aten::hardtanh|4| +|41|Tensor<[1,134,28,28]>,
Tensor<[1,134,28,28]>,
|ttnn.maximum|aten::hardtanh|4| +|42|Tensor<[1,20,28,28]>,
Tensor<[1,20,28,28]>,
|ttnn.maximum|aten::hardtanh|4| +|43|Tensor<[1,34,28,28]>,
Tensor<[1,34,28,28]>,
|ttnn.maximum|aten::hardtanh|4| +|44|Tensor<[1,58,28,28]>,
Tensor<[1,58,28,28]>,
|ttnn.maximum|aten::hardtanh|4| +|45|Tensor<[1,98,28,28]>,
Tensor<[1,98,28,28]>,
|ttnn.maximum|aten::hardtanh|4| +|46|Tensor<[1,168,28,28]>,
Tensor<[1,168,28,28]>,
|ttnn.maximum|aten::hardtanh|4| +|47|Tensor<[1,320,28,28]>,
Tensor<[1,320,28,28]>,
|ttnn.maximum|aten::hardtanh|4| +|48|Tensor<[1,40,14,14]>,
Tensor<[1,40,14,14]>,
|ttnn.maximum|aten::hardtanh|4| +|49|Tensor<[1,68,14,14]>,
Tensor<[1,68,14,14]>,
|ttnn.maximum|aten::hardtanh|4| +|50|Tensor<[1,116,14,14]>,
Tensor<[1,116,14,14]>,
|ttnn.maximum|aten::hardtanh|4| +|51|Tensor<[1,196,14,14]>,
Tensor<[1,196,14,14]>,
|ttnn.maximum|aten::hardtanh|4| +|52|Tensor<[1,334,14,14]>,
Tensor<[1,334,14,14]>,
|ttnn.maximum|aten::hardtanh|4| +|53|Tensor<[1,640,14,14]>,
Tensor<[1,640,14,14]>,
|ttnn.maximum|aten::hardtanh|4| +|54|Tensor<[1,160,7,7]>,
Tensor<[1,160,7,7]>,
|ttnn.maximum|aten::hardtanh|4| +|55|Tensor<[1,272,7,7]>,
Tensor<[1,272,7,7]>,
|ttnn.maximum|aten::hardtanh|4| +|56|Tensor<[1,462,7,7]>,
Tensor<[1,462,7,7]>,
|ttnn.maximum|aten::hardtanh|4| +|57|Tensor<[1,1024,7,7]>,
Tensor<[1,1024,7,7]>,
|ttnn.maximum|aten::hardtanh|4| +|58|Tensor<[1,32,512,512]>,
Tensor<[1,32,512,512]>,
|ttnn.maximum|aten::leaky_relu|4| +|59|Tensor<[1,64,256,256]>,
Tensor<[1,64,256,256]>,
|ttnn.maximum|aten::leaky_relu|4| +|60|Tensor<[1,32,256,256]>,
Tensor<[1,32,256,256]>,
|ttnn.maximum|aten::leaky_relu|4| +|61|Tensor<[1,128,128,128]>,
Tensor<[1,128,128,128]>,
|ttnn.maximum|aten::leaky_relu|4| +|62|Tensor<[1,64,128,128]>,
Tensor<[1,64,128,128]>,
|ttnn.maximum|aten::leaky_relu|4| +|63|Tensor<[1,256,64,64]>,
Tensor<[1,256,64,64]>,
|ttnn.maximum|aten::leaky_relu|4| +|64|Tensor<[1,128,64,64]>,
Tensor<[1,128,64,64]>,
|ttnn.maximum|aten::leaky_relu|4| +|65|Tensor<[1,512,32,32]>,
Tensor<[1,512,32,32]>,
|ttnn.maximum|aten::leaky_relu|4| +|66|Tensor<[1,256,32,32]>,
Tensor<[1,256,32,32]>,
|ttnn.maximum|aten::leaky_relu|4| +|67|Tensor<[1,1024,16,16]>,
Tensor<[1,1024,16,16]>,
|ttnn.maximum|aten::leaky_relu|4| +|68|Tensor<[1,512,16,16]>,
Tensor<[1,512,16,16]>,
|ttnn.maximum|aten::leaky_relu|4| +|69|Tensor<[1,256,16,16]>,
Tensor<[1,256,16,16]>,
|ttnn.maximum|aten::leaky_relu|4| +|70|Tensor<[1,128,32,32]>,
Tensor<[1,128,32,32]>,
|ttnn.maximum|aten::leaky_relu|4| +|71|Tensor<[1,4,14,14]>,
Tensor<[1,4,14,14]>,
|ttnn.maximum|aten::relu|4| +|72|Tensor<[1,16,14,14]>,
Tensor<[1,16,14,14]>,
|ttnn.maximum|aten::relu|4| +|73|Tensor<[1,64,224,224]>,
Tensor<[1,64,224,224]>,
|ttnn.maximum|aten::relu|4| +|74|Tensor<[1,128,112,112]>,
Tensor<[1,128,112,112]>,
|ttnn.maximum|aten::relu|4| +|75|Tensor<[1,64,30,40]>,
Tensor<[1,64,30,40]>,
|ttnn.maximum|aten::relu|4| +|76|Tensor<[1,32,30,40]>,
Tensor<[1,32,30,40]>,
|ttnn.maximum|aten::relu|4| +|77|Tensor<[1,64,60,80]>,
Tensor<[1,64,60,80]>,
|ttnn.maximum|aten::relu|4| +|78|Tensor<[1,32,60,80]>,
Tensor<[1,32,60,80]>,
|ttnn.maximum|aten::relu|4| +|79|Tensor<[1,64,120,160]>,
Tensor<[1,64,120,160]>,
|ttnn.maximum|aten::relu|4| +|80|Tensor<[1,32,120,160]>,
Tensor<[1,32,120,160]>,
|ttnn.maximum|aten::relu|4| +|81|Tensor<[1,64,480,640]>,
Tensor<[1,64,480,640]>,
|ttnn.maximum|aten::relu|4| +|82|Tensor<[1,256,128,128]>,
Tensor<[1,256,128,128]>,
|ttnn.maximum|aten::relu|4| +|83|Tensor<[1,96,112,112]>,
Tensor<[1,96,112,112]>,
|ttnn.maximum|aten::hardtanh|4| +|84|Tensor<[1,96,56,56]>,
Tensor<[1,96,56,56]>,
|ttnn.maximum|aten::hardtanh|4| +|85|Tensor<[1,144,56,56]>,
Tensor<[1,144,56,56]>,
|ttnn.maximum|aten::hardtanh|4| +|86|Tensor<[1,144,28,28]>,
Tensor<[1,144,28,28]>,
|ttnn.maximum|aten::hardtanh|4| +|87|Tensor<[1,192,28,28]>,
Tensor<[1,192,28,28]>,
|ttnn.maximum|aten::hardtanh|4| +|88|Tensor<[1,192,14,14]>,
Tensor<[1,192,14,14]>,
|ttnn.maximum|aten::hardtanh|4| +|89|Tensor<[1,384,14,14]>,
Tensor<[1,384,14,14]>,
|ttnn.maximum|aten::hardtanh|4| +|90|Tensor<[1,576,14,14]>,
Tensor<[1,576,14,14]>,
|ttnn.maximum|aten::hardtanh|4| +|91|Tensor<[1,576,7,7]>,
Tensor<[1,576,7,7]>,
|ttnn.maximum|aten::hardtanh|4| +|92|Tensor<[1,960,7,7]>,
Tensor<[1,960,7,7]>,
|ttnn.maximum|aten::hardtanh|4| +|93|Tensor<[1,1280,7,7]>,
Tensor<[1,1280,7,7]>,
|ttnn.maximum|aten::hardtanh|4| +|94|Tensor<[1,64]>,
Tensor<[1,64]>,
|ttnn.maximum|aten::relu|4| +|95|Tensor<[1,12]>,
Tensor<[1,12]>,
|ttnn.maximum|aten::relu|4| diff --git a/docs/ops/stablehlo/stablehlo.minimum.md b/docs/ops/stablehlo/stablehlo.minimum.md new file mode 100644 index 00000000..43b288d3 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.minimum.md @@ -0,0 +1,60 @@ + +### stablehlo.minimum::ttnn.minimum + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,128,28,28]>,
Tensor<[1,128,28,28]>,
|ttnn.minimum|aten::elu|4| +|1|Tensor<[1,32,112,112]>,
Tensor<[1,32,112,112]>,
|ttnn.minimum|aten::hardtanh|4| +|2|Tensor<[1,64,112,112]>,
Tensor<[1,64,112,112]>,
|ttnn.minimum|aten::hardtanh|4| +|3|Tensor<[1,14,56,56]>,
Tensor<[1,14,56,56]>,
|ttnn.minimum|aten::hardtanh|4| +|4|Tensor<[1,24,56,56]>,
Tensor<[1,24,56,56]>,
|ttnn.minimum|aten::hardtanh|4| +|5|Tensor<[1,40,56,56]>,
Tensor<[1,40,56,56]>,
|ttnn.minimum|aten::hardtanh|4| +|6|Tensor<[1,68,56,56]>,
Tensor<[1,68,56,56]>,
|ttnn.minimum|aten::hardtanh|4| +|7|Tensor<[1,128,56,56]>,
Tensor<[1,128,56,56]>,
|ttnn.minimum|aten::hardtanh|4| +|8|Tensor<[1,16,28,28]>,
Tensor<[1,16,28,28]>,
|ttnn.minimum|aten::hardtanh|4| +|9|Tensor<[1,28,28,28]>,
Tensor<[1,28,28,28]>,
|ttnn.minimum|aten::hardtanh|4| +|10|Tensor<[1,46,28,28]>,
Tensor<[1,46,28,28]>,
|ttnn.minimum|aten::hardtanh|4| +|11|Tensor<[1,78,28,28]>,
Tensor<[1,78,28,28]>,
|ttnn.minimum|aten::hardtanh|4| +|12|Tensor<[1,134,28,28]>,
Tensor<[1,134,28,28]>,
|ttnn.minimum|aten::hardtanh|4| +|13|Tensor<[1,256,28,28]>,
Tensor<[1,256,28,28]>,
|ttnn.minimum|aten::hardtanh|4| +|14|Tensor<[1,20,28,28]>,
Tensor<[1,20,28,28]>,
|ttnn.minimum|aten::hardtanh|4| +|15|Tensor<[1,34,28,28]>,
Tensor<[1,34,28,28]>,
|ttnn.minimum|aten::hardtanh|4| +|16|Tensor<[1,58,28,28]>,
Tensor<[1,58,28,28]>,
|ttnn.minimum|aten::hardtanh|4| +|17|Tensor<[1,98,28,28]>,
Tensor<[1,98,28,28]>,
|ttnn.minimum|aten::hardtanh|4| +|18|Tensor<[1,168,28,28]>,
Tensor<[1,168,28,28]>,
|ttnn.minimum|aten::hardtanh|4| +|19|Tensor<[1,320,28,28]>,
Tensor<[1,320,28,28]>,
|ttnn.minimum|aten::hardtanh|4| +|20|Tensor<[1,40,14,14]>,
Tensor<[1,40,14,14]>,
|ttnn.minimum|aten::hardtanh|4| +|21|Tensor<[1,68,14,14]>,
Tensor<[1,68,14,14]>,
|ttnn.minimum|aten::hardtanh|4| +|22|Tensor<[1,116,14,14]>,
Tensor<[1,116,14,14]>,
|ttnn.minimum|aten::hardtanh|4| +|23|Tensor<[1,196,14,14]>,
Tensor<[1,196,14,14]>,
|ttnn.minimum|aten::hardtanh|4| +|24|Tensor<[1,334,14,14]>,
Tensor<[1,334,14,14]>,
|ttnn.minimum|aten::hardtanh|4| +|25|Tensor<[1,640,14,14]>,
Tensor<[1,640,14,14]>,
|ttnn.minimum|aten::hardtanh|4| +|26|Tensor<[1,160,7,7]>,
Tensor<[1,160,7,7]>,
|ttnn.minimum|aten::hardtanh|4| +|27|Tensor<[1,272,7,7]>,
Tensor<[1,272,7,7]>,
|ttnn.minimum|aten::hardtanh|4| +|28|Tensor<[1,462,7,7]>,
Tensor<[1,462,7,7]>,
|ttnn.minimum|aten::hardtanh|4| +|29|Tensor<[1,1024,7,7]>,
Tensor<[1,1024,7,7]>,
|ttnn.minimum|aten::hardtanh|4| +|30|Tensor<[1,32,512,512]>,
Tensor<[1,32,512,512]>,
|ttnn.minimum|aten::leaky_relu|4| +|31|Tensor<[1,64,256,256]>,
Tensor<[1,64,256,256]>,
|ttnn.minimum|aten::leaky_relu|4| +|32|Tensor<[1,32,256,256]>,
Tensor<[1,32,256,256]>,
|ttnn.minimum|aten::leaky_relu|4| +|33|Tensor<[1,128,128,128]>,
Tensor<[1,128,128,128]>,
|ttnn.minimum|aten::leaky_relu|4| +|34|Tensor<[1,64,128,128]>,
Tensor<[1,64,128,128]>,
|ttnn.minimum|aten::leaky_relu|4| +|35|Tensor<[1,256,64,64]>,
Tensor<[1,256,64,64]>,
|ttnn.minimum|aten::leaky_relu|4| +|36|Tensor<[1,128,64,64]>,
Tensor<[1,128,64,64]>,
|ttnn.minimum|aten::leaky_relu|4| +|37|Tensor<[1,512,32,32]>,
Tensor<[1,512,32,32]>,
|ttnn.minimum|aten::leaky_relu|4| +|38|Tensor<[1,256,32,32]>,
Tensor<[1,256,32,32]>,
|ttnn.minimum|aten::leaky_relu|4| +|39|Tensor<[1,1024,16,16]>,
Tensor<[1,1024,16,16]>,
|ttnn.minimum|aten::leaky_relu|4| +|40|Tensor<[1,512,16,16]>,
Tensor<[1,512,16,16]>,
|ttnn.minimum|aten::leaky_relu|4| +|41|Tensor<[1,256,16,16]>,
Tensor<[1,256,16,16]>,
|ttnn.minimum|aten::leaky_relu|4| +|42|Tensor<[1,128,32,32]>,
Tensor<[1,128,32,32]>,
|ttnn.minimum|aten::leaky_relu|4| +|43|Tensor<[1,96,112,112]>,
Tensor<[1,96,112,112]>,
|ttnn.minimum|aten::hardtanh|4| +|44|Tensor<[1,96,56,56]>,
Tensor<[1,96,56,56]>,
|ttnn.minimum|aten::hardtanh|4| +|45|Tensor<[1,144,56,56]>,
Tensor<[1,144,56,56]>,
|ttnn.minimum|aten::hardtanh|4| +|46|Tensor<[1,144,28,28]>,
Tensor<[1,144,28,28]>,
|ttnn.minimum|aten::hardtanh|4| +|47|Tensor<[1,192,28,28]>,
Tensor<[1,192,28,28]>,
|ttnn.minimum|aten::hardtanh|4| +|48|Tensor<[1,192,14,14]>,
Tensor<[1,192,14,14]>,
|ttnn.minimum|aten::hardtanh|4| +|49|Tensor<[1,384,14,14]>,
Tensor<[1,384,14,14]>,
|ttnn.minimum|aten::hardtanh|4| +|50|Tensor<[1,576,14,14]>,
Tensor<[1,576,14,14]>,
|ttnn.minimum|aten::hardtanh|4| +|51|Tensor<[1,576,7,7]>,
Tensor<[1,576,7,7]>,
|ttnn.minimum|aten::hardtanh|4| +|52|Tensor<[1,960,7,7]>,
Tensor<[1,960,7,7]>,
|ttnn.minimum|aten::hardtanh|4| +|53|Tensor<[1,1280,7,7]>,
Tensor<[1,1280,7,7]>,
|ttnn.minimum|aten::hardtanh|4| diff --git a/docs/ops/stablehlo/stablehlo.multiply.md b/docs/ops/stablehlo/stablehlo.multiply.md new file mode 100644 index 00000000..af746ae6 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.multiply.md @@ -0,0 +1,444 @@ + +### stablehlo.multiply::ttnn.multiply + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[32]>,
Tensor<[32]>,
|ttnn.multiply|aten::arange|4| +|1|Tensor<[1,32,32,128]>,
Tensor<[1,32,32,128]>,
|ttnn.multiply|aten::mul.Scalar|4| +|2|Tensor<[1,32,128,32]>,
Tensor<[1,32,128,32]>,
|ttnn.multiply|aten::mul.Scalar|4| +|3|Tensor<[32,32]>,
Tensor<[32,32]>,
|ttnn.multiply|aten::mul.Tensor|5| +|4|Tensor<[1,32,128]>,
Tensor<[1,32,128]>,
|ttnn.multiply|aten::mul.Tensor|4| +|5|Tensor<[1,32,4096]>,
Tensor<[1,32,4096]>,
|ttnn.multiply|aten::mul.Tensor|4| +|6|Tensor<[1,32,11008]>,
Tensor<[1,32,11008]>,
|ttnn.multiply|aten::mul.Tensor|5| +|7|Tensor<[7]>,
Tensor<[7]>,
|ttnn.multiply|aten::arange|4| +|8|Tensor<[1]>,
Tensor<[1]>,
|ttnn.multiply|aten::arange|4| +|9|Tensor<[1,12,7,64]>,
Tensor<[1,12,7,64]>,
|ttnn.multiply|aten::mul.Scalar|4| +|10|Tensor<[1,12,64,7]>,
Tensor<[1,12,64,7]>,
|ttnn.multiply|aten::mul.Scalar|4| +|11|Tensor<[1,7,768]>,
Tensor<[1,7,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|12|Tensor<[7,2304]>,
Tensor<[7,2304]>,
|ttnn.multiply|aten::mul.Tensor|4| +|13|Tensor<[2304]>,
Tensor<[2304]>,
|ttnn.multiply|aten::mul.Tensor|4| +|14|Tensor<[7,768]>,
Tensor<[7,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|15|Tensor<[768]>,
Tensor<[768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|16|Tensor<[7,3072]>,
Tensor<[7,3072]>,
|ttnn.multiply|aten::mul.Tensor|4| +|17|Tensor<[3072]>,
Tensor<[3072]>,
|ttnn.multiply|aten::mul.Tensor|4| +|18|Tensor<[1,7,3072]>,
Tensor<[1,7,3072]>,
|ttnn.multiply|aten::mul.Tensor|4| +|19|Tensor<[1,128,28,28]>,
Tensor<[1,128,28,28]>,
|ttnn.multiply|aten::elu|4| +|20|Tensor<[1,32,112,112]>,
Tensor<[1,32,112,112]>,
|ttnn.multiply|aten::mul.Tensor|4| +|21|Tensor<[64]>,
Tensor<[64]>,
|ttnn.multiply|aten::mul.Tensor|4| +|22|Tensor<[1,64,112,112]>,
Tensor<[1,64,112,112]>,
|ttnn.multiply|aten::mul.Tensor|4| +|23|Tensor<[1,64,56,56]>,
Tensor<[1,64,56,56]>,
|ttnn.multiply|aten::mul.Tensor|4| +|24|Tensor<[128]>,
Tensor<[128]>,
|ttnn.multiply|aten::mul.Tensor|4| +|25|Tensor<[1,128,56,56]>,
Tensor<[1,128,56,56]>,
|ttnn.multiply|aten::mul.Tensor|4| +|26|Tensor<[256]>,
Tensor<[256]>,
|ttnn.multiply|aten::mul.Tensor|4| +|27|Tensor<[1,256,28,28]>,
Tensor<[1,256,28,28]>,
|ttnn.multiply|aten::mul.Tensor|4| +|28|Tensor<[512]>,
Tensor<[512]>,
|ttnn.multiply|aten::mul.Tensor|4| +|29|Tensor<[1,512,28,28]>,
Tensor<[1,512,28,28]>,
|ttnn.multiply|aten::mul.Tensor|4| +|30|Tensor<[1,1024,512]>,
Tensor<[1,1024,512]>,
|ttnn.multiply|aten::gelu|4| +|31|Tensor<[1,256,256]>,
Tensor<[1,256,256]>,
|ttnn.multiply|aten::gelu|4| +|32|Tensor<[256,512]>,
Tensor<[256,512]>,
|ttnn.multiply|aten::mul.Tensor|4| +|33|Tensor<[1,256,512]>,
Tensor<[1,256,512]>,
|ttnn.multiply|aten::mul.Tensor|4| +|34|Tensor<[256,256]>,
Tensor<[256,256]>,
|ttnn.multiply|aten::mul.Tensor|4| +|35|Tensor<[1,1000]>,
Tensor<[1,1000]>,
|ttnn.multiply|aten::mul.Tensor|4| +|36|Tensor<[1000]>,
Tensor<[1000]>,
|ttnn.multiply|aten::mul.Tensor|4| +|37|Tensor<[23]>,
Tensor<[23]>,
|ttnn.multiply|aten::arange|4| +|38|Tensor<[40]>,
Tensor<[40]>,
|ttnn.multiply|aten::arange|4| +|39|Tensor<[8,920,920]>,
Tensor<[8,920,920]>,
|ttnn.multiply|aten::baddbmm|4| +|40|Tensor<[8,100,920]>,
Tensor<[8,100,920]>,
|ttnn.multiply|aten::baddbmm|4| +|41|Tensor<[1,64,1,1]>,
Tensor<[1,64,1,1]>,
|ttnn.multiply|aten::mul.Tensor|5| +|42|Tensor<[1,64,360,640]>,
Tensor<[1,64,360,640]>,
|ttnn.multiply|aten::mul.Tensor|4| +|43|Tensor<[1,64,180,320]>,
Tensor<[1,64,180,320]>,
|ttnn.multiply|aten::mul.Tensor|4| +|44|Tensor<[1,256,1,1]>,
Tensor<[1,256,1,1]>,
|ttnn.multiply|aten::mul.Tensor|5| +|45|Tensor<[1,256,180,320]>,
Tensor<[1,256,180,320]>,
|ttnn.multiply|aten::mul.Tensor|4| +|46|Tensor<[1,128,1,1]>,
Tensor<[1,128,1,1]>,
|ttnn.multiply|aten::mul.Tensor|5| +|47|Tensor<[1,128,180,320]>,
Tensor<[1,128,180,320]>,
|ttnn.multiply|aten::mul.Tensor|4| +|48|Tensor<[1,128,90,160]>,
Tensor<[1,128,90,160]>,
|ttnn.multiply|aten::mul.Tensor|4| +|49|Tensor<[1,512,1,1]>,
Tensor<[1,512,1,1]>,
|ttnn.multiply|aten::mul.Tensor|5| +|50|Tensor<[1,512,90,160]>,
Tensor<[1,512,90,160]>,
|ttnn.multiply|aten::mul.Tensor|4| +|51|Tensor<[1,256,90,160]>,
Tensor<[1,256,90,160]>,
|ttnn.multiply|aten::mul.Tensor|4| +|52|Tensor<[1,256,45,80]>,
Tensor<[1,256,45,80]>,
|ttnn.multiply|aten::mul.Tensor|4| +|53|Tensor<[1,1024,1,1]>,
Tensor<[1,1024,1,1]>,
|ttnn.multiply|aten::mul.Tensor|5| +|54|Tensor<[1,1024,45,80]>,
Tensor<[1,1024,45,80]>,
|ttnn.multiply|aten::mul.Tensor|4| +|55|Tensor<[1,512,45,80]>,
Tensor<[1,512,45,80]>,
|ttnn.multiply|aten::mul.Tensor|4| +|56|Tensor<[1,512,23,40]>,
Tensor<[1,512,23,40]>,
|ttnn.multiply|aten::mul.Tensor|4| +|57|Tensor<[1,2048,1,1]>,
Tensor<[1,2048,1,1]>,
|ttnn.multiply|aten::mul.Tensor|5| +|58|Tensor<[1,2048,23,40]>,
Tensor<[1,2048,23,40]>,
|ttnn.multiply|aten::mul.Tensor|4| +|59|Tensor<[1,23,40]>,
Tensor<[1,23,40]>,
|ttnn.multiply|aten::mul.Tensor|4| +|60|Tensor<[8,920,32]>,
Tensor<[8,920,32]>,
|ttnn.multiply|aten::mul.Tensor|4| +|61|Tensor<[920,256]>,
Tensor<[920,256]>,
|ttnn.multiply|aten::mul.Tensor|4| +|62|Tensor<[920,1,256]>,
Tensor<[920,1,256]>,
|ttnn.multiply|aten::mul.Tensor|4| +|63|Tensor<[920,2048]>,
Tensor<[920,2048]>,
|ttnn.multiply|aten::mul.Tensor|4| +|64|Tensor<[2048]>,
Tensor<[2048]>,
|ttnn.multiply|aten::mul.Tensor|4| +|65|Tensor<[100,256]>,
Tensor<[100,256]>,
|ttnn.multiply|aten::mul.Tensor|4| +|66|Tensor<[8,100,32]>,
Tensor<[8,100,32]>,
|ttnn.multiply|aten::mul.Tensor|4| +|67|Tensor<[100,1,256]>,
Tensor<[100,1,256]>,
|ttnn.multiply|aten::mul.Tensor|4| +|68|Tensor<[100,2048]>,
Tensor<[100,2048]>,
|ttnn.multiply|aten::mul.Tensor|4| +|69|Tensor<[1,10,3072]>,
Tensor<[1,10,3072]>,
|ttnn.multiply|aten::gelu|4| +|70|Tensor<[1,10,768]>,
Tensor<[1,10,768]>,
|ttnn.multiply|aten::gelu|4| +|71|Tensor<[1,12,10,64]>,
Tensor<[1,12,10,64]>,
|ttnn.multiply|aten::mul.Scalar|4| +|72|Tensor<[1,12,64,10]>,
Tensor<[1,12,64,10]>,
|ttnn.multiply|aten::mul.Scalar|4| +|73|Tensor<[1,10]>,
Tensor<[1,10]>,
|ttnn.multiply|aten::mul.Tensor|5| +|74|Tensor<[10,768]>,
Tensor<[10,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|75|Tensor<[10,3072]>,
Tensor<[10,3072]>,
|ttnn.multiply|aten::mul.Tensor|4| +|76|Tensor<[10,250002]>,
Tensor<[10,250002]>,
|ttnn.multiply|aten::mul.Tensor|4| +|77|Tensor<[250002]>,
Tensor<[250002]>,
|ttnn.multiply|aten::mul.Tensor|4| +|78|Tensor<[16]>,
Tensor<[16]>,
|ttnn.multiply|aten::arange|4| +|79|Tensor<[160]>,
Tensor<[160]>,
|ttnn.multiply|aten::arange.start|4| +|80|Tensor<[1,4096,1280]>,
Tensor<[1,4096,1280]>,
|ttnn.multiply|aten::gelu|4| +|81|Tensor<[1,1024,2560]>,
Tensor<[1,1024,2560]>,
|ttnn.multiply|aten::gelu|4| +|82|Tensor<[1,256,5120]>,
Tensor<[1,256,5120]>,
|ttnn.multiply|aten::gelu|4| +|83|Tensor<[1,64,5120]>,
Tensor<[1,64,5120]>,
|ttnn.multiply|aten::gelu|4| +|84|Tensor<[1280]>,
Tensor<[1280]>,
|ttnn.multiply|aten::index.Tensor|4| +|85|Tensor<[640]>,
Tensor<[640]>,
|ttnn.multiply|aten::index.Tensor|4| +|86|Tensor<[1,8,4096,40]>,
Tensor<[1,8,4096,40]>,
|ttnn.multiply|aten::mul.Scalar|4| +|87|Tensor<[1,8,40,4096]>,
Tensor<[1,8,40,4096]>,
|ttnn.multiply|aten::mul.Scalar|4| +|88|Tensor<[1,8,40,9]>,
Tensor<[1,8,40,9]>,
|ttnn.multiply|aten::mul.Scalar|4| +|89|Tensor<[1,8,1024,80]>,
Tensor<[1,8,1024,80]>,
|ttnn.multiply|aten::mul.Scalar|4| +|90|Tensor<[1,8,80,1024]>,
Tensor<[1,8,80,1024]>,
|ttnn.multiply|aten::mul.Scalar|4| +|91|Tensor<[1,8,80,9]>,
Tensor<[1,8,80,9]>,
|ttnn.multiply|aten::mul.Scalar|4| +|92|Tensor<[1,8,256,160]>,
Tensor<[1,8,256,160]>,
|ttnn.multiply|aten::mul.Scalar|4| +|93|Tensor<[1,8,160,256]>,
Tensor<[1,8,160,256]>,
|ttnn.multiply|aten::mul.Scalar|4| +|94|Tensor<[1,8,160,9]>,
Tensor<[1,8,160,9]>,
|ttnn.multiply|aten::mul.Scalar|4| +|95|Tensor<[1,8,64,160]>,
Tensor<[1,8,64,160]>,
|ttnn.multiply|aten::mul.Scalar|4| +|96|Tensor<[1,8,160,64]>,
Tensor<[1,8,160,64]>,
|ttnn.multiply|aten::mul.Scalar|4| +|97|Tensor<[1,160]>,
Tensor<[1,160]>,
|ttnn.multiply|aten::mul.Tensor|4| +|98|Tensor<[1,1280]>,
Tensor<[1,1280]>,
|ttnn.multiply|aten::mul.Tensor|4| +|99|Tensor<[1,32,10,4096]>,
Tensor<[1,32,10,4096]>,
|ttnn.multiply|aten::mul.Tensor|4| +|100|Tensor<[1,320,64,64]>,
Tensor<[1,320,64,64]>,
|ttnn.multiply|aten::mul.Tensor|4| +|101|Tensor<[1,320]>,
Tensor<[1,320]>,
|ttnn.multiply|aten::mul.Tensor|4| +|102|Tensor<[320]>,
Tensor<[320]>,
|ttnn.multiply|aten::mul.Tensor|4| +|103|Tensor<[1,4096,320]>,
Tensor<[1,4096,320]>,
|ttnn.multiply|aten::mul.Tensor|4| +|104|Tensor<[4096,320]>,
Tensor<[4096,320]>,
|ttnn.multiply|aten::mul.Tensor|4| +|105|Tensor<[4096,2560]>,
Tensor<[4096,2560]>,
|ttnn.multiply|aten::mul.Tensor|4| +|106|Tensor<[2560]>,
Tensor<[2560]>,
|ttnn.multiply|aten::mul.Tensor|4| +|107|Tensor<[1,32,10,1024]>,
Tensor<[1,32,10,1024]>,
|ttnn.multiply|aten::mul.Tensor|4| +|108|Tensor<[1,320,32,32]>,
Tensor<[1,320,32,32]>,
|ttnn.multiply|aten::mul.Tensor|4| +|109|Tensor<[1,640]>,
Tensor<[1,640]>,
|ttnn.multiply|aten::mul.Tensor|4| +|110|Tensor<[1,32,20,1024]>,
Tensor<[1,32,20,1024]>,
|ttnn.multiply|aten::mul.Tensor|4| +|111|Tensor<[1,640,32,32]>,
Tensor<[1,640,32,32]>,
|ttnn.multiply|aten::mul.Tensor|4| +|112|Tensor<[1,1024,640]>,
Tensor<[1,1024,640]>,
|ttnn.multiply|aten::mul.Tensor|4| +|113|Tensor<[1024,640]>,
Tensor<[1024,640]>,
|ttnn.multiply|aten::mul.Tensor|4| +|114|Tensor<[1024,5120]>,
Tensor<[1024,5120]>,
|ttnn.multiply|aten::mul.Tensor|4| +|115|Tensor<[5120]>,
Tensor<[5120]>,
|ttnn.multiply|aten::mul.Tensor|4| +|116|Tensor<[1,32,20,256]>,
Tensor<[1,32,20,256]>,
|ttnn.multiply|aten::mul.Tensor|4| +|117|Tensor<[1,640,16,16]>,
Tensor<[1,640,16,16]>,
|ttnn.multiply|aten::mul.Tensor|4| +|118|Tensor<[1,32,40,256]>,
Tensor<[1,32,40,256]>,
|ttnn.multiply|aten::mul.Tensor|4| +|119|Tensor<[1,1280,16,16]>,
Tensor<[1,1280,16,16]>,
|ttnn.multiply|aten::mul.Tensor|4| +|120|Tensor<[1,256,1280]>,
Tensor<[1,256,1280]>,
|ttnn.multiply|aten::mul.Tensor|4| +|121|Tensor<[256,1280]>,
Tensor<[256,1280]>,
|ttnn.multiply|aten::mul.Tensor|4| +|122|Tensor<[256,10240]>,
Tensor<[256,10240]>,
|ttnn.multiply|aten::mul.Tensor|4| +|123|Tensor<[10240]>,
Tensor<[10240]>,
|ttnn.multiply|aten::mul.Tensor|4| +|124|Tensor<[1,32,40,64]>,
Tensor<[1,32,40,64]>,
|ttnn.multiply|aten::mul.Tensor|4| +|125|Tensor<[1,1280,8,8]>,
Tensor<[1,1280,8,8]>,
|ttnn.multiply|aten::mul.Tensor|4| +|126|Tensor<[1,64,1280]>,
Tensor<[1,64,1280]>,
|ttnn.multiply|aten::mul.Tensor|4| +|127|Tensor<[64,1280]>,
Tensor<[64,1280]>,
|ttnn.multiply|aten::mul.Tensor|4| +|128|Tensor<[64,10240]>,
Tensor<[64,10240]>,
|ttnn.multiply|aten::mul.Tensor|4| +|129|Tensor<[1,32,80,64]>,
Tensor<[1,32,80,64]>,
|ttnn.multiply|aten::mul.Tensor|4| +|130|Tensor<[1,2560,8,8]>,
Tensor<[1,2560,8,8]>,
|ttnn.multiply|aten::mul.Tensor|4| +|131|Tensor<[1,32,80,256]>,
Tensor<[1,32,80,256]>,
|ttnn.multiply|aten::mul.Tensor|4| +|132|Tensor<[1,2560,16,16]>,
Tensor<[1,2560,16,16]>,
|ttnn.multiply|aten::mul.Tensor|4| +|133|Tensor<[1,32,60,256]>,
Tensor<[1,32,60,256]>,
|ttnn.multiply|aten::mul.Tensor|4| +|134|Tensor<[1,1920,16,16]>,
Tensor<[1,1920,16,16]>,
|ttnn.multiply|aten::mul.Tensor|4| +|135|Tensor<[1,32,60,1024]>,
Tensor<[1,32,60,1024]>,
|ttnn.multiply|aten::mul.Tensor|4| +|136|Tensor<[1,1920,32,32]>,
Tensor<[1,1920,32,32]>,
|ttnn.multiply|aten::mul.Tensor|4| +|137|Tensor<[1,32,40,1024]>,
Tensor<[1,32,40,1024]>,
|ttnn.multiply|aten::mul.Tensor|4| +|138|Tensor<[1,1280,32,32]>,
Tensor<[1,1280,32,32]>,
|ttnn.multiply|aten::mul.Tensor|4| +|139|Tensor<[1,32,30,1024]>,
Tensor<[1,32,30,1024]>,
|ttnn.multiply|aten::mul.Tensor|4| +|140|Tensor<[1,960,32,32]>,
Tensor<[1,960,32,32]>,
|ttnn.multiply|aten::mul.Tensor|4| +|141|Tensor<[1,32,30,4096]>,
Tensor<[1,32,30,4096]>,
|ttnn.multiply|aten::mul.Tensor|4| +|142|Tensor<[1,960,64,64]>,
Tensor<[1,960,64,64]>,
|ttnn.multiply|aten::mul.Tensor|4| +|143|Tensor<[1,32,20,4096]>,
Tensor<[1,32,20,4096]>,
|ttnn.multiply|aten::mul.Tensor|4| +|144|Tensor<[1,640,64,64]>,
Tensor<[1,640,64,64]>,
|ttnn.multiply|aten::mul.Tensor|4| +|145|Tensor<[1,25,3072]>,
Tensor<[1,25,3072]>,
|ttnn.multiply|aten::gelu|4| +|146|Tensor<[1,12,25,64]>,
Tensor<[1,12,25,64]>,
|ttnn.multiply|aten::mul.Scalar|4| +|147|Tensor<[1,12,64,25]>,
Tensor<[1,12,64,25]>,
|ttnn.multiply|aten::mul.Scalar|4| +|148|Tensor<[1,25,768]>,
Tensor<[1,25,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|149|Tensor<[25,768]>,
Tensor<[25,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|150|Tensor<[25,3072]>,
Tensor<[25,3072]>,
|ttnn.multiply|aten::mul.Tensor|4| +|151|Tensor<[25,2]>,
Tensor<[25,2]>,
|ttnn.multiply|aten::mul.Tensor|4| +|152|Tensor<[2]>,
Tensor<[2]>,
|ttnn.multiply|aten::mul.Tensor|4| +|153|Tensor<[1,1]>,
Tensor<[1,1]>,
|ttnn.multiply|aten::mul.Tensor|4| +|154|Tensor<[1,1445,768]>,
Tensor<[1,1445,768]>,
|ttnn.multiply|aten::gelu|4| +|155|Tensor<[1,3,1445,64]>,
Tensor<[1,3,1445,64]>,
|ttnn.multiply|aten::mul.Scalar|4| +|156|Tensor<[1,3,64,1445]>,
Tensor<[1,3,64,1445]>,
|ttnn.multiply|aten::mul.Scalar|4| +|157|Tensor<[1,1445,192]>,
Tensor<[1,1445,192]>,
|ttnn.multiply|aten::mul.Tensor|4| +|158|Tensor<[1445,192]>,
Tensor<[1445,192]>,
|ttnn.multiply|aten::mul.Tensor|4| +|159|Tensor<[192]>,
Tensor<[192]>,
|ttnn.multiply|aten::mul.Tensor|4| +|160|Tensor<[1445,768]>,
Tensor<[1445,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|161|Tensor<[100,192]>,
Tensor<[100,192]>,
|ttnn.multiply|aten::mul.Tensor|4| +|162|Tensor<[100,92]>,
Tensor<[100,92]>,
|ttnn.multiply|aten::mul.Tensor|4| +|163|Tensor<[92]>,
Tensor<[92]>,
|ttnn.multiply|aten::mul.Tensor|4| +|164|Tensor<[100,4]>,
Tensor<[100,4]>,
|ttnn.multiply|aten::mul.Tensor|4| +|165|Tensor<[4]>,
Tensor<[4]>,
|ttnn.multiply|aten::mul.Tensor|4| +|166|Tensor<[1,256,14,14]>,
Tensor<[1,256,14,14]>,
|ttnn.multiply|aten::mul.Tensor|4| +|167|Tensor<[1,512,7,7]>,
Tensor<[1,512,7,7]>,
|ttnn.multiply|aten::mul.Tensor|4| +|168|Tensor<[1,3072,8]>,
Tensor<[1,3072,8]>,
|ttnn.multiply|aten::gelu|4| +|169|Tensor<[1,1,1,8]>,
Tensor<[1,1,1,8]>,
|ttnn.multiply|aten::mul.Tensor|4| +|170|Tensor<[1,8,768]>,
Tensor<[1,8,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|171|Tensor<[1,768]>,
Tensor<[1,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|172|Tensor<[1,3]>,
Tensor<[1,3]>,
|ttnn.multiply|aten::mul.Tensor|4| +|173|Tensor<[3]>,
Tensor<[3]>,
|ttnn.multiply|aten::mul.Tensor|4| +|174|Tensor<[1,2048,768]>,
Tensor<[1,2048,768]>,
|ttnn.multiply|aten::gelu|4| +|175|Tensor<[1,1,1,2048]>,
Tensor<[1,1,1,2048]>,
|ttnn.multiply|aten::mul.Tensor|4| +|176|Tensor<[2048,256]>,
Tensor<[2048,256]>,
|ttnn.multiply|aten::mul.Tensor|4| +|177|Tensor<[2048,1280]>,
Tensor<[2048,1280]>,
|ttnn.multiply|aten::mul.Tensor|4| +|178|Tensor<[256,768]>,
Tensor<[256,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|179|Tensor<[2048,768]>,
Tensor<[2048,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|180|Tensor<[1,256,56,56]>,
Tensor<[1,256,56,56]>,
|ttnn.multiply|aten::mul.Tensor|4| +|181|Tensor<[1024]>,
Tensor<[1024]>,
|ttnn.multiply|aten::mul.Tensor|4| +|182|Tensor<[1,1024,14,14]>,
Tensor<[1,1024,14,14]>,
|ttnn.multiply|aten::mul.Tensor|4| +|183|Tensor<[1,512,14,14]>,
Tensor<[1,512,14,14]>,
|ttnn.multiply|aten::mul.Tensor|4| +|184|Tensor<[1,2048,7,7]>,
Tensor<[1,2048,7,7]>,
|ttnn.multiply|aten::mul.Tensor|4| +|185|Tensor<[12]>,
Tensor<[12]>,
|ttnn.multiply|aten::arange|4| +|186|Tensor<[1,201,3072]>,
Tensor<[1,201,3072]>,
|ttnn.multiply|aten::gelu|4| +|187|Tensor<[1,1536]>,
Tensor<[1,1536]>,
|ttnn.multiply|aten::gelu|4| +|188|Tensor<[1,1,1,201]>,
Tensor<[1,1,1,201]>,
|ttnn.multiply|aten::mul.Tensor|4| +|189|Tensor<[1,201,768]>,
Tensor<[1,201,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|190|Tensor<[201,768]>,
Tensor<[201,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|191|Tensor<[201,3072]>,
Tensor<[201,3072]>,
|ttnn.multiply|aten::mul.Tensor|4| +|192|Tensor<[1536]>,
Tensor<[1536]>,
|ttnn.multiply|aten::mul.Tensor|4| +|193|Tensor<[1,3129]>,
Tensor<[1,3129]>,
|ttnn.multiply|aten::mul.Tensor|4| +|194|Tensor<[3129]>,
Tensor<[3129]>,
|ttnn.multiply|aten::mul.Tensor|4| +|195|Tensor<[1,128]>,
Tensor<[1,128]>,
|ttnn.multiply|aten::mul.Tensor|4| +|196|Tensor<[10]>,
Tensor<[10]>,
|ttnn.multiply|aten::mul.Tensor|4| +|197|Tensor<[19]>,
Tensor<[19]>,
|ttnn.multiply|aten::arange|4| +|198|Tensor<[1,19,4096]>,
Tensor<[1,19,4096]>,
|ttnn.multiply|aten::gelu|4| +|199|Tensor<[1,19,1024]>,
Tensor<[1,19,1024]>,
|ttnn.multiply|aten::mul.Tensor|4| +|200|Tensor<[19,1024]>,
Tensor<[19,1024]>,
|ttnn.multiply|aten::mul.Tensor|4| +|201|Tensor<[19,4096]>,
Tensor<[19,4096]>,
|ttnn.multiply|aten::mul.Tensor|4| +|202|Tensor<[4096]>,
Tensor<[4096]>,
|ttnn.multiply|aten::mul.Tensor|4| +|203|Tensor<[14]>,
Tensor<[14]>,
|ttnn.multiply|aten::mul.Tensor|4| +|204|Tensor<[1,14,56,56]>,
Tensor<[1,14,56,56]>,
|ttnn.multiply|aten::mul.Tensor|4| +|205|Tensor<[24]>,
Tensor<[24]>,
|ttnn.multiply|aten::mul.Tensor|4| +|206|Tensor<[1,24,56,56]>,
Tensor<[1,24,56,56]>,
|ttnn.multiply|aten::mul.Tensor|4| +|207|Tensor<[1,40,56,56]>,
Tensor<[1,40,56,56]>,
|ttnn.multiply|aten::mul.Tensor|4| +|208|Tensor<[68]>,
Tensor<[68]>,
|ttnn.multiply|aten::mul.Tensor|4| +|209|Tensor<[1,68,56,56]>,
Tensor<[1,68,56,56]>,
|ttnn.multiply|aten::mul.Tensor|4| +|210|Tensor<[1,16,28,28]>,
Tensor<[1,16,28,28]>,
|ttnn.multiply|aten::mul.Tensor|4| +|211|Tensor<[28]>,
Tensor<[28]>,
|ttnn.multiply|aten::mul.Tensor|4| +|212|Tensor<[1,28,28,28]>,
Tensor<[1,28,28,28]>,
|ttnn.multiply|aten::mul.Tensor|4| +|213|Tensor<[46]>,
Tensor<[46]>,
|ttnn.multiply|aten::mul.Tensor|4| +|214|Tensor<[1,46,28,28]>,
Tensor<[1,46,28,28]>,
|ttnn.multiply|aten::mul.Tensor|4| +|215|Tensor<[78]>,
Tensor<[78]>,
|ttnn.multiply|aten::mul.Tensor|4| +|216|Tensor<[1,78,28,28]>,
Tensor<[1,78,28,28]>,
|ttnn.multiply|aten::mul.Tensor|4| +|217|Tensor<[134]>,
Tensor<[134]>,
|ttnn.multiply|aten::mul.Tensor|4| +|218|Tensor<[1,134,28,28]>,
Tensor<[1,134,28,28]>,
|ttnn.multiply|aten::mul.Tensor|4| +|219|Tensor<[20]>,
Tensor<[20]>,
|ttnn.multiply|aten::mul.Tensor|4| +|220|Tensor<[1,20,28,28]>,
Tensor<[1,20,28,28]>,
|ttnn.multiply|aten::mul.Tensor|4| +|221|Tensor<[34]>,
Tensor<[34]>,
|ttnn.multiply|aten::mul.Tensor|4| +|222|Tensor<[1,34,28,28]>,
Tensor<[1,34,28,28]>,
|ttnn.multiply|aten::mul.Tensor|4| +|223|Tensor<[58]>,
Tensor<[58]>,
|ttnn.multiply|aten::mul.Tensor|4| +|224|Tensor<[1,58,28,28]>,
Tensor<[1,58,28,28]>,
|ttnn.multiply|aten::mul.Tensor|4| +|225|Tensor<[98]>,
Tensor<[98]>,
|ttnn.multiply|aten::mul.Tensor|4| +|226|Tensor<[1,98,28,28]>,
Tensor<[1,98,28,28]>,
|ttnn.multiply|aten::mul.Tensor|4| +|227|Tensor<[168]>,
Tensor<[168]>,
|ttnn.multiply|aten::mul.Tensor|4| +|228|Tensor<[1,168,28,28]>,
Tensor<[1,168,28,28]>,
|ttnn.multiply|aten::mul.Tensor|4| +|229|Tensor<[1,320,28,28]>,
Tensor<[1,320,28,28]>,
|ttnn.multiply|aten::mul.Tensor|4| +|230|Tensor<[1,40,14,14]>,
Tensor<[1,40,14,14]>,
|ttnn.multiply|aten::mul.Tensor|4| +|231|Tensor<[1,68,14,14]>,
Tensor<[1,68,14,14]>,
|ttnn.multiply|aten::mul.Tensor|4| +|232|Tensor<[116]>,
Tensor<[116]>,
|ttnn.multiply|aten::mul.Tensor|4| +|233|Tensor<[1,116,14,14]>,
Tensor<[1,116,14,14]>,
|ttnn.multiply|aten::mul.Tensor|4| +|234|Tensor<[196]>,
Tensor<[196]>,
|ttnn.multiply|aten::mul.Tensor|4| +|235|Tensor<[1,196,14,14]>,
Tensor<[1,196,14,14]>,
|ttnn.multiply|aten::mul.Tensor|4| +|236|Tensor<[334]>,
Tensor<[334]>,
|ttnn.multiply|aten::mul.Tensor|4| +|237|Tensor<[1,334,14,14]>,
Tensor<[1,334,14,14]>,
|ttnn.multiply|aten::mul.Tensor|4| +|238|Tensor<[1,640,14,14]>,
Tensor<[1,640,14,14]>,
|ttnn.multiply|aten::mul.Tensor|4| +|239|Tensor<[1,160,7,7]>,
Tensor<[1,160,7,7]>,
|ttnn.multiply|aten::mul.Tensor|4| +|240|Tensor<[272]>,
Tensor<[272]>,
|ttnn.multiply|aten::mul.Tensor|4| +|241|Tensor<[1,272,7,7]>,
Tensor<[1,272,7,7]>,
|ttnn.multiply|aten::mul.Tensor|4| +|242|Tensor<[462]>,
Tensor<[462]>,
|ttnn.multiply|aten::mul.Tensor|4| +|243|Tensor<[1,462,7,7]>,
Tensor<[1,462,7,7]>,
|ttnn.multiply|aten::mul.Tensor|4| +|244|Tensor<[1,1024,7,7]>,
Tensor<[1,1024,7,7]>,
|ttnn.multiply|aten::mul.Tensor|4| +|245|Tensor<[1,32,512,512]>,
Tensor<[1,32,512,512]>,
|ttnn.multiply|aten::leaky_relu|4| +|246|Tensor<[1,64,256,256]>,
Tensor<[1,64,256,256]>,
|ttnn.multiply|aten::leaky_relu|4| +|247|Tensor<[1,32,256,256]>,
Tensor<[1,32,256,256]>,
|ttnn.multiply|aten::leaky_relu|4| +|248|Tensor<[1,128,128,128]>,
Tensor<[1,128,128,128]>,
|ttnn.multiply|aten::leaky_relu|4| +|249|Tensor<[1,64,128,128]>,
Tensor<[1,64,128,128]>,
|ttnn.multiply|aten::leaky_relu|4| +|250|Tensor<[1,256,64,64]>,
Tensor<[1,256,64,64]>,
|ttnn.multiply|aten::leaky_relu|4| +|251|Tensor<[1,128,64,64]>,
Tensor<[1,128,64,64]>,
|ttnn.multiply|aten::leaky_relu|4| +|252|Tensor<[1,512,32,32]>,
Tensor<[1,512,32,32]>,
|ttnn.multiply|aten::leaky_relu|4| +|253|Tensor<[1,256,32,32]>,
Tensor<[1,256,32,32]>,
|ttnn.multiply|aten::leaky_relu|4| +|254|Tensor<[1,1024,16,16]>,
Tensor<[1,1024,16,16]>,
|ttnn.multiply|aten::leaky_relu|4| +|255|Tensor<[1,512,16,16]>,
Tensor<[1,512,16,16]>,
|ttnn.multiply|aten::leaky_relu|4| +|256|Tensor<[1,256,16,16]>,
Tensor<[1,256,16,16]>,
|ttnn.multiply|aten::leaky_relu|4| +|257|Tensor<[1,128,32,32]>,
Tensor<[1,128,32,32]>,
|ttnn.multiply|aten::leaky_relu|4| +|258|Tensor<[16,32,32]>,
Tensor<[16,32,32]>,
|ttnn.multiply|aten::baddbmm|4| +|259|Tensor<[1,32,1536]>,
Tensor<[1,32,1536]>,
|ttnn.multiply|aten::mul.Tensor|4| +|260|Tensor<[1,32]>,
Tensor<[1,32]>,
|ttnn.multiply|aten::mul.Tensor|4| +|261|Tensor<[1,16,32]>,
Tensor<[1,16,32]>,
|ttnn.multiply|aten::mul.Tensor|4| +|262|Tensor<[32,4608]>,
Tensor<[32,4608]>,
|ttnn.multiply|aten::mul.Tensor|4| +|263|Tensor<[4608]>,
Tensor<[4608]>,
|ttnn.multiply|aten::mul.Tensor|4| +|264|Tensor<[32,1536]>,
Tensor<[32,1536]>,
|ttnn.multiply|aten::mul.Tensor|4| +|265|Tensor<[32,6144]>,
Tensor<[32,6144]>,
|ttnn.multiply|aten::mul.Tensor|4| +|266|Tensor<[6144]>,
Tensor<[6144]>,
|ttnn.multiply|aten::mul.Tensor|4| +|267|Tensor<[1,32,6144]>,
Tensor<[1,32,6144]>,
|ttnn.multiply|aten::mul.Tensor|4| +|268|Tensor<[1,16,3072]>,
Tensor<[1,16,3072]>,
|ttnn.multiply|aten::gelu|4| +|269|Tensor<[1,12,16,64]>,
Tensor<[1,12,16,64]>,
|ttnn.multiply|aten::mul.Scalar|4| +|270|Tensor<[1,12,64,16]>,
Tensor<[1,12,64,16]>,
|ttnn.multiply|aten::mul.Scalar|4| +|271|Tensor<[1,16,768]>,
Tensor<[1,16,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|272|Tensor<[16,768]>,
Tensor<[16,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|273|Tensor<[16,3072]>,
Tensor<[16,3072]>,
|ttnn.multiply|aten::mul.Tensor|4| +|274|Tensor<[1,64,224,224]>,
Tensor<[1,64,224,224]>,
|ttnn.multiply|aten::mul.Tensor|4| +|275|Tensor<[1,128,112,112]>,
Tensor<[1,128,112,112]>,
|ttnn.multiply|aten::mul.Tensor|4| +|276|Tensor<[30]>,
Tensor<[30]>,
|ttnn.multiply|aten::arange|4| +|277|Tensor<[60]>,
Tensor<[60]>,
|ttnn.multiply|aten::arange|4| +|278|Tensor<[80]>,
Tensor<[80]>,
|ttnn.multiply|aten::arange|4| +|279|Tensor<[120]>,
Tensor<[120]>,
|ttnn.multiply|aten::arange|4| +|280|Tensor<[240]>,
Tensor<[240]>,
|ttnn.multiply|aten::arange|4| +|281|Tensor<[480]>,
Tensor<[480]>,
|ttnn.multiply|aten::arange|4| +|282|Tensor<[1,19200,256]>,
Tensor<[1,19200,256]>,
|ttnn.multiply|aten::gelu|4| +|283|Tensor<[1,4800,512]>,
Tensor<[1,4800,512]>,
|ttnn.multiply|aten::gelu|4| +|284|Tensor<[1,1200,1280]>,
Tensor<[1,1200,1280]>,
|ttnn.multiply|aten::gelu|4| +|285|Tensor<[1,300,2048]>,
Tensor<[1,300,2048]>,
|ttnn.multiply|aten::gelu|4| +|286|Tensor<[1,19200,64]>,
Tensor<[1,19200,64]>,
|ttnn.multiply|aten::mul.Tensor|4| +|287|Tensor<[19200,64]>,
Tensor<[19200,64]>,
|ttnn.multiply|aten::mul.Tensor|4| +|288|Tensor<[1,300,64]>,
Tensor<[1,300,64]>,
|ttnn.multiply|aten::mul.Tensor|4| +|289|Tensor<[300,64]>,
Tensor<[300,64]>,
|ttnn.multiply|aten::mul.Tensor|4| +|290|Tensor<[19200,256]>,
Tensor<[19200,256]>,
|ttnn.multiply|aten::mul.Tensor|4| +|291|Tensor<[1,4800,128]>,
Tensor<[1,4800,128]>,
|ttnn.multiply|aten::mul.Tensor|4| +|292|Tensor<[4800,128]>,
Tensor<[4800,128]>,
|ttnn.multiply|aten::mul.Tensor|4| +|293|Tensor<[1,300,128]>,
Tensor<[1,300,128]>,
|ttnn.multiply|aten::mul.Tensor|4| +|294|Tensor<[300,128]>,
Tensor<[300,128]>,
|ttnn.multiply|aten::mul.Tensor|4| +|295|Tensor<[4800,512]>,
Tensor<[4800,512]>,
|ttnn.multiply|aten::mul.Tensor|4| +|296|Tensor<[1,1200,320]>,
Tensor<[1,1200,320]>,
|ttnn.multiply|aten::mul.Tensor|4| +|297|Tensor<[1200,320]>,
Tensor<[1200,320]>,
|ttnn.multiply|aten::mul.Tensor|4| +|298|Tensor<[1,300,320]>,
Tensor<[1,300,320]>,
|ttnn.multiply|aten::mul.Tensor|4| +|299|Tensor<[300,320]>,
Tensor<[300,320]>,
|ttnn.multiply|aten::mul.Tensor|4| +|300|Tensor<[1200,1280]>,
Tensor<[1200,1280]>,
|ttnn.multiply|aten::mul.Tensor|4| +|301|Tensor<[1,300,512]>,
Tensor<[1,300,512]>,
|ttnn.multiply|aten::mul.Tensor|4| +|302|Tensor<[300,512]>,
Tensor<[300,512]>,
|ttnn.multiply|aten::mul.Tensor|4| +|303|Tensor<[300,2048]>,
Tensor<[300,2048]>,
|ttnn.multiply|aten::mul.Tensor|4| +|304|Tensor<[1,64,30,40]>,
Tensor<[1,64,30,40]>,
|ttnn.multiply|aten::mul.Tensor|4| +|305|Tensor<[1,32,30,40]>,
Tensor<[1,32,30,40]>,
|ttnn.multiply|aten::mul.Tensor|4| +|306|Tensor<[1,64,60,80]>,
Tensor<[1,64,60,80]>,
|ttnn.multiply|aten::mul.Tensor|4| +|307|Tensor<[1,32,60,80]>,
Tensor<[1,32,60,80]>,
|ttnn.multiply|aten::mul.Tensor|4| +|308|Tensor<[1,64,120,160]>,
Tensor<[1,64,120,160]>,
|ttnn.multiply|aten::mul.Tensor|4| +|309|Tensor<[1,32,120,160]>,
Tensor<[1,32,120,160]>,
|ttnn.multiply|aten::mul.Tensor|4| +|310|Tensor<[1,64,240,320]>,
Tensor<[1,64,240,320]>,
|ttnn.multiply|aten::mul.Tensor|4| +|311|Tensor<[1,64,480,640]>,
Tensor<[1,64,480,640]>,
|ttnn.multiply|aten::mul.Tensor|4| +|312|Tensor<[1,1,480,640]>,
Tensor<[1,1,480,640]>,
|ttnn.multiply|aten::mul.Tensor|4| +|313|Tensor<[1,197,3072]>,
Tensor<[1,197,3072]>,
|ttnn.multiply|aten::gelu|4| +|314|Tensor<[1,12,197,64]>,
Tensor<[1,12,197,64]>,
|ttnn.multiply|aten::mul.Scalar|4| +|315|Tensor<[1,12,64,197]>,
Tensor<[1,12,64,197]>,
|ttnn.multiply|aten::mul.Scalar|4| +|316|Tensor<[1,197,768]>,
Tensor<[1,197,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|317|Tensor<[197,768]>,
Tensor<[197,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|318|Tensor<[197,3072]>,
Tensor<[197,3072]>,
|ttnn.multiply|aten::mul.Tensor|4| +|319|Tensor<[1,16384,128]>,
Tensor<[1,16384,128]>,
|ttnn.multiply|aten::gelu|4| +|320|Tensor<[1,4096,256]>,
Tensor<[1,4096,256]>,
|ttnn.multiply|aten::gelu|4| +|321|Tensor<[1,256,1024]>,
Tensor<[1,256,1024]>,
|ttnn.multiply|aten::gelu|4| +|322|Tensor<[1,16384,32]>,
Tensor<[1,16384,32]>,
|ttnn.multiply|aten::mul.Tensor|4| +|323|Tensor<[16384,32]>,
Tensor<[16384,32]>,
|ttnn.multiply|aten::mul.Tensor|4| +|324|Tensor<[1,256,32]>,
Tensor<[1,256,32]>,
|ttnn.multiply|aten::mul.Tensor|4| +|325|Tensor<[256,32]>,
Tensor<[256,32]>,
|ttnn.multiply|aten::mul.Tensor|4| +|326|Tensor<[16384,128]>,
Tensor<[16384,128]>,
|ttnn.multiply|aten::mul.Tensor|4| +|327|Tensor<[1,4096,64]>,
Tensor<[1,4096,64]>,
|ttnn.multiply|aten::mul.Tensor|4| +|328|Tensor<[4096,64]>,
Tensor<[4096,64]>,
|ttnn.multiply|aten::mul.Tensor|4| +|329|Tensor<[1,256,64]>,
Tensor<[1,256,64]>,
|ttnn.multiply|aten::mul.Tensor|4| +|330|Tensor<[256,64]>,
Tensor<[256,64]>,
|ttnn.multiply|aten::mul.Tensor|4| +|331|Tensor<[4096,256]>,
Tensor<[4096,256]>,
|ttnn.multiply|aten::mul.Tensor|4| +|332|Tensor<[1,1024,160]>,
Tensor<[1,1024,160]>,
|ttnn.multiply|aten::mul.Tensor|4| +|333|Tensor<[1024,160]>,
Tensor<[1024,160]>,
|ttnn.multiply|aten::mul.Tensor|4| +|334|Tensor<[1,256,160]>,
Tensor<[1,256,160]>,
|ttnn.multiply|aten::mul.Tensor|4| +|335|Tensor<[256,160]>,
Tensor<[256,160]>,
|ttnn.multiply|aten::mul.Tensor|4| +|336|Tensor<[256,1024]>,
Tensor<[256,1024]>,
|ttnn.multiply|aten::mul.Tensor|4| +|337|Tensor<[1,256,128,128]>,
Tensor<[1,256,128,128]>,
|ttnn.multiply|aten::mul.Tensor|4| +|338|Tensor<[1,7,18176]>,
Tensor<[1,7,18176]>,
|ttnn.multiply|aten::gelu|4| +|339|Tensor<[1,71,7,64]>,
Tensor<[1,71,7,64]>,
|ttnn.multiply|aten::mul.Scalar|4| +|340|Tensor<[1,1,64,7]>,
Tensor<[1,1,64,7]>,
|ttnn.multiply|aten::mul.Scalar|4| +|341|Tensor<[7,7]>,
Tensor<[7,7]>,
|ttnn.multiply|aten::mul.Tensor|5| +|342|Tensor<[1,7,64]>,
Tensor<[1,7,64]>,
|ttnn.multiply|aten::mul.Tensor|4| +|343|Tensor<[1,7,4544]>,
Tensor<[1,7,4544]>,
|ttnn.multiply|aten::mul.Tensor|4| +|344|Tensor<[1,1,7,64]>,
Tensor<[1,1,7,64]>,
|ttnn.multiply|aten::mul.Tensor|5| +|345|Tensor<[1,16,112,112]>,
Tensor<[1,16,112,112]>,
|ttnn.multiply|aten::mul.Tensor|4| +|346|Tensor<[96]>,
Tensor<[96]>,
|ttnn.multiply|aten::mul.Tensor|4| +|347|Tensor<[1,96,112,112]>,
Tensor<[1,96,112,112]>,
|ttnn.multiply|aten::mul.Tensor|4| +|348|Tensor<[1,96,56,56]>,
Tensor<[1,96,56,56]>,
|ttnn.multiply|aten::mul.Tensor|4| +|349|Tensor<[144]>,
Tensor<[144]>,
|ttnn.multiply|aten::mul.Tensor|4| +|350|Tensor<[1,144,56,56]>,
Tensor<[1,144,56,56]>,
|ttnn.multiply|aten::mul.Tensor|4| +|351|Tensor<[1,144,28,28]>,
Tensor<[1,144,28,28]>,
|ttnn.multiply|aten::mul.Tensor|4| +|352|Tensor<[1,32,28,28]>,
Tensor<[1,32,28,28]>,
|ttnn.multiply|aten::mul.Tensor|4| +|353|Tensor<[1,192,28,28]>,
Tensor<[1,192,28,28]>,
|ttnn.multiply|aten::mul.Tensor|4| +|354|Tensor<[1,192,14,14]>,
Tensor<[1,192,14,14]>,
|ttnn.multiply|aten::mul.Tensor|4| +|355|Tensor<[1,64,14,14]>,
Tensor<[1,64,14,14]>,
|ttnn.multiply|aten::mul.Tensor|4| +|356|Tensor<[384]>,
Tensor<[384]>,
|ttnn.multiply|aten::mul.Tensor|4| +|357|Tensor<[1,384,14,14]>,
Tensor<[1,384,14,14]>,
|ttnn.multiply|aten::mul.Tensor|4| +|358|Tensor<[1,96,14,14]>,
Tensor<[1,96,14,14]>,
|ttnn.multiply|aten::mul.Tensor|4| +|359|Tensor<[576]>,
Tensor<[576]>,
|ttnn.multiply|aten::mul.Tensor|4| +|360|Tensor<[1,576,14,14]>,
Tensor<[1,576,14,14]>,
|ttnn.multiply|aten::mul.Tensor|4| +|361|Tensor<[1,576,7,7]>,
Tensor<[1,576,7,7]>,
|ttnn.multiply|aten::mul.Tensor|4| +|362|Tensor<[960]>,
Tensor<[960]>,
|ttnn.multiply|aten::mul.Tensor|4| +|363|Tensor<[1,960,7,7]>,
Tensor<[1,960,7,7]>,
|ttnn.multiply|aten::mul.Tensor|4| +|364|Tensor<[1,320,7,7]>,
Tensor<[1,320,7,7]>,
|ttnn.multiply|aten::mul.Tensor|4| +|365|Tensor<[1,1280,7,7]>,
Tensor<[1,1280,7,7]>,
|ttnn.multiply|aten::mul.Tensor|4| +|366|Tensor<[1,12,12,64]>,
Tensor<[1,12,12,64]>,
|ttnn.multiply|aten::mul.Scalar|4| +|367|Tensor<[1,12,64,12]>,
Tensor<[1,12,64,12]>,
|ttnn.multiply|aten::mul.Scalar|4| +|368|Tensor<[1,12,128]>,
Tensor<[1,12,128]>,
|ttnn.multiply|aten::mul.Tensor|4| +|369|Tensor<[12,768]>,
Tensor<[12,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|370|Tensor<[1,12,768]>,
Tensor<[1,12,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|371|Tensor<[12,3072]>,
Tensor<[12,3072]>,
|ttnn.multiply|aten::mul.Tensor|4| +|372|Tensor<[1,12,3072]>,
Tensor<[1,12,3072]>,
|ttnn.multiply|aten::mul.Tensor|4| +|373|Tensor<[12,2]>,
Tensor<[12,2]>,
|ttnn.multiply|aten::mul.Tensor|4| +|374|Tensor<[1,12,9,64]>,
Tensor<[1,12,9,64]>,
|ttnn.multiply|aten::mul.Scalar|4| +|375|Tensor<[1,12,64,9]>,
Tensor<[1,12,64,9]>,
|ttnn.multiply|aten::mul.Scalar|4| +|376|Tensor<[1,9,128]>,
Tensor<[1,9,128]>,
|ttnn.multiply|aten::mul.Tensor|4| +|377|Tensor<[9,768]>,
Tensor<[9,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|378|Tensor<[1,9,768]>,
Tensor<[1,9,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|379|Tensor<[9,3072]>,
Tensor<[9,3072]>,
|ttnn.multiply|aten::mul.Tensor|4| +|380|Tensor<[1,9,3072]>,
Tensor<[1,9,3072]>,
|ttnn.multiply|aten::mul.Tensor|4| +|381|Tensor<[9,128]>,
Tensor<[9,128]>,
|ttnn.multiply|aten::mul.Tensor|4| +|382|Tensor<[9,30000]>,
Tensor<[9,30000]>,
|ttnn.multiply|aten::mul.Tensor|4| +|383|Tensor<[30000]>,
Tensor<[30000]>,
|ttnn.multiply|aten::mul.Tensor|4| +|384|Tensor<[1,16,9,128]>,
Tensor<[1,16,9,128]>,
|ttnn.multiply|aten::mul.Scalar|4| +|385|Tensor<[1,16,128,9]>,
Tensor<[1,16,128,9]>,
|ttnn.multiply|aten::mul.Scalar|4| +|386|Tensor<[9,2048]>,
Tensor<[9,2048]>,
|ttnn.multiply|aten::mul.Tensor|4| +|387|Tensor<[1,9,2048]>,
Tensor<[1,9,2048]>,
|ttnn.multiply|aten::mul.Tensor|4| +|388|Tensor<[9,8192]>,
Tensor<[9,8192]>,
|ttnn.multiply|aten::mul.Tensor|4| +|389|Tensor<[8192]>,
Tensor<[8192]>,
|ttnn.multiply|aten::mul.Tensor|4| +|390|Tensor<[1,9,8192]>,
Tensor<[1,9,8192]>,
|ttnn.multiply|aten::mul.Tensor|4| +|391|Tensor<[1,16,9,64]>,
Tensor<[1,16,9,64]>,
|ttnn.multiply|aten::mul.Scalar|4| +|392|Tensor<[1,16,64,9]>,
Tensor<[1,16,64,9]>,
|ttnn.multiply|aten::mul.Scalar|4| +|393|Tensor<[9,1024]>,
Tensor<[9,1024]>,
|ttnn.multiply|aten::mul.Tensor|4| +|394|Tensor<[1,9,1024]>,
Tensor<[1,9,1024]>,
|ttnn.multiply|aten::mul.Tensor|4| +|395|Tensor<[9,4096]>,
Tensor<[9,4096]>,
|ttnn.multiply|aten::mul.Tensor|4| +|396|Tensor<[1,9,4096]>,
Tensor<[1,9,4096]>,
|ttnn.multiply|aten::mul.Tensor|4| +|397|Tensor<[1,64,9,64]>,
Tensor<[1,64,9,64]>,
|ttnn.multiply|aten::mul.Scalar|4| +|398|Tensor<[1,64,64,9]>,
Tensor<[1,64,64,9]>,
|ttnn.multiply|aten::mul.Scalar|4| +|399|Tensor<[9,16384]>,
Tensor<[9,16384]>,
|ttnn.multiply|aten::mul.Tensor|4| +|400|Tensor<[16384]>,
Tensor<[16384]>,
|ttnn.multiply|aten::mul.Tensor|4| +|401|Tensor<[1,9,16384]>,
Tensor<[1,9,16384]>,
|ttnn.multiply|aten::mul.Tensor|4| +|402|Tensor<[1,2]>,
Tensor<[1,2]>,
|ttnn.multiply|aten::mul.Tensor|4| +|403|Tensor<[1,12,14,64]>,
Tensor<[1,12,14,64]>,
|ttnn.multiply|aten::mul.Scalar|4| +|404|Tensor<[1,12,64,14]>,
Tensor<[1,12,64,14]>,
|ttnn.multiply|aten::mul.Scalar|4| +|405|Tensor<[1,14,128]>,
Tensor<[1,14,128]>,
|ttnn.multiply|aten::mul.Tensor|4| +|406|Tensor<[14,768]>,
Tensor<[14,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|407|Tensor<[1,14,768]>,
Tensor<[1,14,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|408|Tensor<[14,3072]>,
Tensor<[14,3072]>,
|ttnn.multiply|aten::mul.Tensor|4| +|409|Tensor<[1,14,3072]>,
Tensor<[1,14,3072]>,
|ttnn.multiply|aten::mul.Tensor|4| +|410|Tensor<[14,2]>,
Tensor<[14,2]>,
|ttnn.multiply|aten::mul.Tensor|4| +|411|Tensor<[1,12,50,64]>,
Tensor<[1,12,50,64]>,
|ttnn.multiply|aten::mul.Scalar|4| +|412|Tensor<[1,12,64,50]>,
Tensor<[1,12,64,50]>,
|ttnn.multiply|aten::mul.Scalar|4| +|413|Tensor<[2,8,7,64]>,
Tensor<[2,8,7,64]>,
|ttnn.multiply|aten::mul.Scalar|4| +|414|Tensor<[2,8,64,7]>,
Tensor<[2,8,64,7]>,
|ttnn.multiply|aten::mul.Scalar|4| +|415|Tensor<[1,50,768]>,
Tensor<[1,50,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|416|Tensor<[50,768]>,
Tensor<[50,768]>,
|ttnn.multiply|aten::mul.Tensor|4| +|417|Tensor<[50,3072]>,
Tensor<[50,3072]>,
|ttnn.multiply|aten::mul.Tensor|4| +|418|Tensor<[1,50,3072]>,
Tensor<[1,50,3072]>,
|ttnn.multiply|aten::mul.Tensor|4| +|419|Tensor<[2,7,512]>,
Tensor<[2,7,512]>,
|ttnn.multiply|aten::mul.Tensor|4| +|420|Tensor<[14,512]>,
Tensor<[14,512]>,
|ttnn.multiply|aten::mul.Tensor|4| +|421|Tensor<[14,2048]>,
Tensor<[14,2048]>,
|ttnn.multiply|aten::mul.Tensor|4| +|422|Tensor<[2,7,2048]>,
Tensor<[2,7,2048]>,
|ttnn.multiply|aten::mul.Tensor|4| +|423|Tensor<[2,1]>,
Tensor<[2,1]>,
|ttnn.multiply|aten::mul.Tensor|4| +|424|Tensor<[27]>,
Tensor<[27]>,
|ttnn.multiply|aten::arange|4| +|425|Tensor<[197]>,
Tensor<[197]>,
|ttnn.multiply|aten::arange|4| +|426|Tensor<[1,197,4096]>,
Tensor<[1,197,4096]>,
|ttnn.multiply|aten::gelu|4| +|427|Tensor<[1,197,1024]>,
Tensor<[1,197,1024]>,
|ttnn.multiply|aten::mul.Tensor|4| +|428|Tensor<[197,1024]>,
Tensor<[197,1024]>,
|ttnn.multiply|aten::mul.Tensor|4| +|429|Tensor<[1,16,27,27]>,
Tensor<[1,16,27,27]>,
|ttnn.multiply|aten::mul.Tensor|4| +|430|Tensor<[196,196]>,
Tensor<[196,196]>,
|ttnn.multiply|aten::mul.Tensor|4| +|431|Tensor<[197,4096]>,
Tensor<[197,4096]>,
|ttnn.multiply|aten::mul.Tensor|4| +|432|Tensor<[1,1024]>,
Tensor<[1,1024]>,
|ttnn.multiply|aten::mul.Tensor|4| +|433|Tensor<[1,12,27,27]>,
Tensor<[1,12,27,27]>,
|ttnn.multiply|aten::mul.Tensor|4| +|434|Tensor<[1,64]>,
Tensor<[1,64]>,
|ttnn.multiply|aten::mul.Tensor|4| +|435|Tensor<[1,12]>,
Tensor<[1,12]>,
|ttnn.multiply|aten::mul.Tensor|4| +|436|Tensor<[1,784]>,
Tensor<[1,784]>,
|ttnn.multiply|aten::mul.Tensor|4| +|437|Tensor<[784]>,
Tensor<[784]>,
|ttnn.multiply|aten::mul.Tensor|4| diff --git a/docs/ops/stablehlo/stablehlo.negate.md b/docs/ops/stablehlo/stablehlo.negate.md new file mode 100644 index 00000000..80ce34c5 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.negate.md @@ -0,0 +1,10 @@ + +### stablehlo.negate::ttnn.neg + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,32,32,64]>,
|ttnn.neg|aten::neg|5| +|1|Tensor<[19]>,
|ttnn.neg|aten::neg|5| +|2|Tensor<[1,71,7,32]>,
|ttnn.neg|aten::neg|5| +|3|Tensor<[1,1,7,32]>,
|ttnn.neg|aten::neg|5| diff --git a/docs/ops/stablehlo/stablehlo.not.md b/docs/ops/stablehlo/stablehlo.not.md new file mode 100644 index 00000000..9009f977 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.not.md @@ -0,0 +1,7 @@ + +### stablehlo.not::ttnn.not + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,23,40]>,
|ttnn.not|aten::bitwise_not|5| diff --git a/docs/ops/stablehlo/stablehlo.power.md b/docs/ops/stablehlo/stablehlo.power.md new file mode 100644 index 00000000..f0e90b03 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.power.md @@ -0,0 +1,21 @@ + +### stablehlo.power::ttnn.pow + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,32,4096]>,
Tensor<[1,32,4096]>,
|ttnn.pow|aten::pow.Tensor_Scalar|4| +|1|Tensor<[1,7,3072]>,
Tensor<[1,7,3072]>,
|ttnn.pow|aten::pow.Tensor_Scalar|4| +|2|Tensor<[128]>,
Tensor<[128]>,
|ttnn.pow|aten::pow.Scalar|4| +|3|Tensor<[16]>,
Tensor<[16]>,
|ttnn.pow|aten::pow.Tensor_Tensor|4| +|4|Tensor<[1,12,3072]>,
Tensor<[1,12,3072]>,
|ttnn.pow|aten::pow.Tensor_Scalar|4| +|5|Tensor<[1,9,3072]>,
Tensor<[1,9,3072]>,
|ttnn.pow|aten::pow.Tensor_Scalar|4| +|6|Tensor<[1,9,128]>,
Tensor<[1,9,128]>,
|ttnn.pow|aten::pow.Tensor_Scalar|4| +|7|Tensor<[1,9,8192]>,
Tensor<[1,9,8192]>,
|ttnn.pow|aten::pow.Tensor_Scalar|4| +|8|Tensor<[1,9,4096]>,
Tensor<[1,9,4096]>,
|ttnn.pow|aten::pow.Tensor_Scalar|4| +|9|Tensor<[1,9,16384]>,
Tensor<[1,9,16384]>,
|ttnn.pow|aten::pow.Tensor_Scalar|4| +|10|Tensor<[1,14,3072]>,
Tensor<[1,14,3072]>,
|ttnn.pow|aten::pow.Tensor_Scalar|4| +|11|Tensor<[1,512]>,
Tensor<[1,512]>,
|ttnn.pow|aten::pow.Tensor_Scalar|4| +|12|Tensor<[1,1]>,
Tensor<[1,1]>,
|ttnn.pow|aten::pow.Tensor_Scalar|4| +|13|Tensor<[2,512]>,
Tensor<[2,512]>,
|ttnn.pow|aten::pow.Tensor_Scalar|4| +|14|Tensor<[2,1]>,
Tensor<[2,1]>,
|ttnn.pow|aten::pow.Tensor_Scalar|4| diff --git a/docs/ops/stablehlo/stablehlo.reduce_stablehlo.add.md b/docs/ops/stablehlo/stablehlo.reduce_stablehlo.add.md new file mode 100644 index 00000000..6199412d --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.reduce_stablehlo.add.md @@ -0,0 +1,62 @@ + +### stablehlo.reduce_stablehlo.add::ttnn.sum + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,32,32,32]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|1|Tensor<[1,32,4096]>,
Scalar,
dim: [2]
|ttnn.sum|aten::mean.dim|4| +|2|Tensor<[1,12,7,7]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|3|Tensor<[1,512,256]>,
Scalar,
dim: [2]
|ttnn.sum|aten::mean.dim|4| +|4|Tensor<[8,920,920]>,
Scalar,
dim: [2]
|ttnn.sum|aten::_softmax|4| +|5|Tensor<[8,100,100]>,
Scalar,
dim: [2]
|ttnn.sum|aten::_softmax|4| +|6|Tensor<[8,100,920]>,
Scalar,
dim: [2]
|ttnn.sum|aten::_softmax|4| +|7|Tensor<[1,12,10,10]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|8|Tensor<[1,8,4096,4096]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|9|Tensor<[1,8,4096,9]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|10|Tensor<[1,8,1024,1024]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|11|Tensor<[1,8,1024,9]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|12|Tensor<[1,8,256,256]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|13|Tensor<[1,8,256,9]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|14|Tensor<[1,8,64,64]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|15|Tensor<[1,8,64,9]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|16|Tensor<[1,12,25,25]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|17|Tensor<[1,3,1445,1445]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|18|Tensor<[1,512,7,7]>,
Scalar,
dim: [2,
|ttnn.sum|aten::mean.dim|4| +|19|Tensor<[1,12,8,8]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_softmax|4| +|20|Tensor<[1,8,256,2048]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_softmax|4| +|21|Tensor<[1,8,2048,256]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_softmax|4| +|22|Tensor<[1,2048,7,7]>,
Scalar,
dim: [2,
|ttnn.sum|aten::mean.dim|4| +|23|Tensor<[1,12,201,201]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_softmax|4| +|24|Tensor<[1,12,16]>,
Scalar,
dim: [1]
|ttnn.sum|aten::sum.dim_IntList|4| +|25|Tensor<[1,12,16]>,
Scalar,
dim: [2]
|ttnn.sum|aten::sum.dim_IntList|4| +|26|Tensor<[1,10]>,
Scalar,
dim: [1]
|ttnn.sum|aten::sum.dim_IntList|5| +|27|Tensor<[16,19,19]>,
Scalar,
dim: [2]
|ttnn.sum|aten::_softmax|4| +|28|Tensor<[19]>,
Scalar,
dim: [0]
|ttnn.sum|aten::sum|4| +|29|Tensor<[19,256008]>,
Scalar,
dim: [1]
|ttnn.sum|aten::sum.dim_IntList|5| +|30|Tensor<[1,1024,7,7]>,
Scalar,
dim: [2,
|ttnn.sum|aten::mean.dim|4| +|31|Tensor<[1,16,32,32]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_softmax|4| +|32|Tensor<[1,12,16,16]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|33|Tensor<[1,1,19200,300]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_softmax|4| +|34|Tensor<[1,2,4800,300]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_softmax|4| +|35|Tensor<[1,5,1200,300]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_softmax|4| +|36|Tensor<[1,8,300,300]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_softmax|4| +|37|Tensor<[1,12,197,197]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|38|Tensor<[1,1,16384,256]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_softmax|4| +|39|Tensor<[1,2,4096,256]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_softmax|4| +|40|Tensor<[1,5,1024,256]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_softmax|4| +|41|Tensor<[1,71,7,7]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|42|Tensor<[1,1280,7,7]>,
Scalar,
dim: [2,
|ttnn.sum|aten::mean.dim|4| +|43|Tensor<[1,12,12,12]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|44|Tensor<[1,12,9,9]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|45|Tensor<[1,16,9,9]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|46|Tensor<[1,64,9,9]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|47|Tensor<[1,12,14,14]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|48|Tensor<[1,12,50,50]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|49|Tensor<[2,8,7,7]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_safe_softmax|4| +|50|Tensor<[1,512]>,
Scalar,
dim: [1]
|ttnn.sum|aten::sum.dim_IntList|4| +|51|Tensor<[2,512]>,
Scalar,
dim: [1]
|ttnn.sum|aten::sum.dim_IntList|4| +|52|Tensor<[1,16,197,197]>,
Scalar,
dim: [3]
|ttnn.sum|aten::_softmax|4| +|53|Tensor<[1,196,1024]>,
Scalar,
dim: [1]
|ttnn.sum|aten::mean.dim|4| +|54|Tensor<[196,196,2]>,
Scalar,
dim: [2]
|ttnn.sum|aten::sum.dim_IntList|4| +|55|Tensor<[1,196,768]>,
Scalar,
dim: [1]
|ttnn.sum|aten::mean.dim|4| diff --git a/docs/ops/stablehlo/stablehlo.reduce_stablehlo.and.md b/docs/ops/stablehlo/stablehlo.reduce_stablehlo.and.md new file mode 100644 index 00000000..ed5b9ba0 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.reduce_stablehlo.and.md @@ -0,0 +1,29 @@ + +### stablehlo.reduce_stablehlo.and::ttnn.? + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,32,32,32]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|1|Tensor<[1,12,7,7]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|2|Tensor<[1,12,10,10]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|3|Tensor<[1,8,4096,4096]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|4|Tensor<[1,8,4096,9]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|5|Tensor<[1,8,1024,1024]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|6|Tensor<[1,8,1024,9]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|7|Tensor<[1,8,256,256]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|8|Tensor<[1,8,256,9]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|9|Tensor<[1,8,64,64]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|10|Tensor<[1,8,64,9]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|11|Tensor<[1,12,25,25]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|12|Tensor<[1,3,1445,1445]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|13|Tensor<[1,12,16,16]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|14|Tensor<[1,12,197,197]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|15|Tensor<[1,71,7,7]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|16|Tensor<[1,12,12,12]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|17|Tensor<[1,12,9,9]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|18|Tensor<[1,16,9,9]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|19|Tensor<[1,64,9,9]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|20|Tensor<[1,12,14,14]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|21|Tensor<[1,12,50,50]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| +|22|Tensor<[2,8,7,7]>,
Scalar,
dim: [3]
|ttnn.?|aten::_safe_softmax|4| diff --git a/docs/ops/stablehlo/stablehlo.reduce_stablehlo.maximum.md b/docs/ops/stablehlo/stablehlo.reduce_stablehlo.maximum.md new file mode 100644 index 00000000..bf74d9db --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.reduce_stablehlo.maximum.md @@ -0,0 +1,49 @@ + +### stablehlo.reduce_stablehlo.maximum::ttnn.max + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,32,32,32]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|1|Tensor<[1,12,7,7]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|2|Tensor<[8,920,920]>,
Scalar,
dim: [2]
|ttnn.max|aten::_softmax|4| +|3|Tensor<[8,100,100]>,
Scalar,
dim: [2]
|ttnn.max|aten::_softmax|4| +|4|Tensor<[8,100,920]>,
Scalar,
dim: [2]
|ttnn.max|aten::_softmax|4| +|5|Tensor<[1,12,10,10]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|6|Tensor<[1,8,4096,4096]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|7|Tensor<[1,8,4096,9]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|8|Tensor<[1,8,1024,1024]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|9|Tensor<[1,8,1024,9]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|10|Tensor<[1,8,256,256]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|11|Tensor<[1,8,256,9]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|12|Tensor<[1,8,64,64]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|13|Tensor<[1,8,64,9]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|14|Tensor<[1,12,25,25]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|15|Tensor<[1,3,1445,1445]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|16|Tensor<[1,12,8,8]>,
Scalar,
dim: [3]
|ttnn.max|aten::_softmax|4| +|17|Tensor<[1,8,256,2048]>,
Scalar,
dim: [3]
|ttnn.max|aten::_softmax|4| +|18|Tensor<[1,8,2048,256]>,
Scalar,
dim: [3]
|ttnn.max|aten::_softmax|4| +|19|Tensor<[1,12,201,201]>,
Scalar,
dim: [3]
|ttnn.max|aten::_softmax|4| +|20|Tensor<[1]>,
Scalar,
dim: [0]
|ttnn.max|aten::max|4| +|21|Tensor<[1,10]>,
Scalar,
dim: [1]
|ttnn.max|aten::amax|5| +|22|Tensor<[16,19,19]>,
Scalar,
dim: [2]
|ttnn.max|aten::_softmax|4| +|23|Tensor<[19,256008]>,
Scalar,
dim: [1]
|ttnn.max|aten::amax|5| +|24|Tensor<[1,16,32,32]>,
Scalar,
dim: [3]
|ttnn.max|aten::_softmax|4| +|25|Tensor<[1,12,16,16]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|26|Tensor<[1,1,19200,300]>,
Scalar,
dim: [3]
|ttnn.max|aten::_softmax|4| +|27|Tensor<[1,2,4800,300]>,
Scalar,
dim: [3]
|ttnn.max|aten::_softmax|4| +|28|Tensor<[1,5,1200,300]>,
Scalar,
dim: [3]
|ttnn.max|aten::_softmax|4| +|29|Tensor<[1,8,300,300]>,
Scalar,
dim: [3]
|ttnn.max|aten::_softmax|4| +|30|Tensor<[1,12,197,197]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|31|Tensor<[1,1,16384,256]>,
Scalar,
dim: [3]
|ttnn.max|aten::_softmax|4| +|32|Tensor<[1,2,4096,256]>,
Scalar,
dim: [3]
|ttnn.max|aten::_softmax|4| +|33|Tensor<[1,5,1024,256]>,
Scalar,
dim: [3]
|ttnn.max|aten::_softmax|4| +|34|Tensor<[1,71,7,7]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|35|Tensor<[1,12,12,12]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|36|Tensor<[1,12,9,9]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|37|Tensor<[1,16,9,9]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|38|Tensor<[1,64,9,9]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|39|Tensor<[1,12,14,14]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|40|Tensor<[1,12,50,50]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|41|Tensor<[2,8,7,7]>,
Scalar,
dim: [3]
|ttnn.max|aten::_safe_softmax|4| +|42|Tensor<[1,16,197,197]>,
Scalar,
dim: [3]
|ttnn.max|aten::_softmax|4| diff --git a/docs/ops/stablehlo/stablehlo.reduce_window_stablehlo.add.md b/docs/ops/stablehlo/stablehlo.reduce_window_stablehlo.add.md new file mode 100644 index 00000000..6e17bf81 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.reduce_window_stablehlo.add.md @@ -0,0 +1,9 @@ + +### stablehlo.reduce_window_stablehlo.add::ttnn.avg_pool2d + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,23,40]>,
Scalar,
|ttnn.avg_pool2d|aten::cumsum|4| +|1|Tensor<[1,10]>,
Scalar,
|ttnn.avg_pool2d|aten::cumsum|4| +|2|Tensor<[1,32]>,
Scalar,
|ttnn.avg_pool2d|aten::cumsum|4| diff --git a/docs/ops/stablehlo/stablehlo.remainder.md b/docs/ops/stablehlo/stablehlo.remainder.md new file mode 100644 index 00000000..36f717ba --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.remainder.md @@ -0,0 +1,7 @@ + +### stablehlo.remainder::ttnn.remainder + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1]>,
Tensor<[1]>,
|ttnn.remainder|aten::remainder.Scalar|4| diff --git a/docs/ops/stablehlo/stablehlo.reshape.md b/docs/ops/stablehlo/stablehlo.reshape.md new file mode 100644 index 00000000..92798671 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.reshape.md @@ -0,0 +1,896 @@ + +### stablehlo.reshape::ttnn.reshape + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,32,32]>,
Tensor<[1,32,32,1]>,
|ttnn.reshape|aten::_safe_softmax|4| +|1|Tensor<[1]>,
Scalar,
|ttnn.reshape|aten::_safe_softmax|4| +|2|Tensor<[1,64,32]>,
Tensor<[1,64,32]>,
|ttnn.reshape|aten::_unsafe_view|5| +|3|Tensor<[32,4096]>,
Tensor<[1,32,4096]>,
|ttnn.reshape|aten::_unsafe_view|5| +|4|Tensor<[32,11008]>,
Tensor<[1,32,11008]>,
|ttnn.reshape|aten::_unsafe_view|5| +|5|Tensor<[32,32000]>,
Tensor<[1,32,32000]>,
|ttnn.reshape|aten::_unsafe_view|5| +|6|Scalar,
Tensor<[1]>,
|ttnn.reshape|aten::arange|4| +|7|Tensor<[1,32]>,
Tensor<[1,32,1]>,
|ttnn.reshape|aten::mean.dim|4| +|8|Tensor<[32]>,
Tensor<[32,1]>,
|ttnn.reshape|aten::triu|4| +|9|Tensor<[32]>,
Tensor<[1,32]>,
|ttnn.reshape|aten::triu|4| +|10|Tensor<[32,32]>,
Tensor<[1,32,32]>,
|ttnn.reshape|aten::unsqueeze|5| +|11|Tensor<[1,32,32]>,
Tensor<[1,1,32,32]>,
|ttnn.reshape|aten::unsqueeze|5| +|12|Tensor<[1,32]>,
Tensor<[1,1,32]>,
|ttnn.reshape|aten::unsqueeze|4| +|13|Tensor<[1,1,32]>,
Tensor<[1,1,1,32]>,
|ttnn.reshape|aten::unsqueeze|4| +|14|Tensor<[64]>,
Tensor<[1,64]>,
|ttnn.reshape|aten::unsqueeze|5| +|15|Tensor<[1,64]>,
Tensor<[1,64,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|16|Tensor<[1,32,128]>,
Tensor<[1,1,32,128]>,
|ttnn.reshape|aten::unsqueeze|5| +|17|Tensor<[1,64,1]>,
Tensor<[1,64,1]>,
|ttnn.reshape|aten::view|5| +|18|Tensor<[1,1,32]>,
Tensor<[1,1,32]>,
|ttnn.reshape|aten::view|5| +|19|Tensor<[1,32,4096]>,
Tensor<[32,4096]>,
|ttnn.reshape|aten::view|5| +|20|Tensor<[1,32,4096]>,
Tensor<[1,32,32,128]>,
|ttnn.reshape|aten::view|5| +|21|Tensor<[1,32,32,128]>,
Tensor<[32,32,128]>,
|ttnn.reshape|aten::view|5| +|22|Tensor<[1,32,128,32]>,
Tensor<[32,128,32]>,
|ttnn.reshape|aten::view|5| +|23|Tensor<[32,32,32]>,
Tensor<[1,32,32,32]>,
|ttnn.reshape|aten::view|5| +|24|Tensor<[1,32,32,32]>,
Tensor<[32,32,32]>,
|ttnn.reshape|aten::view|5| +|25|Tensor<[32,32,128]>,
Tensor<[1,32,32,128]>,
|ttnn.reshape|aten::view|5| +|26|Tensor<[1,32,32,128]>,
Tensor<[1,32,4096]>,
|ttnn.reshape|aten::view|5| +|27|Tensor<[1,32,11008]>,
Tensor<[32,11008]>,
|ttnn.reshape|aten::view|5| +|28|Tensor<[1,12,7]>,
Tensor<[1,12,7,1]>,
|ttnn.reshape|aten::_safe_softmax|4| +|29|Tensor<[7,2]>,
Tensor<[1,7,2]>,
|ttnn.reshape|aten::_unsafe_view|5| +|30|Tensor<[1]>,
Tensor<[1,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|31|Tensor<[7]>,
Tensor<[1,7]>,
|ttnn.reshape|aten::unsqueeze|4| +|32|Tensor<[7,7]>,
Tensor<[1,7,7]>,
|ttnn.reshape|aten::unsqueeze|5| +|33|Tensor<[1,7,7]>,
Tensor<[1,1,7,7]>,
|ttnn.reshape|aten::unsqueeze|5| +|34|Tensor<[1,7]>,
Tensor<[1,1,7]>,
|ttnn.reshape|aten::unsqueeze|4| +|35|Tensor<[1,1,7]>,
Tensor<[1,1,1,7]>,
|ttnn.reshape|aten::unsqueeze|4| +|36|Tensor<[1,7]>,
Tensor<[1,7]>,
|ttnn.reshape|aten::view|4| +|37|Tensor<[7]>,
Tensor<[7,1]>,
|ttnn.reshape|aten::view|4| +|38|Tensor<[1,7,768]>,
Tensor<[7,768]>,
|ttnn.reshape|aten::view|5| +|39|Tensor<[7,2304]>,
Tensor<[1,7,2304]>,
|ttnn.reshape|aten::view|5| +|40|Tensor<[1,7,768]>,
Tensor<[1,7,12,64]>,
|ttnn.reshape|aten::view|5| +|41|Tensor<[1,12,7,64]>,
Tensor<[12,7,64]>,
|ttnn.reshape|aten::view|5| +|42|Tensor<[1,12,64,7]>,
Tensor<[12,64,7]>,
|ttnn.reshape|aten::view|5| +|43|Tensor<[12,7,7]>,
Tensor<[1,12,7,7]>,
|ttnn.reshape|aten::view|5| +|44|Tensor<[1,12,7,7]>,
Tensor<[12,7,7]>,
|ttnn.reshape|aten::view|5| +|45|Tensor<[12,7,64]>,
Tensor<[1,12,7,64]>,
|ttnn.reshape|aten::view|5| +|46|Tensor<[1,7,12,64]>,
Tensor<[1,7,768]>,
|ttnn.reshape|aten::view|5| +|47|Tensor<[7,768]>,
Tensor<[1,7,768]>,
|ttnn.reshape|aten::view|5| +|48|Tensor<[7,3072]>,
Tensor<[1,7,3072]>,
|ttnn.reshape|aten::view|5| +|49|Tensor<[1,7,3072]>,
Tensor<[7,3072]>,
|ttnn.reshape|aten::view|5| +|50|Tensor<[1,7,768]>,
Tensor<[1,7,768]>,
|ttnn.reshape|aten::view|5| +|51|Tensor<[128]>,
Tensor<[128,1,1]>,
|ttnn.reshape|aten::convolution|4| +|52|Tensor<[512]>,
Tensor<[512,1,1]>,
|ttnn.reshape|aten::convolution|4| +|53|Tensor<[19]>,
Tensor<[19,1,1]>,
|ttnn.reshape|aten::convolution|4| +|54|Tensor<[38]>,
Tensor<[38,1,1]>,
|ttnn.reshape|aten::convolution|4| +|55|Tensor<[32,1]>,
Tensor<[32,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|56|Tensor<[64]>,
Tensor<[64,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|57|Tensor<[64,1]>,
Tensor<[64,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|58|Tensor<[128]>,
Tensor<[128,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|59|Tensor<[128,1]>,
Tensor<[128,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|60|Tensor<[256]>,
Tensor<[256,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|61|Tensor<[256,1]>,
Tensor<[256,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|62|Tensor<[512]>,
Tensor<[512,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|63|Tensor<[512,1]>,
Tensor<[512,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|64|Tensor<[1,16,16,16,16,3]>,
Tensor<[1,256,768]>,
|ttnn.reshape|aten::_unsafe_view|5| +|65|Tensor<[1024]>,
Tensor<[1024,1]>,
|ttnn.reshape|aten::convolution|4| +|66|Tensor<[1,3,256,256]>,
Tensor<[1,3,16,16,16,16]>,
|ttnn.reshape|aten::view|4| +|67|Tensor<[1,256,768]>,
Tensor<[256,768]>,
|ttnn.reshape|aten::view|5| +|68|Tensor<[256,512]>,
Tensor<[1,256,512]>,
|ttnn.reshape|aten::view|5| +|69|Tensor<[1,256,512]>,
Tensor<[256,512]>,
|ttnn.reshape|aten::view|5| +|70|Tensor<[256,256]>,
Tensor<[1,256,256]>,
|ttnn.reshape|aten::view|5| +|71|Tensor<[1,256,256]>,
Tensor<[256,256]>,
|ttnn.reshape|aten::view|5| +|72|Tensor<[8,920]>,
Tensor<[8,920,1]>,
|ttnn.reshape|aten::_softmax|4| +|73|Tensor<[8,100]>,
Tensor<[8,100,1]>,
|ttnn.reshape|aten::_softmax|4| +|74|Tensor<[920,1,256]>,
Tensor<[920,1,256]>,
|ttnn.reshape|aten::_unsafe_view|5| +|75|Tensor<[6,100,92]>,
Tensor<[6,1,100,92]>,
|ttnn.reshape|aten::_unsafe_view|5| +|76|Tensor<[6,1,100,92]>,
Tensor<[6,100,92]>,
|ttnn.reshape|aten::_unsafe_view|5| +|77|Tensor<[6,100,256]>,
Tensor<[6,1,100,256]>,
|ttnn.reshape|aten::_unsafe_view|5| +|78|Tensor<[6,1,100,256]>,
Tensor<[6,100,256]>,
|ttnn.reshape|aten::_unsafe_view|5| +|79|Tensor<[600,256]>,
Tensor<[6,1,100,256]>,
|ttnn.reshape|aten::_unsafe_view|5| +|80|Tensor<[6,1,100,256]>,
Tensor<[600,256]>,
|ttnn.reshape|aten::_unsafe_view|5| +|81|Tensor<[600,4]>,
Tensor<[6,1,100,4]>,
|ttnn.reshape|aten::_unsafe_view|5| +|82|Tensor<[6,1,100,4]>,
Tensor<[600,4]>,
|ttnn.reshape|aten::_unsafe_view|5| +|83|Tensor<[256]>,
Tensor<[256,1,1]>,
|ttnn.reshape|aten::convolution|4| +|84|Tensor<[1,1]>,
Tensor<[1,1,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|85|Tensor<[1,1,1]>,
Tensor<[1,1,1,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|86|Tensor<[1,1,23,40]>,
Tensor<[1,1,23,40,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|87|Tensor<[100,1,256]>,
Tensor<[1,100,1,256]>,
|ttnn.reshape|aten::repeat|4| +|88|Tensor<[1,100,1,256]>,
Tensor<[1,100,1,1,256]>,
|ttnn.reshape|aten::repeat|4| +|89|Tensor<[1,100,1,1,256]>,
Tensor<[1,100,1,1,1,256]>,
|ttnn.reshape|aten::repeat|4| +|90|Tensor<[1,100,1,1,1,256]>,
Tensor<[100,1,1,1,256]>,
|ttnn.reshape|aten::repeat|4| +|91|Tensor<[100,1,1,1,256]>,
Tensor<[100,1,1,256]>,
|ttnn.reshape|aten::repeat|4| +|92|Tensor<[100,1,1,256]>,
Tensor<[100,1,256]>,
|ttnn.reshape|aten::repeat|4| +|93|Tensor<[1,3,720,1280]>,
Tensor<[3,720,1280]>,
|ttnn.reshape|aten::select.int|5| +|94|Tensor<[1,720,1280]>,
Tensor<[720,1280]>,
|ttnn.reshape|aten::select.int|4| +|95|Tensor<[1,1,23,40]>,
Tensor<[1,23,40]>,
|ttnn.reshape|aten::select.int|4| +|96|Tensor<[1,1,100,4]>,
Tensor<[1,100,4]>,
|ttnn.reshape|aten::select.int|4| +|97|Tensor<[1,1,100,92]>,
Tensor<[1,100,92]>,
|ttnn.reshape|aten::select.int|4| +|98|Tensor<[3,720,1280]>,
Tensor<[1,3,720,1280]>,
|ttnn.reshape|aten::select_scatter|4| +|99|Tensor<[720,1280]>,
Tensor<[1,720,1280]>,
|ttnn.reshape|aten::select_scatter|4| +|100|Tensor<[1,23,40,64]>,
Tensor<[1,23,40,64,1]>,
|ttnn.reshape|aten::stack|5| +|101|Tensor<[1,720,1280]>,
Tensor<[1,1,720,1280]>,
|ttnn.reshape|aten::unsqueeze|4| +|102|Tensor<[23]>,
Tensor<[23,1]>,
|ttnn.reshape|aten::unsqueeze|4| +|103|Tensor<[1,23,40]>,
Tensor<[1,23,40,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|104|Tensor<[100,256]>,
Tensor<[100,1,256]>,
|ttnn.reshape|aten::unsqueeze|5| +|105|Tensor<[64]>,
Tensor<[1,64,1,1]>,
|ttnn.reshape|aten::view|5| +|106|Tensor<[256]>,
Tensor<[1,256,1,1]>,
|ttnn.reshape|aten::view|5| +|107|Tensor<[128]>,
Tensor<[1,128,1,1]>,
|ttnn.reshape|aten::view|5| +|108|Tensor<[512]>,
Tensor<[1,512,1,1]>,
|ttnn.reshape|aten::view|5| +|109|Tensor<[1024]>,
Tensor<[1,1024,1,1]>,
|ttnn.reshape|aten::view|5| +|110|Tensor<[2048]>,
Tensor<[1,2048,1,1]>,
|ttnn.reshape|aten::view|5| +|111|Tensor<[1,23,40,64,2]>,
Tensor<[1,23,40,128]>,
|ttnn.reshape|aten::view|5| +|112|Tensor<[1,256,23,40]>,
Tensor<[1,256,920]>,
|ttnn.reshape|aten::view|5| +|113|Tensor<[1,23,40]>,
Tensor<[1,920]>,
|ttnn.reshape|aten::view|4| +|114|Tensor<[920,256,256]>,
Tensor<[920,256,256]>,
|ttnn.reshape|aten::view|5| +|115|Tensor<[920,1,256]>,
Tensor<[920,8,32]>,
|ttnn.reshape|aten::view|5| +|116|Tensor<[1,920]>,
Tensor<[1,1,1,920]>,
|ttnn.reshape|aten::view|5| +|117|Tensor<[1,8,1,920]>,
Tensor<[8,1,920]>,
|ttnn.reshape|aten::view|5| +|118|Tensor<[920,8,32]>,
Tensor<[920,256]>,
|ttnn.reshape|aten::view|5| +|119|Tensor<[920,256]>,
Tensor<[920,1,256]>,
|ttnn.reshape|aten::view|5| +|120|Tensor<[920,1,256]>,
Tensor<[920,256]>,
|ttnn.reshape|aten::view|5| +|121|Tensor<[920,2048]>,
Tensor<[920,1,2048]>,
|ttnn.reshape|aten::view|5| +|122|Tensor<[920,1,2048]>,
Tensor<[920,2048]>,
|ttnn.reshape|aten::view|5| +|123|Tensor<[100,1,256]>,
Tensor<[100,256]>,
|ttnn.reshape|aten::view|5| +|124|Tensor<[100,1,256]>,
Tensor<[100,8,32]>,
|ttnn.reshape|aten::view|5| +|125|Tensor<[100,8,32]>,
Tensor<[100,256]>,
|ttnn.reshape|aten::view|5| +|126|Tensor<[100,2048]>,
Tensor<[100,1,2048]>,
|ttnn.reshape|aten::view|5| +|127|Tensor<[100,1,2048]>,
Tensor<[100,2048]>,
|ttnn.reshape|aten::view|5| +|128|Tensor<[6,1,256,92]>,
Tensor<[6,256,92]>,
|ttnn.reshape|aten::view|5| +|129|Tensor<[6,1,256,256]>,
Tensor<[6,256,256]>,
|ttnn.reshape|aten::view|5| +|130|Tensor<[1,12,10]>,
Tensor<[1,12,10,1]>,
|ttnn.reshape|aten::_safe_softmax|4| +|131|Tensor<[1,10]>,
Tensor<[1,1,10]>,
|ttnn.reshape|aten::unsqueeze|4| +|132|Tensor<[1,1,10]>,
Tensor<[1,1,1,10]>,
|ttnn.reshape|aten::unsqueeze|4| +|133|Tensor<[1,10,768]>,
Tensor<[10,768]>,
|ttnn.reshape|aten::view|5| +|134|Tensor<[10,768]>,
Tensor<[1,10,768]>,
|ttnn.reshape|aten::view|5| +|135|Tensor<[1,10,768]>,
Tensor<[1,10,12,64]>,
|ttnn.reshape|aten::view|5| +|136|Tensor<[1,12,10,64]>,
Tensor<[12,10,64]>,
|ttnn.reshape|aten::view|5| +|137|Tensor<[1,12,64,10]>,
Tensor<[12,64,10]>,
|ttnn.reshape|aten::view|5| +|138|Tensor<[12,10,10]>,
Tensor<[1,12,10,10]>,
|ttnn.reshape|aten::view|5| +|139|Tensor<[1,12,10,10]>,
Tensor<[12,10,10]>,
|ttnn.reshape|aten::view|5| +|140|Tensor<[12,10,64]>,
Tensor<[1,12,10,64]>,
|ttnn.reshape|aten::view|5| +|141|Tensor<[1,10,12,64]>,
Tensor<[1,10,768]>,
|ttnn.reshape|aten::view|5| +|142|Tensor<[10,3072]>,
Tensor<[1,10,3072]>,
|ttnn.reshape|aten::view|5| +|143|Tensor<[1,10,3072]>,
Tensor<[10,3072]>,
|ttnn.reshape|aten::view|5| +|144|Tensor<[10,250002]>,
Tensor<[1,10,250002]>,
|ttnn.reshape|aten::view|5| +|145|Tensor<[1,8,4096]>,
Tensor<[1,8,4096,1]>,
|ttnn.reshape|aten::_safe_softmax|4| +|146|Tensor<[1,8,1024]>,
Tensor<[1,8,1024,1]>,
|ttnn.reshape|aten::_safe_softmax|4| +|147|Tensor<[1,8,256]>,
Tensor<[1,8,256,1]>,
|ttnn.reshape|aten::_safe_softmax|4| +|148|Tensor<[1,8,64]>,
Tensor<[1,8,64,1]>,
|ttnn.reshape|aten::_safe_softmax|4| +|149|Tensor<[4096,320]>,
Tensor<[1,4096,320]>,
|ttnn.reshape|aten::_unsafe_view|5| +|150|Tensor<[9,320]>,
Tensor<[1,9,320]>,
|ttnn.reshape|aten::_unsafe_view|5| +|151|Tensor<[1024,640]>,
Tensor<[1,1024,640]>,
|ttnn.reshape|aten::_unsafe_view|5| +|152|Tensor<[9,640]>,
Tensor<[1,9,640]>,
|ttnn.reshape|aten::_unsafe_view|5| +|153|Tensor<[256,1280]>,
Tensor<[1,256,1280]>,
|ttnn.reshape|aten::_unsafe_view|5| +|154|Tensor<[9,1280]>,
Tensor<[1,9,1280]>,
|ttnn.reshape|aten::_unsafe_view|5| +|155|Tensor<[64,1280]>,
Tensor<[1,64,1280]>,
|ttnn.reshape|aten::_unsafe_view|5| +|156|Tensor<[320]>,
Tensor<[320,1,1]>,
|ttnn.reshape|aten::convolution|4| +|157|Tensor<[640]>,
Tensor<[640,1,1]>,
|ttnn.reshape|aten::convolution|4| +|158|Tensor<[1280]>,
Tensor<[1280,1,1]>,
|ttnn.reshape|aten::convolution|4| +|159|Tensor<[4]>,
Tensor<[4,1,1]>,
|ttnn.reshape|aten::convolution|4| +|160|Tensor<[1280]>,
Tensor<[1280,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|161|Tensor<[1280,1]>,
Tensor<[1280,1,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|162|Tensor<[1,1280,16,16]>,
Tensor<[1,1280,16,16,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|163|Tensor<[1,1280,32,32]>,
Tensor<[1,1280,32,32,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|164|Tensor<[640]>,
Tensor<[640,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|165|Tensor<[640,1]>,
Tensor<[640,1,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|166|Tensor<[1,640,64,64]>,
Tensor<[1,640,64,64,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|167|Tensor<[160]>,
Tensor<[1,160]>,
|ttnn.reshape|aten::unsqueeze|5| +|168|Tensor<[320]>,
Tensor<[1,320]>,
|ttnn.reshape|aten::unsqueeze|5| +|169|Tensor<[1,320]>,
Tensor<[1,320,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|170|Tensor<[1,320,1]>,
Tensor<[1,320,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|171|Tensor<[1,640]>,
Tensor<[1,640,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|172|Tensor<[1,640,1]>,
Tensor<[1,640,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|173|Tensor<[640]>,
Tensor<[1,640]>,
|ttnn.reshape|aten::unsqueeze|5| +|174|Tensor<[1,1280]>,
Tensor<[1,1280,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|175|Tensor<[1,1280,1]>,
Tensor<[1,1280,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|176|Tensor<[1280]>,
Tensor<[1,1280]>,
|ttnn.reshape|aten::unsqueeze|5| +|177|Tensor<[2560]>,
Tensor<[1,2560]>,
|ttnn.reshape|aten::unsqueeze|5| +|178|Tensor<[1,2560]>,
Tensor<[1,2560,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|179|Tensor<[1,2560,1]>,
Tensor<[1,2560,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|180|Tensor<[16]>,
Tensor<[16,1]>,
|ttnn.reshape|aten::unsqueeze|4| +|181|Tensor<[1920]>,
Tensor<[1,1920]>,
|ttnn.reshape|aten::unsqueeze|5| +|182|Tensor<[1,1920]>,
Tensor<[1,1920,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|183|Tensor<[1,1920,1]>,
Tensor<[1,1920,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|184|Tensor<[960]>,
Tensor<[1,960]>,
|ttnn.reshape|aten::unsqueeze|5| +|185|Tensor<[1,960]>,
Tensor<[1,960,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|186|Tensor<[1,960,1]>,
Tensor<[1,960,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|187|Tensor<[1,320,64,64]>,
Tensor<[1,32,10,4096]>,
|ttnn.reshape|aten::view|5| +|188|Tensor<[1,32,10,4096]>,
Tensor<[1,320,64,64]>,
|ttnn.reshape|aten::view|5| +|189|Tensor<[1,64,64,320]>,
Tensor<[1,4096,320]>,
|ttnn.reshape|aten::view|5| +|190|Tensor<[1,4096,320]>,
Tensor<[4096,320]>,
|ttnn.reshape|aten::view|5| +|191|Tensor<[1,4096,320]>,
Tensor<[1,4096,8,40]>,
|ttnn.reshape|aten::view|5| +|192|Tensor<[1,8,4096,40]>,
Tensor<[8,4096,40]>,
|ttnn.reshape|aten::view|5| +|193|Tensor<[1,8,40,4096]>,
Tensor<[8,40,4096]>,
|ttnn.reshape|aten::view|5| +|194|Tensor<[8,4096,4096]>,
Tensor<[1,8,4096,4096]>,
|ttnn.reshape|aten::view|5| +|195|Tensor<[1,8,4096,4096]>,
Tensor<[8,4096,4096]>,
|ttnn.reshape|aten::view|5| +|196|Tensor<[8,4096,40]>,
Tensor<[1,8,4096,40]>,
|ttnn.reshape|aten::view|5| +|197|Tensor<[1,4096,8,40]>,
Tensor<[1,4096,320]>,
|ttnn.reshape|aten::view|5| +|198|Tensor<[1,9,768]>,
Tensor<[9,768]>,
|ttnn.reshape|aten::view|5| +|199|Tensor<[1,9,320]>,
Tensor<[1,9,8,40]>,
|ttnn.reshape|aten::view|5| +|200|Tensor<[1,8,40,9]>,
Tensor<[8,40,9]>,
|ttnn.reshape|aten::view|5| +|201|Tensor<[8,4096,9]>,
Tensor<[1,8,4096,9]>,
|ttnn.reshape|aten::view|5| +|202|Tensor<[1,8,4096,9]>,
Tensor<[8,4096,9]>,
|ttnn.reshape|aten::view|5| +|203|Tensor<[1,8,9,40]>,
Tensor<[8,9,40]>,
|ttnn.reshape|aten::view|5| +|204|Tensor<[4096,2560]>,
Tensor<[1,4096,2560]>,
|ttnn.reshape|aten::view|5| +|205|Tensor<[1,4096,1280]>,
Tensor<[4096,1280]>,
|ttnn.reshape|aten::view|5| +|206|Tensor<[1,4096,320]>,
Tensor<[1,64,64,320]>,
|ttnn.reshape|aten::view|5| +|207|Tensor<[1,320,32,32]>,
Tensor<[1,32,10,1024]>,
|ttnn.reshape|aten::view|5| +|208|Tensor<[1,32,10,1024]>,
Tensor<[1,320,32,32]>,
|ttnn.reshape|aten::view|5| +|209|Tensor<[1,640,32,32]>,
Tensor<[1,32,20,1024]>,
|ttnn.reshape|aten::view|5| +|210|Tensor<[1,32,20,1024]>,
Tensor<[1,640,32,32]>,
|ttnn.reshape|aten::view|5| +|211|Tensor<[1,32,32,640]>,
Tensor<[1,1024,640]>,
|ttnn.reshape|aten::view|5| +|212|Tensor<[1,1024,640]>,
Tensor<[1024,640]>,
|ttnn.reshape|aten::view|5| +|213|Tensor<[1,1024,640]>,
Tensor<[1,1024,8,80]>,
|ttnn.reshape|aten::view|5| +|214|Tensor<[1,8,1024,80]>,
Tensor<[8,1024,80]>,
|ttnn.reshape|aten::view|5| +|215|Tensor<[1,8,80,1024]>,
Tensor<[8,80,1024]>,
|ttnn.reshape|aten::view|5| +|216|Tensor<[8,1024,1024]>,
Tensor<[1,8,1024,1024]>,
|ttnn.reshape|aten::view|5| +|217|Tensor<[1,8,1024,1024]>,
Tensor<[8,1024,1024]>,
|ttnn.reshape|aten::view|5| +|218|Tensor<[8,1024,80]>,
Tensor<[1,8,1024,80]>,
|ttnn.reshape|aten::view|5| +|219|Tensor<[1,1024,8,80]>,
Tensor<[1,1024,640]>,
|ttnn.reshape|aten::view|5| +|220|Tensor<[1,9,640]>,
Tensor<[1,9,8,80]>,
|ttnn.reshape|aten::view|5| +|221|Tensor<[1,8,80,9]>,
Tensor<[8,80,9]>,
|ttnn.reshape|aten::view|5| +|222|Tensor<[8,1024,9]>,
Tensor<[1,8,1024,9]>,
|ttnn.reshape|aten::view|5| +|223|Tensor<[1,8,1024,9]>,
Tensor<[8,1024,9]>,
|ttnn.reshape|aten::view|5| +|224|Tensor<[1,8,9,80]>,
Tensor<[8,9,80]>,
|ttnn.reshape|aten::view|5| +|225|Tensor<[1024,5120]>,
Tensor<[1,1024,5120]>,
|ttnn.reshape|aten::view|5| +|226|Tensor<[1,1024,2560]>,
Tensor<[1024,2560]>,
|ttnn.reshape|aten::view|5| +|227|Tensor<[1,1024,640]>,
Tensor<[1,32,32,640]>,
|ttnn.reshape|aten::view|5| +|228|Tensor<[1,640,16,16]>,
Tensor<[1,32,20,256]>,
|ttnn.reshape|aten::view|5| +|229|Tensor<[1,32,20,256]>,
Tensor<[1,640,16,16]>,
|ttnn.reshape|aten::view|5| +|230|Tensor<[1,1280,16,16]>,
Tensor<[1,32,40,256]>,
|ttnn.reshape|aten::view|5| +|231|Tensor<[1,32,40,256]>,
Tensor<[1,1280,16,16]>,
|ttnn.reshape|aten::view|5| +|232|Tensor<[1,16,16,1280]>,
Tensor<[1,256,1280]>,
|ttnn.reshape|aten::view|5| +|233|Tensor<[1,256,1280]>,
Tensor<[256,1280]>,
|ttnn.reshape|aten::view|5| +|234|Tensor<[1,256,1280]>,
Tensor<[1,256,8,160]>,
|ttnn.reshape|aten::view|5| +|235|Tensor<[1,8,256,160]>,
Tensor<[8,256,160]>,
|ttnn.reshape|aten::view|5| +|236|Tensor<[1,8,160,256]>,
Tensor<[8,160,256]>,
|ttnn.reshape|aten::view|5| +|237|Tensor<[8,256,256]>,
Tensor<[1,8,256,256]>,
|ttnn.reshape|aten::view|5| +|238|Tensor<[1,8,256,256]>,
Tensor<[8,256,256]>,
|ttnn.reshape|aten::view|5| +|239|Tensor<[8,256,160]>,
Tensor<[1,8,256,160]>,
|ttnn.reshape|aten::view|5| +|240|Tensor<[1,256,8,160]>,
Tensor<[1,256,1280]>,
|ttnn.reshape|aten::view|5| +|241|Tensor<[1,9,1280]>,
Tensor<[1,9,8,160]>,
|ttnn.reshape|aten::view|5| +|242|Tensor<[1,8,160,9]>,
Tensor<[8,160,9]>,
|ttnn.reshape|aten::view|5| +|243|Tensor<[8,256,9]>,
Tensor<[1,8,256,9]>,
|ttnn.reshape|aten::view|5| +|244|Tensor<[1,8,256,9]>,
Tensor<[8,256,9]>,
|ttnn.reshape|aten::view|5| +|245|Tensor<[1,8,9,160]>,
Tensor<[8,9,160]>,
|ttnn.reshape|aten::view|5| +|246|Tensor<[256,10240]>,
Tensor<[1,256,10240]>,
|ttnn.reshape|aten::view|5| +|247|Tensor<[1,256,5120]>,
Tensor<[256,5120]>,
|ttnn.reshape|aten::view|5| +|248|Tensor<[1,256,1280]>,
Tensor<[1,16,16,1280]>,
|ttnn.reshape|aten::view|5| +|249|Tensor<[1,1280,8,8]>,
Tensor<[1,32,40,64]>,
|ttnn.reshape|aten::view|5| +|250|Tensor<[1,32,40,64]>,
Tensor<[1,1280,8,8]>,
|ttnn.reshape|aten::view|5| +|251|Tensor<[1,8,8,1280]>,
Tensor<[1,64,1280]>,
|ttnn.reshape|aten::view|5| +|252|Tensor<[1,64,1280]>,
Tensor<[64,1280]>,
|ttnn.reshape|aten::view|5| +|253|Tensor<[1,64,1280]>,
Tensor<[1,64,8,160]>,
|ttnn.reshape|aten::view|5| +|254|Tensor<[1,8,64,160]>,
Tensor<[8,64,160]>,
|ttnn.reshape|aten::view|5| +|255|Tensor<[1,8,160,64]>,
Tensor<[8,160,64]>,
|ttnn.reshape|aten::view|5| +|256|Tensor<[8,64,64]>,
Tensor<[1,8,64,64]>,
|ttnn.reshape|aten::view|5| +|257|Tensor<[1,8,64,64]>,
Tensor<[8,64,64]>,
|ttnn.reshape|aten::view|5| +|258|Tensor<[8,64,160]>,
Tensor<[1,8,64,160]>,
|ttnn.reshape|aten::view|5| +|259|Tensor<[1,64,8,160]>,
Tensor<[1,64,1280]>,
|ttnn.reshape|aten::view|5| +|260|Tensor<[8,64,9]>,
Tensor<[1,8,64,9]>,
|ttnn.reshape|aten::view|5| +|261|Tensor<[1,8,64,9]>,
Tensor<[8,64,9]>,
|ttnn.reshape|aten::view|5| +|262|Tensor<[64,10240]>,
Tensor<[1,64,10240]>,
|ttnn.reshape|aten::view|5| +|263|Tensor<[1,64,5120]>,
Tensor<[64,5120]>,
|ttnn.reshape|aten::view|5| +|264|Tensor<[1,64,1280]>,
Tensor<[1,8,8,1280]>,
|ttnn.reshape|aten::view|5| +|265|Tensor<[1,2560,8,8]>,
Tensor<[1,32,80,64]>,
|ttnn.reshape|aten::view|5| +|266|Tensor<[1,32,80,64]>,
Tensor<[1,2560,8,8]>,
|ttnn.reshape|aten::view|5| +|267|Tensor<[1,2560,16,16]>,
Tensor<[1,32,80,256]>,
|ttnn.reshape|aten::view|5| +|268|Tensor<[1,32,80,256]>,
Tensor<[1,2560,16,16]>,
|ttnn.reshape|aten::view|5| +|269|Tensor<[1,1920,16,16]>,
Tensor<[1,32,60,256]>,
|ttnn.reshape|aten::view|5| +|270|Tensor<[1,32,60,256]>,
Tensor<[1,1920,16,16]>,
|ttnn.reshape|aten::view|5| +|271|Tensor<[1,1920,32,32]>,
Tensor<[1,32,60,1024]>,
|ttnn.reshape|aten::view|5| +|272|Tensor<[1,32,60,1024]>,
Tensor<[1,1920,32,32]>,
|ttnn.reshape|aten::view|5| +|273|Tensor<[1,1280,32,32]>,
Tensor<[1,32,40,1024]>,
|ttnn.reshape|aten::view|5| +|274|Tensor<[1,32,40,1024]>,
Tensor<[1,1280,32,32]>,
|ttnn.reshape|aten::view|5| +|275|Tensor<[1,960,32,32]>,
Tensor<[1,32,30,1024]>,
|ttnn.reshape|aten::view|5| +|276|Tensor<[1,32,30,1024]>,
Tensor<[1,960,32,32]>,
|ttnn.reshape|aten::view|5| +|277|Tensor<[1,960,64,64]>,
Tensor<[1,32,30,4096]>,
|ttnn.reshape|aten::view|5| +|278|Tensor<[1,32,30,4096]>,
Tensor<[1,960,64,64]>,
|ttnn.reshape|aten::view|5| +|279|Tensor<[1,640,64,64]>,
Tensor<[1,32,20,4096]>,
|ttnn.reshape|aten::view|5| +|280|Tensor<[1,32,20,4096]>,
Tensor<[1,640,64,64]>,
|ttnn.reshape|aten::view|5| +|281|Tensor<[1,12,25]>,
Tensor<[1,12,25,1]>,
|ttnn.reshape|aten::_safe_softmax|4| +|282|Tensor<[1,1,768]>,
Tensor<[1,768]>,
|ttnn.reshape|aten::select.int|4| +|283|Tensor<[1,25,1]>,
Tensor<[1,25]>,
|ttnn.reshape|aten::squeeze.dim|5| +|284|Tensor<[1,25]>,
Tensor<[1,1,25]>,
|ttnn.reshape|aten::unsqueeze|4| +|285|Tensor<[1,1,25]>,
Tensor<[1,1,1,25]>,
|ttnn.reshape|aten::unsqueeze|4| +|286|Tensor<[1,25,768]>,
Tensor<[25,768]>,
|ttnn.reshape|aten::view|5| +|287|Tensor<[25,768]>,
Tensor<[1,25,768]>,
|ttnn.reshape|aten::view|5| +|288|Tensor<[1,25,768]>,
Tensor<[1,25,12,64]>,
|ttnn.reshape|aten::view|5| +|289|Tensor<[1,12,25,64]>,
Tensor<[12,25,64]>,
|ttnn.reshape|aten::view|5| +|290|Tensor<[1,12,64,25]>,
Tensor<[12,64,25]>,
|ttnn.reshape|aten::view|5| +|291|Tensor<[12,25,25]>,
Tensor<[1,12,25,25]>,
|ttnn.reshape|aten::view|5| +|292|Tensor<[1,12,25,25]>,
Tensor<[12,25,25]>,
|ttnn.reshape|aten::view|5| +|293|Tensor<[12,25,64]>,
Tensor<[1,12,25,64]>,
|ttnn.reshape|aten::view|5| +|294|Tensor<[1,25,12,64]>,
Tensor<[1,25,768]>,
|ttnn.reshape|aten::view|5| +|295|Tensor<[25,3072]>,
Tensor<[1,25,3072]>,
|ttnn.reshape|aten::view|5| +|296|Tensor<[1,25,3072]>,
Tensor<[25,3072]>,
|ttnn.reshape|aten::view|5| +|297|Tensor<[25,2]>,
Tensor<[1,25,2]>,
|ttnn.reshape|aten::view|5| +|298|Tensor<[1,25]>,
Tensor<[1,25]>,
|ttnn.reshape|aten::view|5| +|299|Tensor<[1,1]>,
Tensor<[1]>,
|ttnn.reshape|aten::view|5| +|300|Tensor<[1,3,1445]>,
Tensor<[1,3,1445,1]>,
|ttnn.reshape|aten::_safe_softmax|4| +|301|Tensor<[192]>,
Tensor<[192,1,1]>,
|ttnn.reshape|aten::convolution|4| +|302|Tensor<[1,1,192]>,
Tensor<[1,192]>,
|ttnn.reshape|aten::select.int|4| +|303|Tensor<[1,192]>,
Tensor<[1,1,192]>,
|ttnn.reshape|aten::unsqueeze|5| +|304|Tensor<[1,192,32,42]>,
Tensor<[1,192,1344]>,
|ttnn.reshape|aten::view|5| +|305|Tensor<[1,192,4150]>,
Tensor<[1,192,50,83]>,
|ttnn.reshape|aten::view|5| +|306|Tensor<[1,1445,192]>,
Tensor<[1445,192]>,
|ttnn.reshape|aten::view|5| +|307|Tensor<[1445,192]>,
Tensor<[1,1445,192]>,
|ttnn.reshape|aten::view|5| +|308|Tensor<[1,1445,192]>,
Tensor<[1,1445,3,64]>,
|ttnn.reshape|aten::view|5| +|309|Tensor<[1,3,1445,64]>,
Tensor<[3,1445,64]>,
|ttnn.reshape|aten::view|5| +|310|Tensor<[1,3,64,1445]>,
Tensor<[3,64,1445]>,
|ttnn.reshape|aten::view|5| +|311|Tensor<[3,1445,1445]>,
Tensor<[1,3,1445,1445]>,
|ttnn.reshape|aten::view|5| +|312|Tensor<[1,3,1445,1445]>,
Tensor<[3,1445,1445]>,
|ttnn.reshape|aten::view|5| +|313|Tensor<[3,1445,64]>,
Tensor<[1,3,1445,64]>,
|ttnn.reshape|aten::view|5| +|314|Tensor<[1,1445,3,64]>,
Tensor<[1,1445,192]>,
|ttnn.reshape|aten::view|5| +|315|Tensor<[1445,768]>,
Tensor<[1,1445,768]>,
|ttnn.reshape|aten::view|5| +|316|Tensor<[1,1445,768]>,
Tensor<[1445,768]>,
|ttnn.reshape|aten::view|5| +|317|Tensor<[1,100,192]>,
Tensor<[100,192]>,
|ttnn.reshape|aten::view|5| +|318|Tensor<[100,192]>,
Tensor<[1,100,192]>,
|ttnn.reshape|aten::view|5| +|319|Tensor<[100,92]>,
Tensor<[1,100,92]>,
|ttnn.reshape|aten::view|5| +|320|Tensor<[100,4]>,
Tensor<[1,100,4]>,
|ttnn.reshape|aten::view|5| +|321|Tensor<[1,512]>,
Tensor<[1,512,1,1]>,
|ttnn.reshape|aten::mean.dim|4| +|322|Tensor<[1,512,1,1]>,
Tensor<[1,512]>,
|ttnn.reshape|aten::view|5| +|323|Tensor<[1,12,8]>,
Tensor<[1,12,8,1]>,
|ttnn.reshape|aten::_softmax|4| +|324|Tensor<[12,8,8]>,
Tensor<[1,12,8,8]>,
|ttnn.reshape|aten::_unsafe_view|5| +|325|Tensor<[12,8,64]>,
Tensor<[1,12,8,64]>,
|ttnn.reshape|aten::_unsafe_view|5| +|326|Tensor<[768]>,
Tensor<[768,1]>,
|ttnn.reshape|aten::convolution|4| +|327|Tensor<[3072]>,
Tensor<[3072,1]>,
|ttnn.reshape|aten::convolution|4| +|328|Tensor<[1,8]>,
Tensor<[1,1,8]>,
|ttnn.reshape|aten::unsqueeze|4| +|329|Tensor<[1,1,8]>,
Tensor<[1,1,1,8]>,
|ttnn.reshape|aten::unsqueeze|4| +|330|Tensor<[1,768,8]>,
Tensor<[1,12,64,8]>,
|ttnn.reshape|aten::view|5| +|331|Tensor<[1,12,8,64]>,
Tensor<[12,8,64]>,
|ttnn.reshape|aten::view|5| +|332|Tensor<[1,12,64,8]>,
Tensor<[12,64,8]>,
|ttnn.reshape|aten::view|5| +|333|Tensor<[1,12,8,8]>,
Tensor<[12,8,8]>,
|ttnn.reshape|aten::view|5| +|334|Tensor<[1,12,64,8]>,
Tensor<[1,768,8]>,
|ttnn.reshape|aten::view|5| +|335|Tensor<[1,8,2048]>,
Tensor<[1,8,2048,1]>,
|ttnn.reshape|aten::_softmax|4| +|336|Tensor<[8,256,2048]>,
Tensor<[1,8,256,2048]>,
|ttnn.reshape|aten::_unsafe_view|5| +|337|Tensor<[8,2048,256]>,
Tensor<[1,8,2048,256]>,
|ttnn.reshape|aten::_unsafe_view|5| +|338|Tensor<[8,2048,96]>,
Tensor<[1,8,2048,96]>,
|ttnn.reshape|aten::_unsafe_view|5| +|339|Tensor<[1,2048]>,
Tensor<[1,1,2048]>,
|ttnn.reshape|aten::unsqueeze|4| +|340|Tensor<[1,1,2048]>,
Tensor<[1,1,1,2048]>,
|ttnn.reshape|aten::unsqueeze|4| +|341|Tensor<[1,2048,768]>,
Tensor<[2048,768]>,
|ttnn.reshape|aten::view|5| +|342|Tensor<[2048,256]>,
Tensor<[1,2048,256]>,
|ttnn.reshape|aten::view|5| +|343|Tensor<[2048,1280]>,
Tensor<[1,2048,1280]>,
|ttnn.reshape|aten::view|5| +|344|Tensor<[1,256,256]>,
Tensor<[1,256,8,32]>,
|ttnn.reshape|aten::view|5| +|345|Tensor<[1,2048,256]>,
Tensor<[1,2048,8,32]>,
|ttnn.reshape|aten::view|5| +|346|Tensor<[1,2048,1280]>,
Tensor<[1,2048,8,160]>,
|ttnn.reshape|aten::view|5| +|347|Tensor<[1,8,256,32]>,
Tensor<[8,256,32]>,
|ttnn.reshape|aten::view|5| +|348|Tensor<[1,8,32,2048]>,
Tensor<[8,32,2048]>,
|ttnn.reshape|aten::view|5| +|349|Tensor<[1,8,256,2048]>,
Tensor<[8,256,2048]>,
|ttnn.reshape|aten::view|5| +|350|Tensor<[1,8,2048,160]>,
Tensor<[8,2048,160]>,
|ttnn.reshape|aten::view|5| +|351|Tensor<[1,8,32,256]>,
Tensor<[8,32,256]>,
|ttnn.reshape|aten::view|5| +|352|Tensor<[256,768]>,
Tensor<[1,256,768]>,
|ttnn.reshape|aten::view|5| +|353|Tensor<[1,256,768]>,
Tensor<[1,256,8,96]>,
|ttnn.reshape|aten::view|5| +|354|Tensor<[1,8,2048,32]>,
Tensor<[8,2048,32]>,
|ttnn.reshape|aten::view|5| +|355|Tensor<[1,8,2048,256]>,
Tensor<[8,2048,256]>,
|ttnn.reshape|aten::view|5| +|356|Tensor<[1,8,256,96]>,
Tensor<[8,256,96]>,
|ttnn.reshape|aten::view|5| +|357|Tensor<[1,2048,8,96]>,
Tensor<[1,2048,768]>,
|ttnn.reshape|aten::view|5| +|358|Tensor<[2048,768]>,
Tensor<[1,2048,768]>,
|ttnn.reshape|aten::view|5| +|359|Tensor<[2048,262]>,
Tensor<[1,2048,262]>,
|ttnn.reshape|aten::view|5| +|360|Tensor<[1,2048]>,
Tensor<[1,2048,1,1]>,
|ttnn.reshape|aten::mean.dim|4| +|361|Tensor<[1024,1]>,
Tensor<[1024,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|362|Tensor<[2048]>,
Tensor<[2048,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|363|Tensor<[2048,1]>,
Tensor<[2048,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|364|Tensor<[1,2048,1,1]>,
Tensor<[1,2048]>,
|ttnn.reshape|aten::view|5| +|365|Tensor<[1,12,201]>,
Tensor<[1,12,201,1]>,
|ttnn.reshape|aten::_softmax|4| +|366|Tensor<[12,201,201]>,
Tensor<[1,12,201,201]>,
|ttnn.reshape|aten::_unsafe_view|5| +|367|Tensor<[12,201,64]>,
Tensor<[1,12,201,64]>,
|ttnn.reshape|aten::_unsafe_view|5| +|368|Tensor<[768]>,
Tensor<[768,1,1]>,
|ttnn.reshape|aten::convolution|4| +|369|Tensor<[1,1,12,16]>,
Tensor<[1,1,12,16,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|370|Tensor<[1,1,12,16]>,
Tensor<[1,12,16]>,
|ttnn.reshape|aten::select.int|4| +|371|Tensor<[192,1]>,
Tensor<[192]>,
|ttnn.reshape|aten::select.int|4| +|372|Tensor<[12,16]>,
Tensor<[12,16,1]>,
|ttnn.reshape|aten::stack|4| +|373|Tensor<[1,384,512]>,
Tensor<[1,1,384,512]>,
|ttnn.reshape|aten::unsqueeze|4| +|374|Tensor<[12]>,
Tensor<[12,1]>,
|ttnn.reshape|aten::unsqueeze|4| +|375|Tensor<[12,16,2]>,
Tensor<[1,12,16,2]>,
|ttnn.reshape|aten::unsqueeze|4| +|376|Tensor<[1,12,16,2]>,
Tensor<[1,1,12,16,2]>,
|ttnn.reshape|aten::unsqueeze|4| +|377|Tensor<[1,201]>,
Tensor<[1,1,201]>,
|ttnn.reshape|aten::unsqueeze|4| +|378|Tensor<[1,1,201]>,
Tensor<[1,1,1,201]>,
|ttnn.reshape|aten::unsqueeze|4| +|379|Tensor<[1,768,144]>,
Tensor<[1,768,12,12]>,
|ttnn.reshape|aten::view|5| +|380|Tensor<[1,768,12,16]>,
Tensor<[1,768,192]>,
|ttnn.reshape|aten::view|5| +|381|Tensor<[16]>,
Tensor<[1,16]>,
|ttnn.reshape|aten::view|4| +|382|Tensor<[1,1,12,16,2]>,
Tensor<[1,192,2]>,
|ttnn.reshape|aten::view|4| +|383|Tensor<[1,1,12,16]>,
Tensor<[1,192]>,
|ttnn.reshape|aten::view|4| +|384|Tensor<[1,201,768]>,
Tensor<[201,768]>,
|ttnn.reshape|aten::view|5| +|385|Tensor<[201,768]>,
Tensor<[1,201,768]>,
|ttnn.reshape|aten::view|5| +|386|Tensor<[1,201,768]>,
Tensor<[1,201,12,64]>,
|ttnn.reshape|aten::view|5| +|387|Tensor<[1,12,201,64]>,
Tensor<[12,201,64]>,
|ttnn.reshape|aten::view|5| +|388|Tensor<[1,12,64,201]>,
Tensor<[12,64,201]>,
|ttnn.reshape|aten::view|5| +|389|Tensor<[1,12,201,201]>,
Tensor<[12,201,201]>,
|ttnn.reshape|aten::view|5| +|390|Tensor<[1,201,12,64]>,
Tensor<[1,201,768]>,
|ttnn.reshape|aten::view|5| +|391|Tensor<[201,3072]>,
Tensor<[1,201,3072]>,
|ttnn.reshape|aten::view|5| +|392|Tensor<[1,201,3072]>,
Tensor<[201,3072]>,
|ttnn.reshape|aten::view|5| +|393|Tensor<[32]>,
Tensor<[32,1,1]>,
|ttnn.reshape|aten::convolution|4| +|394|Tensor<[64]>,
Tensor<[64,1,1]>,
|ttnn.reshape|aten::convolution|4| +|395|Tensor<[1,64,12,12]>,
Tensor<[1,9216]>,
|ttnn.reshape|aten::view|5| +|396|Tensor<[16,19]>,
Tensor<[16,19,1]>,
|ttnn.reshape|aten::_softmax|4| +|397|Tensor<[1,19,16,64]>,
Tensor<[1,19,1024]>,
|ttnn.reshape|aten::_unsafe_view|5| +|398|Tensor<[19,256008]>,
Tensor<[1,19,256008]>,
|ttnn.reshape|aten::_unsafe_view|5| +|399|Tensor<[19]>,
Tensor<[19,1]>,
|ttnn.reshape|aten::amax|5| +|400|Tensor<[19,1]>,
Tensor<[19,1,1]>,
|ttnn.reshape|aten::gather|4| +|401|Tensor<[1,19]>,
Tensor<[19]>,
|ttnn.reshape|aten::squeeze.dim|4| +|402|Tensor<[19,1]>,
Tensor<[19]>,
|ttnn.reshape|aten::squeeze.dim|5| +|403|Tensor<[19]>,
Tensor<[1,19]>,
|ttnn.reshape|aten::unsqueeze|4| +|404|Tensor<[19,19]>,
Tensor<[1,19,19]>,
|ttnn.reshape|aten::unsqueeze|5| +|405|Tensor<[1,19,19]>,
Tensor<[1,1,19,19]>,
|ttnn.reshape|aten::unsqueeze|5| +|406|Tensor<[1,19]>,
Tensor<[1,1,19]>,
|ttnn.reshape|aten::unsqueeze|4| +|407|Tensor<[1,1,19]>,
Tensor<[1,1,1,19]>,
|ttnn.reshape|aten::unsqueeze|4| +|408|Tensor<[1,19]>,
Tensor<[1,19]>,
|ttnn.reshape|aten::view|4| +|409|Tensor<[19,1024]>,
Tensor<[1,19,1024]>,
|ttnn.reshape|aten::view|5| +|410|Tensor<[1,19,1024]>,
Tensor<[19,1024]>,
|ttnn.reshape|aten::view|5| +|411|Tensor<[1,19,1024]>,
Tensor<[1,19,16,64]>,
|ttnn.reshape|aten::view|5| +|412|Tensor<[1,16,19,64]>,
Tensor<[16,19,64]>,
|ttnn.reshape|aten::view|5| +|413|Tensor<[16,19,19]>,
Tensor<[1,16,19,19]>,
|ttnn.reshape|aten::view|5| +|414|Tensor<[1,16,19,19]>,
Tensor<[16,19,19]>,
|ttnn.reshape|aten::view|5| +|415|Tensor<[16,19,64]>,
Tensor<[1,16,19,64]>,
|ttnn.reshape|aten::view|5| +|416|Tensor<[19,4096]>,
Tensor<[1,19,4096]>,
|ttnn.reshape|aten::view|5| +|417|Tensor<[1,19,4096]>,
Tensor<[19,4096]>,
|ttnn.reshape|aten::view|5| +|418|Tensor<[1,19,256008]>,
Tensor<[19,256008]>,
|ttnn.reshape|aten::view|5| +|419|Tensor<[1,1024]>,
Tensor<[1,1024,1,1]>,
|ttnn.reshape|aten::mean.dim|4| +|420|Tensor<[14]>,
Tensor<[14,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|421|Tensor<[14,1]>,
Tensor<[14,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|422|Tensor<[24]>,
Tensor<[24,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|423|Tensor<[24,1]>,
Tensor<[24,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|424|Tensor<[40]>,
Tensor<[40,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|425|Tensor<[40,1]>,
Tensor<[40,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|426|Tensor<[68]>,
Tensor<[68,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|427|Tensor<[68,1]>,
Tensor<[68,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|428|Tensor<[16,1]>,
Tensor<[16,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|429|Tensor<[28]>,
Tensor<[28,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|430|Tensor<[28,1]>,
Tensor<[28,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|431|Tensor<[46]>,
Tensor<[46,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|432|Tensor<[46,1]>,
Tensor<[46,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|433|Tensor<[78]>,
Tensor<[78,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|434|Tensor<[78,1]>,
Tensor<[78,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|435|Tensor<[134]>,
Tensor<[134,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|436|Tensor<[134,1]>,
Tensor<[134,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|437|Tensor<[20]>,
Tensor<[20,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|438|Tensor<[20,1]>,
Tensor<[20,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|439|Tensor<[34]>,
Tensor<[34,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|440|Tensor<[34,1]>,
Tensor<[34,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|441|Tensor<[58]>,
Tensor<[58,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|442|Tensor<[58,1]>,
Tensor<[58,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|443|Tensor<[98]>,
Tensor<[98,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|444|Tensor<[98,1]>,
Tensor<[98,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|445|Tensor<[168]>,
Tensor<[168,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|446|Tensor<[168,1]>,
Tensor<[168,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|447|Tensor<[320]>,
Tensor<[320,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|448|Tensor<[320,1]>,
Tensor<[320,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|449|Tensor<[116]>,
Tensor<[116,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|450|Tensor<[116,1]>,
Tensor<[116,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|451|Tensor<[196]>,
Tensor<[196,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|452|Tensor<[196,1]>,
Tensor<[196,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|453|Tensor<[334]>,
Tensor<[334,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|454|Tensor<[334,1]>,
Tensor<[334,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|455|Tensor<[160]>,
Tensor<[160,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|456|Tensor<[160,1]>,
Tensor<[160,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|457|Tensor<[272]>,
Tensor<[272,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|458|Tensor<[272,1]>,
Tensor<[272,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|459|Tensor<[462]>,
Tensor<[462,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|460|Tensor<[462,1]>,
Tensor<[462,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|461|Tensor<[1,1024,1,1]>,
Tensor<[1,1024]>,
|ttnn.reshape|aten::view|5| +|462|Tensor<[255]>,
Tensor<[255,1,1]>,
|ttnn.reshape|aten::convolution|4| +|463|Tensor<[1,256,32,32]>,
Tensor<[1,256,32,32,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|464|Tensor<[1,128,64,64]>,
Tensor<[1,128,64,64,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|465|Tensor<[1]>,
Tensor<[1,1,1]>,
|ttnn.reshape|aten::convolution|4| +|466|Tensor<[16]>,
Tensor<[16,1,1]>,
|ttnn.reshape|aten::convolution|4| +|467|Tensor<[1,16,32]>,
Tensor<[1,16,32,1]>,
|ttnn.reshape|aten::_softmax|4| +|468|Tensor<[1,32,16,96]>,
Tensor<[1,32,1536]>,
|ttnn.reshape|aten::_unsafe_view|5| +|469|Tensor<[32,250880]>,
Tensor<[1,32,250880]>,
|ttnn.reshape|aten::_unsafe_view|5| +|470|Tensor<[1,32,16,1,96]>,
Tensor<[1,32,16,96]>,
|ttnn.reshape|aten::select.int|4| +|471|Tensor<[1,16,32]>,
Tensor<[16,1,32]>,
|ttnn.reshape|aten::view|5| +|472|Tensor<[1,32,1536]>,
Tensor<[32,1536]>,
|ttnn.reshape|aten::view|5| +|473|Tensor<[32,4608]>,
Tensor<[1,32,4608]>,
|ttnn.reshape|aten::view|5| +|474|Tensor<[1,32,4608]>,
Tensor<[1,32,16,3,96]>,
|ttnn.reshape|aten::view|5| +|475|Tensor<[1,16,32,96]>,
Tensor<[16,32,96]>,
|ttnn.reshape|aten::view|5| +|476|Tensor<[16,32,32]>,
Tensor<[1,16,32,32]>,
|ttnn.reshape|aten::view|5| +|477|Tensor<[1,16,32,32]>,
Tensor<[16,32,32]>,
|ttnn.reshape|aten::view|5| +|478|Tensor<[16,32,96]>,
Tensor<[1,16,32,96]>,
|ttnn.reshape|aten::view|5| +|479|Tensor<[32,1536]>,
Tensor<[1,32,1536]>,
|ttnn.reshape|aten::view|5| +|480|Tensor<[32,6144]>,
Tensor<[1,32,6144]>,
|ttnn.reshape|aten::view|5| +|481|Tensor<[1,32,6144]>,
Tensor<[32,6144]>,
|ttnn.reshape|aten::view|5| +|482|Tensor<[1,12,16]>,
Tensor<[1,12,16,1]>,
|ttnn.reshape|aten::_safe_softmax|4| +|483|Tensor<[1,16]>,
Tensor<[1,1,16]>,
|ttnn.reshape|aten::unsqueeze|4| +|484|Tensor<[1,1,16]>,
Tensor<[1,1,1,16]>,
|ttnn.reshape|aten::unsqueeze|4| +|485|Tensor<[1,16,768]>,
Tensor<[16,768]>,
|ttnn.reshape|aten::view|5| +|486|Tensor<[16,768]>,
Tensor<[1,16,768]>,
|ttnn.reshape|aten::view|5| +|487|Tensor<[1,16,768]>,
Tensor<[1,16,12,64]>,
|ttnn.reshape|aten::view|5| +|488|Tensor<[1,12,16,64]>,
Tensor<[12,16,64]>,
|ttnn.reshape|aten::view|5| +|489|Tensor<[1,12,64,16]>,
Tensor<[12,64,16]>,
|ttnn.reshape|aten::view|5| +|490|Tensor<[12,16,16]>,
Tensor<[1,12,16,16]>,
|ttnn.reshape|aten::view|5| +|491|Tensor<[1,12,16,16]>,
Tensor<[12,16,16]>,
|ttnn.reshape|aten::view|5| +|492|Tensor<[12,16,64]>,
Tensor<[1,12,16,64]>,
|ttnn.reshape|aten::view|5| +|493|Tensor<[1,16,12,64]>,
Tensor<[1,16,768]>,
|ttnn.reshape|aten::view|5| +|494|Tensor<[16,3072]>,
Tensor<[1,16,3072]>,
|ttnn.reshape|aten::view|5| +|495|Tensor<[1,16,3072]>,
Tensor<[16,3072]>,
|ttnn.reshape|aten::view|5| +|496|Tensor<[1,1,19200]>,
Tensor<[1,1,19200,1]>,
|ttnn.reshape|aten::_softmax|4| +|497|Tensor<[1,2,4800]>,
Tensor<[1,2,4800,1]>,
|ttnn.reshape|aten::_softmax|4| +|498|Tensor<[1,5,1200]>,
Tensor<[1,5,1200,1]>,
|ttnn.reshape|aten::_softmax|4| +|499|Tensor<[1,8,300]>,
Tensor<[1,8,300,1]>,
|ttnn.reshape|aten::_softmax|4| +|500|Tensor<[1,19200,300]>,
Tensor<[1,1,19200,300]>,
|ttnn.reshape|aten::_unsafe_view|5| +|501|Tensor<[1,19200,64]>,
Tensor<[1,1,19200,64]>,
|ttnn.reshape|aten::_unsafe_view|5| +|502|Tensor<[1,19200,64]>,
Tensor<[1,19200,64]>,
|ttnn.reshape|aten::_unsafe_view|5| +|503|Tensor<[2,4800,300]>,
Tensor<[1,2,4800,300]>,
|ttnn.reshape|aten::_unsafe_view|5| +|504|Tensor<[2,4800,64]>,
Tensor<[1,2,4800,64]>,
|ttnn.reshape|aten::_unsafe_view|5| +|505|Tensor<[1,4800,128]>,
Tensor<[1,4800,128]>,
|ttnn.reshape|aten::_unsafe_view|5| +|506|Tensor<[5,1200,300]>,
Tensor<[1,5,1200,300]>,
|ttnn.reshape|aten::_unsafe_view|5| +|507|Tensor<[5,1200,64]>,
Tensor<[1,5,1200,64]>,
|ttnn.reshape|aten::_unsafe_view|5| +|508|Tensor<[1,1200,320]>,
Tensor<[1,1200,320]>,
|ttnn.reshape|aten::_unsafe_view|5| +|509|Tensor<[8,300,300]>,
Tensor<[1,8,300,300]>,
|ttnn.reshape|aten::_unsafe_view|5| +|510|Tensor<[8,300,64]>,
Tensor<[1,8,300,64]>,
|ttnn.reshape|aten::_unsafe_view|5| +|511|Tensor<[1,300,512]>,
Tensor<[1,300,512]>,
|ttnn.reshape|aten::_unsafe_view|5| +|512|Tensor<[2048]>,
Tensor<[2048,1,1]>,
|ttnn.reshape|aten::convolution|4| +|513|Tensor<[2]>,
Tensor<[2,1,1]>,
|ttnn.reshape|aten::convolution|4| +|514|Tensor<[1,64,30,40]>,
Tensor<[1,64,30,40,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|515|Tensor<[1,64,60,80]>,
Tensor<[1,64,60,80,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|516|Tensor<[1,64,120,160]>,
Tensor<[1,64,120,160,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|517|Tensor<[1,64,240,320]>,
Tensor<[1,64,240,320,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|518|Tensor<[1,64,480,640]>,
Tensor<[1,64,480,640,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|519|Tensor<[1,1,30,40]>,
Tensor<[1,30,40]>,
|ttnn.reshape|aten::select.int|4| +|520|Tensor<[1,1,60,80]>,
Tensor<[1,60,80]>,
|ttnn.reshape|aten::select.int|4| +|521|Tensor<[1,1,120,160]>,
Tensor<[1,120,160]>,
|ttnn.reshape|aten::select.int|4| +|522|Tensor<[1,1,480,640]>,
Tensor<[1,480,640]>,
|ttnn.reshape|aten::squeeze.dim|5| +|523|Tensor<[1,30,40]>,
Tensor<[1,1,30,40]>,
|ttnn.reshape|aten::unsqueeze|5| +|524|Tensor<[1,60,80]>,
Tensor<[1,1,60,80]>,
|ttnn.reshape|aten::unsqueeze|5| +|525|Tensor<[1,120,160]>,
Tensor<[1,1,120,160]>,
|ttnn.reshape|aten::unsqueeze|5| +|526|Tensor<[1,64,120,160]>,
Tensor<[1,64,19200]>,
|ttnn.reshape|aten::view|5| +|527|Tensor<[1,19200,64]>,
Tensor<[19200,64]>,
|ttnn.reshape|aten::view|5| +|528|Tensor<[19200,64]>,
Tensor<[1,19200,64]>,
|ttnn.reshape|aten::view|5| +|529|Tensor<[1,19200,64]>,
Tensor<[1,19200,1,64]>,
|ttnn.reshape|aten::view|5| +|530|Tensor<[1,64,19200]>,
Tensor<[1,64,120,160]>,
|ttnn.reshape|aten::view|5| +|531|Tensor<[1,64,15,20]>,
Tensor<[1,64,300]>,
|ttnn.reshape|aten::view|5| +|532|Tensor<[1,300,64]>,
Tensor<[300,64]>,
|ttnn.reshape|aten::view|5| +|533|Tensor<[300,64]>,
Tensor<[1,300,64]>,
|ttnn.reshape|aten::view|5| +|534|Tensor<[1,300,64]>,
Tensor<[1,300,1,64]>,
|ttnn.reshape|aten::view|5| +|535|Tensor<[1,1,19200,64]>,
Tensor<[1,19200,64]>,
|ttnn.reshape|aten::view|5| +|536|Tensor<[1,1,64,300]>,
Tensor<[1,64,300]>,
|ttnn.reshape|aten::view|5| +|537|Tensor<[1,1,19200,300]>,
Tensor<[1,19200,300]>,
|ttnn.reshape|aten::view|5| +|538|Tensor<[1,1,300,64]>,
Tensor<[1,300,64]>,
|ttnn.reshape|aten::view|5| +|539|Tensor<[1,19200,1,64]>,
Tensor<[1,19200,64]>,
|ttnn.reshape|aten::view|5| +|540|Tensor<[19200,256]>,
Tensor<[1,19200,256]>,
|ttnn.reshape|aten::view|5| +|541|Tensor<[1,256,19200]>,
Tensor<[1,256,120,160]>,
|ttnn.reshape|aten::view|5| +|542|Tensor<[1,256,120,160]>,
Tensor<[1,256,19200]>,
|ttnn.reshape|aten::view|5| +|543|Tensor<[1,19200,256]>,
Tensor<[1,19200,256]>,
|ttnn.reshape|aten::view|5| +|544|Tensor<[1,256,64]>,
Tensor<[1,256,64]>,
|ttnn.reshape|aten::view|5| +|545|Tensor<[1,19200,64]>,
Tensor<[1,120,160,64]>,
|ttnn.reshape|aten::view|5| +|546|Tensor<[1,128,60,80]>,
Tensor<[1,128,4800]>,
|ttnn.reshape|aten::view|5| +|547|Tensor<[1,4800,128]>,
Tensor<[4800,128]>,
|ttnn.reshape|aten::view|5| +|548|Tensor<[4800,128]>,
Tensor<[1,4800,128]>,
|ttnn.reshape|aten::view|5| +|549|Tensor<[1,4800,128]>,
Tensor<[1,4800,2,64]>,
|ttnn.reshape|aten::view|5| +|550|Tensor<[1,128,4800]>,
Tensor<[1,128,60,80]>,
|ttnn.reshape|aten::view|5| +|551|Tensor<[1,128,15,20]>,
Tensor<[1,128,300]>,
|ttnn.reshape|aten::view|5| +|552|Tensor<[1,300,128]>,
Tensor<[300,128]>,
|ttnn.reshape|aten::view|5| +|553|Tensor<[300,128]>,
Tensor<[1,300,128]>,
|ttnn.reshape|aten::view|5| +|554|Tensor<[1,300,128]>,
Tensor<[1,300,2,64]>,
|ttnn.reshape|aten::view|5| +|555|Tensor<[1,2,4800,64]>,
Tensor<[2,4800,64]>,
|ttnn.reshape|aten::view|5| +|556|Tensor<[1,2,64,300]>,
Tensor<[2,64,300]>,
|ttnn.reshape|aten::view|5| +|557|Tensor<[1,2,4800,300]>,
Tensor<[2,4800,300]>,
|ttnn.reshape|aten::view|5| +|558|Tensor<[1,2,300,64]>,
Tensor<[2,300,64]>,
|ttnn.reshape|aten::view|5| +|559|Tensor<[1,4800,2,64]>,
Tensor<[1,4800,128]>,
|ttnn.reshape|aten::view|5| +|560|Tensor<[4800,512]>,
Tensor<[1,4800,512]>,
|ttnn.reshape|aten::view|5| +|561|Tensor<[1,512,4800]>,
Tensor<[1,512,60,80]>,
|ttnn.reshape|aten::view|5| +|562|Tensor<[1,512,60,80]>,
Tensor<[1,512,4800]>,
|ttnn.reshape|aten::view|5| +|563|Tensor<[1,4800,512]>,
Tensor<[1,4800,512]>,
|ttnn.reshape|aten::view|5| +|564|Tensor<[1,512,128]>,
Tensor<[1,512,128]>,
|ttnn.reshape|aten::view|5| +|565|Tensor<[1,4800,128]>,
Tensor<[1,60,80,128]>,
|ttnn.reshape|aten::view|5| +|566|Tensor<[1,320,30,40]>,
Tensor<[1,320,1200]>,
|ttnn.reshape|aten::view|5| +|567|Tensor<[1,1200,320]>,
Tensor<[1200,320]>,
|ttnn.reshape|aten::view|5| +|568|Tensor<[1200,320]>,
Tensor<[1,1200,320]>,
|ttnn.reshape|aten::view|5| +|569|Tensor<[1,1200,320]>,
Tensor<[1,1200,5,64]>,
|ttnn.reshape|aten::view|5| +|570|Tensor<[1,320,1200]>,
Tensor<[1,320,30,40]>,
|ttnn.reshape|aten::view|5| +|571|Tensor<[1,320,15,20]>,
Tensor<[1,320,300]>,
|ttnn.reshape|aten::view|5| +|572|Tensor<[1,300,320]>,
Tensor<[300,320]>,
|ttnn.reshape|aten::view|5| +|573|Tensor<[300,320]>,
Tensor<[1,300,320]>,
|ttnn.reshape|aten::view|5| +|574|Tensor<[1,300,320]>,
Tensor<[1,300,5,64]>,
|ttnn.reshape|aten::view|5| +|575|Tensor<[1,5,1200,64]>,
Tensor<[5,1200,64]>,
|ttnn.reshape|aten::view|5| +|576|Tensor<[1,5,64,300]>,
Tensor<[5,64,300]>,
|ttnn.reshape|aten::view|5| +|577|Tensor<[1,5,1200,300]>,
Tensor<[5,1200,300]>,
|ttnn.reshape|aten::view|5| +|578|Tensor<[1,5,300,64]>,
Tensor<[5,300,64]>,
|ttnn.reshape|aten::view|5| +|579|Tensor<[1,1200,5,64]>,
Tensor<[1,1200,320]>,
|ttnn.reshape|aten::view|5| +|580|Tensor<[1200,1280]>,
Tensor<[1,1200,1280]>,
|ttnn.reshape|aten::view|5| +|581|Tensor<[1,1280,1200]>,
Tensor<[1,1280,30,40]>,
|ttnn.reshape|aten::view|5| +|582|Tensor<[1,1280,30,40]>,
Tensor<[1,1280,1200]>,
|ttnn.reshape|aten::view|5| +|583|Tensor<[1,1200,1280]>,
Tensor<[1,1200,1280]>,
|ttnn.reshape|aten::view|5| +|584|Tensor<[1,1280,320]>,
Tensor<[1,1280,320]>,
|ttnn.reshape|aten::view|5| +|585|Tensor<[1,1200,320]>,
Tensor<[1,30,40,320]>,
|ttnn.reshape|aten::view|5| +|586|Tensor<[1,512,15,20]>,
Tensor<[1,512,300]>,
|ttnn.reshape|aten::view|5| +|587|Tensor<[1,300,512]>,
Tensor<[300,512]>,
|ttnn.reshape|aten::view|5| +|588|Tensor<[300,512]>,
Tensor<[1,300,512]>,
|ttnn.reshape|aten::view|5| +|589|Tensor<[1,300,512]>,
Tensor<[1,300,8,64]>,
|ttnn.reshape|aten::view|5| +|590|Tensor<[1,8,300,64]>,
Tensor<[8,300,64]>,
|ttnn.reshape|aten::view|5| +|591|Tensor<[1,8,64,300]>,
Tensor<[8,64,300]>,
|ttnn.reshape|aten::view|5| +|592|Tensor<[1,8,300,300]>,
Tensor<[8,300,300]>,
|ttnn.reshape|aten::view|5| +|593|Tensor<[1,300,8,64]>,
Tensor<[1,300,512]>,
|ttnn.reshape|aten::view|5| +|594|Tensor<[300,2048]>,
Tensor<[1,300,2048]>,
|ttnn.reshape|aten::view|5| +|595|Tensor<[1,2048,300]>,
Tensor<[1,2048,15,20]>,
|ttnn.reshape|aten::view|5| +|596|Tensor<[1,2048,15,20]>,
Tensor<[1,2048,300]>,
|ttnn.reshape|aten::view|5| +|597|Tensor<[1,300,2048]>,
Tensor<[1,300,2048]>,
|ttnn.reshape|aten::view|5| +|598|Tensor<[1,2048,512]>,
Tensor<[1,2048,512]>,
|ttnn.reshape|aten::view|5| +|599|Tensor<[1,300,512]>,
Tensor<[1,15,20,512]>,
|ttnn.reshape|aten::view|5| +|600|Tensor<[30]>,
Tensor<[30,1]>,
|ttnn.reshape|aten::view|5| +|601|Tensor<[60]>,
Tensor<[60,1]>,
|ttnn.reshape|aten::view|5| +|602|Tensor<[120]>,
Tensor<[120,1]>,
|ttnn.reshape|aten::view|5| +|603|Tensor<[240]>,
Tensor<[240,1]>,
|ttnn.reshape|aten::view|5| +|604|Tensor<[480]>,
Tensor<[480,1]>,
|ttnn.reshape|aten::view|5| +|605|Tensor<[1,12,197]>,
Tensor<[1,12,197,1]>,
|ttnn.reshape|aten::_safe_softmax|4| +|606|Tensor<[1,768,14,14]>,
Tensor<[1,768,196]>,
|ttnn.reshape|aten::view|5| +|607|Tensor<[1,197,768]>,
Tensor<[197,768]>,
|ttnn.reshape|aten::view|5| +|608|Tensor<[197,768]>,
Tensor<[1,197,768]>,
|ttnn.reshape|aten::view|5| +|609|Tensor<[1,197,768]>,
Tensor<[1,197,12,64]>,
|ttnn.reshape|aten::view|5| +|610|Tensor<[1,12,197,64]>,
Tensor<[12,197,64]>,
|ttnn.reshape|aten::view|5| +|611|Tensor<[1,12,64,197]>,
Tensor<[12,64,197]>,
|ttnn.reshape|aten::view|5| +|612|Tensor<[12,197,197]>,
Tensor<[1,12,197,197]>,
|ttnn.reshape|aten::view|5| +|613|Tensor<[1,12,197,197]>,
Tensor<[12,197,197]>,
|ttnn.reshape|aten::view|5| +|614|Tensor<[12,197,64]>,
Tensor<[1,12,197,64]>,
|ttnn.reshape|aten::view|5| +|615|Tensor<[1,197,12,64]>,
Tensor<[1,197,768]>,
|ttnn.reshape|aten::view|5| +|616|Tensor<[197,3072]>,
Tensor<[1,197,3072]>,
|ttnn.reshape|aten::view|5| +|617|Tensor<[1,197,3072]>,
Tensor<[197,3072]>,
|ttnn.reshape|aten::view|5| +|618|Tensor<[1,1,16384]>,
Tensor<[1,1,16384,1]>,
|ttnn.reshape|aten::_softmax|4| +|619|Tensor<[1,2,4096]>,
Tensor<[1,2,4096,1]>,
|ttnn.reshape|aten::_softmax|4| +|620|Tensor<[1,5,1024]>,
Tensor<[1,5,1024,1]>,
|ttnn.reshape|aten::_softmax|4| +|621|Tensor<[1,16384,256]>,
Tensor<[1,1,16384,256]>,
|ttnn.reshape|aten::_unsafe_view|5| +|622|Tensor<[1,16384,32]>,
Tensor<[1,1,16384,32]>,
|ttnn.reshape|aten::_unsafe_view|5| +|623|Tensor<[1,16384,32]>,
Tensor<[1,16384,32]>,
|ttnn.reshape|aten::_unsafe_view|5| +|624|Tensor<[2,4096,256]>,
Tensor<[1,2,4096,256]>,
|ttnn.reshape|aten::_unsafe_view|5| +|625|Tensor<[2,4096,32]>,
Tensor<[1,2,4096,32]>,
|ttnn.reshape|aten::_unsafe_view|5| +|626|Tensor<[1,4096,64]>,
Tensor<[1,4096,64]>,
|ttnn.reshape|aten::_unsafe_view|5| +|627|Tensor<[5,1024,256]>,
Tensor<[1,5,1024,256]>,
|ttnn.reshape|aten::_unsafe_view|5| +|628|Tensor<[5,1024,32]>,
Tensor<[1,5,1024,32]>,
|ttnn.reshape|aten::_unsafe_view|5| +|629|Tensor<[1,1024,160]>,
Tensor<[1,1024,160]>,
|ttnn.reshape|aten::_unsafe_view|5| +|630|Tensor<[8,256,32]>,
Tensor<[1,8,256,32]>,
|ttnn.reshape|aten::_unsafe_view|5| +|631|Tensor<[1,256,256]>,
Tensor<[1,256,256]>,
|ttnn.reshape|aten::_unsafe_view|5| +|632|Tensor<[1,16384,256]>,
Tensor<[1,16384,256]>,
|ttnn.reshape|aten::_unsafe_view|5| +|633|Tensor<[1,4096,256]>,
Tensor<[1,4096,256]>,
|ttnn.reshape|aten::_unsafe_view|5| +|634|Tensor<[1,1024,256]>,
Tensor<[1,1024,256]>,
|ttnn.reshape|aten::_unsafe_view|5| +|635|Tensor<[160]>,
Tensor<[160,1,1]>,
|ttnn.reshape|aten::convolution|4| +|636|Tensor<[1024]>,
Tensor<[1024,1,1]>,
|ttnn.reshape|aten::convolution|4| +|637|Tensor<[150]>,
Tensor<[150,1,1]>,
|ttnn.reshape|aten::convolution|4| +|638|Tensor<[1,256,128,128]>,
Tensor<[1,256,128,128,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|639|Tensor<[1,32,128,128]>,
Tensor<[1,32,16384]>,
|ttnn.reshape|aten::view|5| +|640|Tensor<[1,16384,32]>,
Tensor<[16384,32]>,
|ttnn.reshape|aten::view|5| +|641|Tensor<[16384,32]>,
Tensor<[1,16384,32]>,
|ttnn.reshape|aten::view|5| +|642|Tensor<[1,16384,32]>,
Tensor<[1,16384,1,32]>,
|ttnn.reshape|aten::view|5| +|643|Tensor<[1,32,16384]>,
Tensor<[1,32,128,128]>,
|ttnn.reshape|aten::view|5| +|644|Tensor<[1,32,16,16]>,
Tensor<[1,32,256]>,
|ttnn.reshape|aten::view|5| +|645|Tensor<[1,256,32]>,
Tensor<[256,32]>,
|ttnn.reshape|aten::view|5| +|646|Tensor<[256,32]>,
Tensor<[1,256,32]>,
|ttnn.reshape|aten::view|5| +|647|Tensor<[1,256,32]>,
Tensor<[1,256,1,32]>,
|ttnn.reshape|aten::view|5| +|648|Tensor<[1,1,16384,32]>,
Tensor<[1,16384,32]>,
|ttnn.reshape|aten::view|5| +|649|Tensor<[1,1,32,256]>,
Tensor<[1,32,256]>,
|ttnn.reshape|aten::view|5| +|650|Tensor<[1,1,16384,256]>,
Tensor<[1,16384,256]>,
|ttnn.reshape|aten::view|5| +|651|Tensor<[1,1,256,32]>,
Tensor<[1,256,32]>,
|ttnn.reshape|aten::view|5| +|652|Tensor<[1,16384,1,32]>,
Tensor<[1,16384,32]>,
|ttnn.reshape|aten::view|5| +|653|Tensor<[16384,128]>,
Tensor<[1,16384,128]>,
|ttnn.reshape|aten::view|5| +|654|Tensor<[1,128,16384]>,
Tensor<[1,128,128,128]>,
|ttnn.reshape|aten::view|5| +|655|Tensor<[1,128,128,128]>,
Tensor<[1,128,16384]>,
|ttnn.reshape|aten::view|5| +|656|Tensor<[1,16384,128]>,
Tensor<[1,16384,128]>,
|ttnn.reshape|aten::view|5| +|657|Tensor<[1,128,32]>,
Tensor<[1,128,32]>,
|ttnn.reshape|aten::view|5| +|658|Tensor<[1,16384,32]>,
Tensor<[1,128,128,32]>,
|ttnn.reshape|aten::view|5| +|659|Tensor<[1,64,64,64]>,
Tensor<[1,64,4096]>,
|ttnn.reshape|aten::view|5| +|660|Tensor<[1,4096,64]>,
Tensor<[4096,64]>,
|ttnn.reshape|aten::view|5| +|661|Tensor<[4096,64]>,
Tensor<[1,4096,64]>,
|ttnn.reshape|aten::view|5| +|662|Tensor<[1,4096,64]>,
Tensor<[1,4096,2,32]>,
|ttnn.reshape|aten::view|5| +|663|Tensor<[1,64,4096]>,
Tensor<[1,64,64,64]>,
|ttnn.reshape|aten::view|5| +|664|Tensor<[1,64,16,16]>,
Tensor<[1,64,256]>,
|ttnn.reshape|aten::view|5| +|665|Tensor<[1,256,64]>,
Tensor<[256,64]>,
|ttnn.reshape|aten::view|5| +|666|Tensor<[256,64]>,
Tensor<[1,256,64]>,
|ttnn.reshape|aten::view|5| +|667|Tensor<[1,256,64]>,
Tensor<[1,256,2,32]>,
|ttnn.reshape|aten::view|5| +|668|Tensor<[1,2,4096,32]>,
Tensor<[2,4096,32]>,
|ttnn.reshape|aten::view|5| +|669|Tensor<[1,2,32,256]>,
Tensor<[2,32,256]>,
|ttnn.reshape|aten::view|5| +|670|Tensor<[1,2,4096,256]>,
Tensor<[2,4096,256]>,
|ttnn.reshape|aten::view|5| +|671|Tensor<[1,2,256,32]>,
Tensor<[2,256,32]>,
|ttnn.reshape|aten::view|5| +|672|Tensor<[1,4096,2,32]>,
Tensor<[1,4096,64]>,
|ttnn.reshape|aten::view|5| +|673|Tensor<[4096,256]>,
Tensor<[1,4096,256]>,
|ttnn.reshape|aten::view|5| +|674|Tensor<[1,256,4096]>,
Tensor<[1,256,64,64]>,
|ttnn.reshape|aten::view|5| +|675|Tensor<[1,256,64,64]>,
Tensor<[1,256,4096]>,
|ttnn.reshape|aten::view|5| +|676|Tensor<[1,4096,64]>,
Tensor<[1,64,64,64]>,
|ttnn.reshape|aten::view|5| +|677|Tensor<[1,160,32,32]>,
Tensor<[1,160,1024]>,
|ttnn.reshape|aten::view|5| +|678|Tensor<[1,1024,160]>,
Tensor<[1024,160]>,
|ttnn.reshape|aten::view|5| +|679|Tensor<[1024,160]>,
Tensor<[1,1024,160]>,
|ttnn.reshape|aten::view|5| +|680|Tensor<[1,1024,160]>,
Tensor<[1,1024,5,32]>,
|ttnn.reshape|aten::view|5| +|681|Tensor<[1,160,1024]>,
Tensor<[1,160,32,32]>,
|ttnn.reshape|aten::view|5| +|682|Tensor<[1,160,16,16]>,
Tensor<[1,160,256]>,
|ttnn.reshape|aten::view|5| +|683|Tensor<[1,256,160]>,
Tensor<[256,160]>,
|ttnn.reshape|aten::view|5| +|684|Tensor<[256,160]>,
Tensor<[1,256,160]>,
|ttnn.reshape|aten::view|5| +|685|Tensor<[1,256,160]>,
Tensor<[1,256,5,32]>,
|ttnn.reshape|aten::view|5| +|686|Tensor<[1,5,1024,32]>,
Tensor<[5,1024,32]>,
|ttnn.reshape|aten::view|5| +|687|Tensor<[1,5,32,256]>,
Tensor<[5,32,256]>,
|ttnn.reshape|aten::view|5| +|688|Tensor<[1,5,1024,256]>,
Tensor<[5,1024,256]>,
|ttnn.reshape|aten::view|5| +|689|Tensor<[1,5,256,32]>,
Tensor<[5,256,32]>,
|ttnn.reshape|aten::view|5| +|690|Tensor<[1,1024,5,32]>,
Tensor<[1,1024,160]>,
|ttnn.reshape|aten::view|5| +|691|Tensor<[1,640,1024]>,
Tensor<[1,640,32,32]>,
|ttnn.reshape|aten::view|5| +|692|Tensor<[1,640,32,32]>,
Tensor<[1,640,1024]>,
|ttnn.reshape|aten::view|5| +|693|Tensor<[1,1024,640]>,
Tensor<[1,1024,640]>,
|ttnn.reshape|aten::view|5| +|694|Tensor<[1,640,160]>,
Tensor<[1,640,160]>,
|ttnn.reshape|aten::view|5| +|695|Tensor<[1,1024,160]>,
Tensor<[1,32,32,160]>,
|ttnn.reshape|aten::view|5| +|696|Tensor<[1,256,16,16]>,
Tensor<[1,256,256]>,
|ttnn.reshape|aten::view|5| +|697|Tensor<[1,256,8,32]>,
Tensor<[1,256,256]>,
|ttnn.reshape|aten::view|5| +|698|Tensor<[256,1024]>,
Tensor<[1,256,1024]>,
|ttnn.reshape|aten::view|5| +|699|Tensor<[1,1024,256]>,
Tensor<[1,1024,16,16]>,
|ttnn.reshape|aten::view|5| +|700|Tensor<[1,1024,16,16]>,
Tensor<[1,1024,256]>,
|ttnn.reshape|aten::view|5| +|701|Tensor<[1,256,1024]>,
Tensor<[1,256,1024]>,
|ttnn.reshape|aten::view|5| +|702|Tensor<[1,256,256]>,
Tensor<[1,16,16,256]>,
|ttnn.reshape|aten::view|5| +|703|Tensor<[1,32,256]>,
Tensor<[1,32,256]>,
|ttnn.reshape|aten::view|5| +|704|Tensor<[1,256,16384]>,
Tensor<[1,256,128,128]>,
|ttnn.reshape|aten::view|5| +|705|Tensor<[1,64,256]>,
Tensor<[1,64,256]>,
|ttnn.reshape|aten::view|5| +|706|Tensor<[1,160,256]>,
Tensor<[1,160,256]>,
|ttnn.reshape|aten::view|5| +|707|Tensor<[1,256,1024]>,
Tensor<[1,256,32,32]>,
|ttnn.reshape|aten::view|5| +|708|Tensor<[1,256,256]>,
Tensor<[1,256,16,16]>,
|ttnn.reshape|aten::view|5| +|709|Tensor<[1,71,7]>,
Tensor<[1,71,7,1]>,
|ttnn.reshape|aten::_safe_softmax|4| +|710|Tensor<[1,32,7]>,
Tensor<[1,32,7]>,
|ttnn.reshape|aten::_unsafe_view|5| +|711|Tensor<[7,4672]>,
Tensor<[1,7,4672]>,
|ttnn.reshape|aten::_unsafe_view|5| +|712|Tensor<[71,7,7]>,
Tensor<[1,71,7,7]>,
|ttnn.reshape|aten::_unsafe_view|5| +|713|Tensor<[71,7,64]>,
Tensor<[1,71,7,64]>,
|ttnn.reshape|aten::_unsafe_view|5| +|714|Tensor<[1,7,71,64]>,
Tensor<[1,7,4544]>,
|ttnn.reshape|aten::_unsafe_view|5| +|715|Tensor<[7,4544]>,
Tensor<[1,7,4544]>,
|ttnn.reshape|aten::_unsafe_view|5| +|716|Tensor<[7,18176]>,
Tensor<[1,7,18176]>,
|ttnn.reshape|aten::_unsafe_view|5| +|717|Tensor<[1,7,4544]>,
Tensor<[7,4544]>,
|ttnn.reshape|aten::_unsafe_view|5| +|718|Tensor<[7,65024]>,
Tensor<[1,7,65024]>,
|ttnn.reshape|aten::_unsafe_view|5| +|719|Tensor<[7,1]>,
Tensor<[7,1,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|720|Tensor<[1,7,1,64]>,
Tensor<[1,7,1,64,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|721|Tensor<[1,7,64]>,
Tensor<[1,1,7,64]>,
|ttnn.reshape|aten::unsqueeze|5| +|722|Tensor<[1,32,1]>,
Tensor<[1,32,1]>,
|ttnn.reshape|aten::view|5| +|723|Tensor<[1,1,7]>,
Tensor<[1,1,7]>,
|ttnn.reshape|aten::view|5| +|724|Tensor<[1,7,4672]>,
Tensor<[1,7,73,64]>,
|ttnn.reshape|aten::view|5| +|725|Tensor<[1,71,7,64]>,
Tensor<[1,71,7,64]>,
|ttnn.reshape|aten::view|5| +|726|Tensor<[1,1,7,64]>,
Tensor<[1,1,7,64]>,
|ttnn.reshape|aten::view|5| +|727|Tensor<[1,71,7,64]>,
Tensor<[71,7,64]>,
|ttnn.reshape|aten::view|5| +|728|Tensor<[1,71,64,7]>,
Tensor<[71,64,7]>,
|ttnn.reshape|aten::view|5| +|729|Tensor<[1,71,7,7]>,
Tensor<[71,7,7]>,
|ttnn.reshape|aten::view|5| +|730|Tensor<[1,7,18176]>,
Tensor<[7,18176]>,
|ttnn.reshape|aten::view|5| +|731|Tensor<[1,1280]>,
Tensor<[1,1280,1,1]>,
|ttnn.reshape|aten::mean.dim|4| +|732|Tensor<[96]>,
Tensor<[96,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|733|Tensor<[96,1]>,
Tensor<[96,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|734|Tensor<[144]>,
Tensor<[144,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|735|Tensor<[144,1]>,
Tensor<[144,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|736|Tensor<[192]>,
Tensor<[192,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|737|Tensor<[192,1]>,
Tensor<[192,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|738|Tensor<[384]>,
Tensor<[384,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|739|Tensor<[384,1]>,
Tensor<[384,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|740|Tensor<[576]>,
Tensor<[576,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|741|Tensor<[576,1]>,
Tensor<[576,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|742|Tensor<[960]>,
Tensor<[960,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|743|Tensor<[960,1]>,
Tensor<[960,1,1]>,
|ttnn.reshape|aten::unsqueeze|5| +|744|Tensor<[1,1280,1,1]>,
Tensor<[1,1280]>,
|ttnn.reshape|aten::view|5| +|745|Tensor<[1,12,12]>,
Tensor<[1,12,12,1]>,
|ttnn.reshape|aten::_safe_softmax|4| +|746|Tensor<[1,12]>,
Tensor<[1,1,12]>,
|ttnn.reshape|aten::unsqueeze|4| +|747|Tensor<[1,1,12]>,
Tensor<[1,1,1,12]>,
|ttnn.reshape|aten::unsqueeze|4| +|748|Tensor<[1,12,128]>,
Tensor<[12,128]>,
|ttnn.reshape|aten::view|5| +|749|Tensor<[12,768]>,
Tensor<[1,12,768]>,
|ttnn.reshape|aten::view|5| +|750|Tensor<[1,12,768]>,
Tensor<[12,768]>,
|ttnn.reshape|aten::view|5| +|751|Tensor<[1,12,768]>,
Tensor<[1,12,12,64]>,
|ttnn.reshape|aten::view|5| +|752|Tensor<[1,12,12,64]>,
Tensor<[12,12,64]>,
|ttnn.reshape|aten::view|5| +|753|Tensor<[1,12,64,12]>,
Tensor<[12,64,12]>,
|ttnn.reshape|aten::view|5| +|754|Tensor<[12,12,12]>,
Tensor<[1,12,12,12]>,
|ttnn.reshape|aten::view|5| +|755|Tensor<[1,12,12,12]>,
Tensor<[12,12,12]>,
|ttnn.reshape|aten::view|5| +|756|Tensor<[12,12,64]>,
Tensor<[1,12,12,64]>,
|ttnn.reshape|aten::view|5| +|757|Tensor<[1,12,12,64]>,
Tensor<[1,12,768]>,
|ttnn.reshape|aten::view|5| +|758|Tensor<[12,3072]>,
Tensor<[1,12,3072]>,
|ttnn.reshape|aten::view|5| +|759|Tensor<[1,12,3072]>,
Tensor<[12,3072]>,
|ttnn.reshape|aten::view|5| +|760|Tensor<[12,2]>,
Tensor<[1,12,2]>,
|ttnn.reshape|aten::view|5| +|761|Tensor<[1,12,9]>,
Tensor<[1,12,9,1]>,
|ttnn.reshape|aten::_safe_softmax|4| +|762|Tensor<[1,9]>,
Tensor<[1,1,9]>,
|ttnn.reshape|aten::unsqueeze|4| +|763|Tensor<[1,1,9]>,
Tensor<[1,1,1,9]>,
|ttnn.reshape|aten::unsqueeze|4| +|764|Tensor<[1,9,128]>,
Tensor<[9,128]>,
|ttnn.reshape|aten::view|5| +|765|Tensor<[9,768]>,
Tensor<[1,9,768]>,
|ttnn.reshape|aten::view|5| +|766|Tensor<[1,9,768]>,
Tensor<[1,9,12,64]>,
|ttnn.reshape|aten::view|5| +|767|Tensor<[1,12,9,64]>,
Tensor<[12,9,64]>,
|ttnn.reshape|aten::view|5| +|768|Tensor<[1,12,64,9]>,
Tensor<[12,64,9]>,
|ttnn.reshape|aten::view|5| +|769|Tensor<[12,9,9]>,
Tensor<[1,12,9,9]>,
|ttnn.reshape|aten::view|5| +|770|Tensor<[1,12,9,9]>,
Tensor<[12,9,9]>,
|ttnn.reshape|aten::view|5| +|771|Tensor<[12,9,64]>,
Tensor<[1,12,9,64]>,
|ttnn.reshape|aten::view|5| +|772|Tensor<[1,9,12,64]>,
Tensor<[1,9,768]>,
|ttnn.reshape|aten::view|5| +|773|Tensor<[9,3072]>,
Tensor<[1,9,3072]>,
|ttnn.reshape|aten::view|5| +|774|Tensor<[1,9,3072]>,
Tensor<[9,3072]>,
|ttnn.reshape|aten::view|5| +|775|Tensor<[9,128]>,
Tensor<[1,9,128]>,
|ttnn.reshape|aten::view|5| +|776|Tensor<[9,30000]>,
Tensor<[1,9,30000]>,
|ttnn.reshape|aten::view|5| +|777|Tensor<[1,16,9]>,
Tensor<[1,16,9,1]>,
|ttnn.reshape|aten::_safe_softmax|4| +|778|Tensor<[9,2048]>,
Tensor<[1,9,2048]>,
|ttnn.reshape|aten::view|5| +|779|Tensor<[1,9,2048]>,
Tensor<[9,2048]>,
|ttnn.reshape|aten::view|5| +|780|Tensor<[1,9,2048]>,
Tensor<[1,9,16,128]>,
|ttnn.reshape|aten::view|5| +|781|Tensor<[1,16,9,128]>,
Tensor<[16,9,128]>,
|ttnn.reshape|aten::view|5| +|782|Tensor<[1,16,128,9]>,
Tensor<[16,128,9]>,
|ttnn.reshape|aten::view|5| +|783|Tensor<[16,9,9]>,
Tensor<[1,16,9,9]>,
|ttnn.reshape|aten::view|5| +|784|Tensor<[1,16,9,9]>,
Tensor<[16,9,9]>,
|ttnn.reshape|aten::view|5| +|785|Tensor<[16,9,128]>,
Tensor<[1,16,9,128]>,
|ttnn.reshape|aten::view|5| +|786|Tensor<[1,9,16,128]>,
Tensor<[1,9,2048]>,
|ttnn.reshape|aten::view|5| +|787|Tensor<[9,8192]>,
Tensor<[1,9,8192]>,
|ttnn.reshape|aten::view|5| +|788|Tensor<[1,9,8192]>,
Tensor<[9,8192]>,
|ttnn.reshape|aten::view|5| +|789|Tensor<[9,1024]>,
Tensor<[1,9,1024]>,
|ttnn.reshape|aten::view|5| +|790|Tensor<[1,9,1024]>,
Tensor<[9,1024]>,
|ttnn.reshape|aten::view|5| +|791|Tensor<[1,9,1024]>,
Tensor<[1,9,16,64]>,
|ttnn.reshape|aten::view|5| +|792|Tensor<[1,16,9,64]>,
Tensor<[16,9,64]>,
|ttnn.reshape|aten::view|5| +|793|Tensor<[1,16,64,9]>,
Tensor<[16,64,9]>,
|ttnn.reshape|aten::view|5| +|794|Tensor<[16,9,64]>,
Tensor<[1,16,9,64]>,
|ttnn.reshape|aten::view|5| +|795|Tensor<[1,9,16,64]>,
Tensor<[1,9,1024]>,
|ttnn.reshape|aten::view|5| +|796|Tensor<[9,4096]>,
Tensor<[1,9,4096]>,
|ttnn.reshape|aten::view|5| +|797|Tensor<[1,9,4096]>,
Tensor<[9,4096]>,
|ttnn.reshape|aten::view|5| +|798|Tensor<[1,64,9]>,
Tensor<[1,64,9,1]>,
|ttnn.reshape|aten::_safe_softmax|4| +|799|Tensor<[1,9,4096]>,
Tensor<[1,9,64,64]>,
|ttnn.reshape|aten::view|5| +|800|Tensor<[1,64,9,64]>,
Tensor<[64,9,64]>,
|ttnn.reshape|aten::view|5| +|801|Tensor<[1,64,64,9]>,
Tensor<[64,64,9]>,
|ttnn.reshape|aten::view|5| +|802|Tensor<[64,9,9]>,
Tensor<[1,64,9,9]>,
|ttnn.reshape|aten::view|5| +|803|Tensor<[1,64,9,9]>,
Tensor<[64,9,9]>,
|ttnn.reshape|aten::view|5| +|804|Tensor<[64,9,64]>,
Tensor<[1,64,9,64]>,
|ttnn.reshape|aten::view|5| +|805|Tensor<[1,9,64,64]>,
Tensor<[1,9,4096]>,
|ttnn.reshape|aten::view|5| +|806|Tensor<[9,16384]>,
Tensor<[1,9,16384]>,
|ttnn.reshape|aten::view|5| +|807|Tensor<[1,9,16384]>,
Tensor<[9,16384]>,
|ttnn.reshape|aten::view|5| +|808|Tensor<[1,12,14]>,
Tensor<[1,12,14,1]>,
|ttnn.reshape|aten::_safe_softmax|4| +|809|Tensor<[1,14,1]>,
Tensor<[1,14]>,
|ttnn.reshape|aten::squeeze.dim|5| +|810|Tensor<[1,14]>,
Tensor<[1,1,14]>,
|ttnn.reshape|aten::unsqueeze|4| +|811|Tensor<[1,1,14]>,
Tensor<[1,1,1,14]>,
|ttnn.reshape|aten::unsqueeze|4| +|812|Tensor<[1,14,128]>,
Tensor<[14,128]>,
|ttnn.reshape|aten::view|5| +|813|Tensor<[14,768]>,
Tensor<[1,14,768]>,
|ttnn.reshape|aten::view|5| +|814|Tensor<[1,14,768]>,
Tensor<[14,768]>,
|ttnn.reshape|aten::view|5| +|815|Tensor<[1,14,768]>,
Tensor<[1,14,12,64]>,
|ttnn.reshape|aten::view|5| +|816|Tensor<[1,12,14,64]>,
Tensor<[12,14,64]>,
|ttnn.reshape|aten::view|5| +|817|Tensor<[1,12,64,14]>,
Tensor<[12,64,14]>,
|ttnn.reshape|aten::view|5| +|818|Tensor<[12,14,14]>,
Tensor<[1,12,14,14]>,
|ttnn.reshape|aten::view|5| +|819|Tensor<[1,12,14,14]>,
Tensor<[12,14,14]>,
|ttnn.reshape|aten::view|5| +|820|Tensor<[12,14,64]>,
Tensor<[1,12,14,64]>,
|ttnn.reshape|aten::view|5| +|821|Tensor<[1,14,12,64]>,
Tensor<[1,14,768]>,
|ttnn.reshape|aten::view|5| +|822|Tensor<[14,3072]>,
Tensor<[1,14,3072]>,
|ttnn.reshape|aten::view|5| +|823|Tensor<[1,14,3072]>,
Tensor<[14,3072]>,
|ttnn.reshape|aten::view|5| +|824|Tensor<[14,2]>,
Tensor<[1,14,2]>,
|ttnn.reshape|aten::view|5| +|825|Tensor<[1,12,50]>,
Tensor<[1,12,50,1]>,
|ttnn.reshape|aten::_safe_softmax|4| +|826|Tensor<[2,8,7]>,
Tensor<[2,8,7,1]>,
|ttnn.reshape|aten::_safe_softmax|4| +|827|Tensor<[2,8,7,64]>,
Tensor<[16,7,64]>,
|ttnn.reshape|aten::_unsafe_view|5| +|828|Tensor<[2,8,64,7]>,
Tensor<[16,64,7]>,
|ttnn.reshape|aten::_unsafe_view|5| +|829|Tensor<[2]>,
Tensor<[2,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|830|Tensor<[2,7]>,
Tensor<[2,1,7]>,
|ttnn.reshape|aten::unsqueeze|4| +|831|Tensor<[2,1,7]>,
Tensor<[2,1,1,7]>,
|ttnn.reshape|aten::unsqueeze|4| +|832|Tensor<[1,768,7,7]>,
Tensor<[1,768,49]>,
|ttnn.reshape|aten::view|5| +|833|Tensor<[1,50,768]>,
Tensor<[50,768]>,
|ttnn.reshape|aten::view|5| +|834|Tensor<[50,768]>,
Tensor<[1,50,768]>,
|ttnn.reshape|aten::view|5| +|835|Tensor<[1,50,768]>,
Tensor<[1,50,12,64]>,
|ttnn.reshape|aten::view|5| +|836|Tensor<[1,12,50,64]>,
Tensor<[12,50,64]>,
|ttnn.reshape|aten::view|5| +|837|Tensor<[1,12,64,50]>,
Tensor<[12,64,50]>,
|ttnn.reshape|aten::view|5| +|838|Tensor<[12,50,50]>,
Tensor<[1,12,50,50]>,
|ttnn.reshape|aten::view|5| +|839|Tensor<[1,12,50,50]>,
Tensor<[12,50,50]>,
|ttnn.reshape|aten::view|5| +|840|Tensor<[12,50,64]>,
Tensor<[1,12,50,64]>,
|ttnn.reshape|aten::view|5| +|841|Tensor<[1,50,12,64]>,
Tensor<[1,50,768]>,
|ttnn.reshape|aten::view|5| +|842|Tensor<[50,3072]>,
Tensor<[1,50,3072]>,
|ttnn.reshape|aten::view|5| +|843|Tensor<[1,50,3072]>,
Tensor<[50,3072]>,
|ttnn.reshape|aten::view|5| +|844|Tensor<[2,7]>,
Tensor<[2,7]>,
|ttnn.reshape|aten::view|4| +|845|Tensor<[2,7,512]>,
Tensor<[14,512]>,
|ttnn.reshape|aten::view|5| +|846|Tensor<[14,512]>,
Tensor<[2,7,512]>,
|ttnn.reshape|aten::view|5| +|847|Tensor<[2,7,512]>,
Tensor<[2,7,8,64]>,
|ttnn.reshape|aten::view|5| +|848|Tensor<[16,7,7]>,
Tensor<[2,8,7,7]>,
|ttnn.reshape|aten::view|5| +|849|Tensor<[2,8,7,7]>,
Tensor<[16,7,7]>,
|ttnn.reshape|aten::view|5| +|850|Tensor<[16,7,64]>,
Tensor<[2,8,7,64]>,
|ttnn.reshape|aten::view|5| +|851|Tensor<[2,7,8,64]>,
Tensor<[2,7,512]>,
|ttnn.reshape|aten::view|5| +|852|Tensor<[14,2048]>,
Tensor<[2,7,2048]>,
|ttnn.reshape|aten::view|5| +|853|Tensor<[2,7,2048]>,
Tensor<[14,2048]>,
|ttnn.reshape|aten::view|5| +|854|Tensor<[1,16,197]>,
Tensor<[1,16,197,1]>,
|ttnn.reshape|aten::_softmax|4| +|855|Tensor<[197,1024]>,
Tensor<[1,197,1024]>,
|ttnn.reshape|aten::_unsafe_view|5| +|856|Tensor<[16,197,197]>,
Tensor<[1,16,197,197]>,
|ttnn.reshape|aten::_unsafe_view|5| +|857|Tensor<[16,197,64]>,
Tensor<[1,16,197,64]>,
|ttnn.reshape|aten::_unsafe_view|5| +|858|Tensor<[1,16,27,27]>,
Tensor<[1,16,27,27,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|859|Tensor<[38809]>,
Tensor<[38809,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|860|Tensor<[196,196,1]>,
Tensor<[196,196]>,
|ttnn.reshape|aten::select.int|4| +|861|Tensor<[1,197]>,
Tensor<[197]>,
|ttnn.reshape|aten::select.int|4| +|862|Tensor<[197,1]>,
Tensor<[197]>,
|ttnn.reshape|aten::select.int|4| +|863|Tensor<[196,196]>,
Tensor<[196,196,1]>,
|ttnn.reshape|aten::select_scatter|4| +|864|Tensor<[197]>,
Tensor<[1,197]>,
|ttnn.reshape|aten::select_scatter|4| +|865|Tensor<[197]>,
Tensor<[197,1]>,
|ttnn.reshape|aten::select_scatter|4| +|866|Tensor<[14,14]>,
Tensor<[1,14,14]>,
|ttnn.reshape|aten::stack|4| +|867|Tensor<[2,196]>,
Tensor<[2,196,1]>,
|ttnn.reshape|aten::unsqueeze|4| +|868|Tensor<[2,196]>,
Tensor<[2,1,196]>,
|ttnn.reshape|aten::unsqueeze|4| +|869|Tensor<[1,1024,14,14]>,
Tensor<[1,1024,196]>,
|ttnn.reshape|aten::view|5| +|870|Tensor<[1,197,1024]>,
Tensor<[197,1024]>,
|ttnn.reshape|aten::view|5| +|871|Tensor<[1,197,1024]>,
Tensor<[1,197,16,64]>,
|ttnn.reshape|aten::view|5| +|872|Tensor<[1,16,197,64]>,
Tensor<[16,197,64]>,
|ttnn.reshape|aten::view|5| +|873|Tensor<[1,16,64,197]>,
Tensor<[16,64,197]>,
|ttnn.reshape|aten::view|5| +|874|Tensor<[729,16]>,
Tensor<[1,27,27,16]>,
|ttnn.reshape|aten::view|5| +|875|Tensor<[27]>,
Tensor<[27,1]>,
|ttnn.reshape|aten::view|5| +|876|Tensor<[1,27,27,16]>,
Tensor<[729,16]>,
|ttnn.reshape|aten::view|5| +|877|Tensor<[14]>,
Tensor<[1,14]>,
|ttnn.reshape|aten::view|4| +|878|Tensor<[2,14,14]>,
Tensor<[2,196]>,
|ttnn.reshape|aten::view|4| +|879|Tensor<[197,197]>,
Tensor<[38809]>,
|ttnn.reshape|aten::view|4| +|880|Tensor<[38809,16]>,
Tensor<[197,197,16]>,
|ttnn.reshape|aten::view|5| +|881|Tensor<[1,16,197,197]>,
Tensor<[16,197,197]>,
|ttnn.reshape|aten::view|5| +|882|Tensor<[1,197,16,64]>,
Tensor<[1,197,1024]>,
|ttnn.reshape|aten::view|5| +|883|Tensor<[197,4096]>,
Tensor<[1,197,4096]>,
|ttnn.reshape|aten::view|5| +|884|Tensor<[1,197,4096]>,
Tensor<[197,4096]>,
|ttnn.reshape|aten::view|5| +|885|Tensor<[12,1]>,
Tensor<[12,1,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|886|Tensor<[1,12,27,27]>,
Tensor<[1,12,27,27,1]>,
|ttnn.reshape|aten::index.Tensor|4| +|887|Tensor<[729,12]>,
Tensor<[1,27,27,12]>,
|ttnn.reshape|aten::view|5| +|888|Tensor<[1,27,27,12]>,
Tensor<[729,12]>,
|ttnn.reshape|aten::view|5| +|889|Tensor<[38809,12]>,
Tensor<[197,197,12]>,
|ttnn.reshape|aten::view|5| diff --git a/docs/ops/stablehlo/stablehlo.reverse.md b/docs/ops/stablehlo/stablehlo.reverse.md new file mode 100644 index 00000000..ce56e7f5 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.reverse.md @@ -0,0 +1,13 @@ + +### stablehlo.reverse::ttnn.? + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[2,2,256,512]>,
dims: [0, 1]
|ttnn.?|aten::convolution|4| +|1|Tensor<[2,2,128,256]>,
dims: [0, 1]
|ttnn.?|aten::convolution|4| +|2|Tensor<[2,2,64,128]>,
dims: [0, 1]
|ttnn.?|aten::convolution|4| +|3|Tensor<[2,2,32,64]>,
dims: [0, 1]
|ttnn.?|aten::convolution|4| +|4|Tensor<[2,2,16,4]>,
dims: [0, 1]
|ttnn.?|aten::convolution|4| +|5|Tensor<[2,2,1,16]>,
dims: [0, 1]
|ttnn.?|aten::convolution|4| +|6|Tensor<[2,2,512,1024]>,
dims: [0, 1]
|ttnn.?|aten::convolution|4| diff --git a/docs/ops/stablehlo/stablehlo.rng.md b/docs/ops/stablehlo/stablehlo.rng.md new file mode 100644 index 00000000..f76616e7 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.rng.md @@ -0,0 +1,7 @@ + +### stablehlo.rng + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Scalar,
Scalar,
Tensor<[3]>,
distribution: UNIFORM
||aten::rand|4| diff --git a/docs/ops/stablehlo/stablehlo.rsqrt.md b/docs/ops/stablehlo/stablehlo.rsqrt.md new file mode 100644 index 00000000..59077d3b --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.rsqrt.md @@ -0,0 +1,69 @@ + +### stablehlo.rsqrt::ttnn.rsqrt + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,32,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|1|Tensor<[1,7,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|2|Tensor<[1,1024,512]>,
|ttnn.rsqrt|aten::gelu|4| +|3|Tensor<[1,256,256]>,
|ttnn.rsqrt|aten::gelu|4| +|4|Tensor<[1,256,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|5|Tensor<[1,64,1,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|6|Tensor<[1,256,1,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|7|Tensor<[1,128,1,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|8|Tensor<[1,512,1,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|9|Tensor<[1,1024,1,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|10|Tensor<[1,2048,1,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|11|Tensor<[920,1,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|12|Tensor<[100,1,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|13|Tensor<[1,10,3072]>,
|ttnn.rsqrt|aten::gelu|4| +|14|Tensor<[1,10,768]>,
|ttnn.rsqrt|aten::gelu|4| +|15|Tensor<[1,10,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|16|Tensor<[1,4096,1280]>,
|ttnn.rsqrt|aten::gelu|4| +|17|Tensor<[1,1024,2560]>,
|ttnn.rsqrt|aten::gelu|4| +|18|Tensor<[1,256,5120]>,
|ttnn.rsqrt|aten::gelu|4| +|19|Tensor<[1,64,5120]>,
|ttnn.rsqrt|aten::gelu|4| +|20|Tensor<[1,32,1,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|21|Tensor<[1,4096,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|22|Tensor<[1,1024,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|23|Tensor<[1,64,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|24|Tensor<[1,25,3072]>,
|ttnn.rsqrt|aten::gelu|4| +|25|Tensor<[1,25,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|26|Tensor<[1,1445,768]>,
|ttnn.rsqrt|aten::gelu|4| +|27|Tensor<[1,1445,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|28|Tensor<[1,3072,8]>,
|ttnn.rsqrt|aten::gelu|4| +|29|Tensor<[1,8,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|30|Tensor<[1,256,1280]>,
|ttnn.rsqrt|aten::gelu|4| +|31|Tensor<[1,2048,768]>,
|ttnn.rsqrt|aten::gelu|4| +|32|Tensor<[1,2048,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|33|Tensor<[1,201,3072]>,
|ttnn.rsqrt|aten::gelu|4| +|34|Tensor<[1,1536]>,
|ttnn.rsqrt|aten::gelu|4| +|35|Tensor<[1,201,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|36|Tensor<[1,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|37|Tensor<[1,19,4096]>,
|ttnn.rsqrt|aten::gelu|4| +|38|Tensor<[1,19,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|39|Tensor<[1,16,3072]>,
|ttnn.rsqrt|aten::gelu|4| +|40|Tensor<[1,16,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|41|Tensor<[1,19200,256]>,
|ttnn.rsqrt|aten::gelu|4| +|42|Tensor<[1,4800,512]>,
|ttnn.rsqrt|aten::gelu|4| +|43|Tensor<[1,1200,1280]>,
|ttnn.rsqrt|aten::gelu|4| +|44|Tensor<[1,300,2048]>,
|ttnn.rsqrt|aten::gelu|4| +|45|Tensor<[1,19200,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|46|Tensor<[1,300,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|47|Tensor<[1,4800,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|48|Tensor<[1,1200,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|49|Tensor<[1,197,3072]>,
|ttnn.rsqrt|aten::gelu|4| +|50|Tensor<[1,197,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|51|Tensor<[1,16384,128]>,
|ttnn.rsqrt|aten::gelu|4| +|52|Tensor<[1,4096,256]>,
|ttnn.rsqrt|aten::gelu|4| +|53|Tensor<[1,1024,640]>,
|ttnn.rsqrt|aten::gelu|4| +|54|Tensor<[1,256,1024]>,
|ttnn.rsqrt|aten::gelu|4| +|55|Tensor<[1,16384,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|56|Tensor<[1,7,18176]>,
|ttnn.rsqrt|aten::gelu|4| +|57|Tensor<[1,12,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|58|Tensor<[1,9,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|59|Tensor<[1,14,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|60|Tensor<[1,50,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|61|Tensor<[2,7,1]>,
|ttnn.rsqrt|aten::rsqrt|5| +|62|Tensor<[1,197,4096]>,
|ttnn.rsqrt|aten::gelu|4| diff --git a/docs/ops/stablehlo/stablehlo.scatter.md b/docs/ops/stablehlo/stablehlo.scatter.md new file mode 100644 index 00000000..e6d055a5 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.scatter.md @@ -0,0 +1,12 @@ + +### stablehlo.scatter::ttnn.scatter + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,3,720,1280]>,
Tensor<[1,1]>,
Tensor<[1,3,720,1280]>,
update_window_dims: [1, 2, 3]
inserted_window_dims: [0]
scatter_dims_to_operand_dims: [0]
index_vector_dim: 1>
|ttnn.scatter|aten::select_scatter|4| +|1|Tensor<[1,720,1280]>,
Tensor<[1,1]>,
Tensor<[1,720,1280]>,
update_window_dims: [1, 2]
inserted_window_dims: [0]
scatter_dims_to_operand_dims: [0]
index_vector_dim: 1>
|ttnn.scatter|aten::select_scatter|4| +|2|Tensor<[196,196,2]>,
Tensor<[1,1]>,
Tensor<[196,196,1]>,
update_window_dims: [0, 1]
inserted_window_dims: [2]
scatter_dims_to_operand_dims: [2]
index_vector_dim: 1>
|ttnn.scatter|aten::select_scatter|4| +|3|Tensor<[197,197]>,
Tensor<[1,1]>,
Tensor<[1,197]>,
update_window_dims: [1]
inserted_window_dims: [0]
scatter_dims_to_operand_dims: [0]
index_vector_dim: 1>
|ttnn.scatter|aten::select_scatter|4| +|4|Tensor<[197,197]>,
Tensor<[1,1]>,
Tensor<[197,1]>,
update_window_dims: [0]
inserted_window_dims: [1]
scatter_dims_to_operand_dims: [1]
index_vector_dim: 1>
|ttnn.scatter|aten::select_scatter|4| +|5|Tensor<[197]>,
Tensor<[1,1]>,
Tensor<[1]>,
inserted_window_dims: [0]
scatter_dims_to_operand_dims: [0]
index_vector_dim: 1>
|ttnn.scatter|aten::select_scatter|4| diff --git a/docs/ops/stablehlo/stablehlo.select.md b/docs/ops/stablehlo/stablehlo.select.md new file mode 100644 index 00000000..f1777843 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.select.md @@ -0,0 +1,47 @@ + +### stablehlo.select::ttnn.where + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,32,32,32]>,
Tensor<[1,32,32,32]>,
|ttnn.where|aten::_safe_softmax|4| +|1|Tensor<[32,32]>,
Tensor<[32,32]>,
|ttnn.where|aten::triu|4| +|2|Tensor<[1,1,32,32]>,
Tensor<[1,1,32,32]>,
|ttnn.where|aten::where.self|4| +|3|Tensor<[1,12,7,7]>,
Tensor<[1,12,7,7]>,
|ttnn.where|aten::_safe_softmax|4| +|4|Tensor<[7,7]>,
Tensor<[7,7]>,
|ttnn.where|aten::where.self|4| +|5|Tensor<[1,1,7,7]>,
Tensor<[1,1,7,7]>,
|ttnn.where|aten::where.self|4| +|6|Tensor<[1,920]>,
Tensor<[1,920]>,
|ttnn.where|aten::where.self|4| +|7|Tensor<[1,12,10,10]>,
Tensor<[1,12,10,10]>,
|ttnn.where|aten::_safe_softmax|4| +|8|Tensor<[1,1,10,10]>,
Tensor<[1,1,10,10]>,
|ttnn.where|aten::where.self|4| +|9|Tensor<[1,8,4096,4096]>,
Tensor<[1,8,4096,4096]>,
|ttnn.where|aten::_safe_softmax|4| +|10|Tensor<[1,8,4096,9]>,
Tensor<[1,8,4096,9]>,
|ttnn.where|aten::_safe_softmax|4| +|11|Tensor<[1,8,1024,1024]>,
Tensor<[1,8,1024,1024]>,
|ttnn.where|aten::_safe_softmax|4| +|12|Tensor<[1,8,1024,9]>,
Tensor<[1,8,1024,9]>,
|ttnn.where|aten::_safe_softmax|4| +|13|Tensor<[1,8,256,256]>,
Tensor<[1,8,256,256]>,
|ttnn.where|aten::_safe_softmax|4| +|14|Tensor<[1,8,256,9]>,
Tensor<[1,8,256,9]>,
|ttnn.where|aten::_safe_softmax|4| +|15|Tensor<[1,8,64,64]>,
Tensor<[1,8,64,64]>,
|ttnn.where|aten::_safe_softmax|4| +|16|Tensor<[1,8,64,9]>,
Tensor<[1,8,64,9]>,
|ttnn.where|aten::_safe_softmax|4| +|17|Tensor<[1,12,25,25]>,
Tensor<[1,12,25,25]>,
|ttnn.where|aten::_safe_softmax|4| +|18|Tensor<[1,1,25,25]>,
Tensor<[1,1,25,25]>,
|ttnn.where|aten::where.self|4| +|19|Tensor<[1,3,1445,1445]>,
Tensor<[1,3,1445,1445]>,
|ttnn.where|aten::_safe_softmax|4| +|20|Tensor<[19,19]>,
Tensor<[19,19]>,
|ttnn.where|aten::where.self|4| +|21|Tensor<[1,1,19,19]>,
Tensor<[1,1,19,19]>,
|ttnn.where|aten::where.self|4| +|22|Tensor<[1,19]>,
Tensor<[1,19]>,
|ttnn.where|aten::where.self|4| +|23|Tensor<[19]>,
Tensor<[19]>,
|ttnn.where|aten::where.self|4| +|24|Tensor<[1,12,16,16]>,
Tensor<[1,12,16,16]>,
|ttnn.where|aten::_safe_softmax|4| +|25|Tensor<[1,1,16,16]>,
Tensor<[1,1,16,16]>,
|ttnn.where|aten::where.self|4| +|26|Tensor<[1,12,197,197]>,
Tensor<[1,12,197,197]>,
|ttnn.where|aten::_safe_softmax|4| +|27|Tensor<[1,71,7,7]>,
Tensor<[1,71,7,7]>,
|ttnn.where|aten::_safe_softmax|4| +|28|Tensor<[1,12,12,12]>,
Tensor<[1,12,12,12]>,
|ttnn.where|aten::_safe_softmax|4| +|29|Tensor<[1,1,12,12]>,
Tensor<[1,1,12,12]>,
|ttnn.where|aten::where.self|4| +|30|Tensor<[1,12,9,9]>,
Tensor<[1,12,9,9]>,
|ttnn.where|aten::_safe_softmax|4| +|31|Tensor<[1,1,9,9]>,
Tensor<[1,1,9,9]>,
|ttnn.where|aten::where.self|4| +|32|Tensor<[1,16,9,9]>,
Tensor<[1,16,9,9]>,
|ttnn.where|aten::_safe_softmax|4| +|33|Tensor<[1,64,9,9]>,
Tensor<[1,64,9,9]>,
|ttnn.where|aten::_safe_softmax|4| +|34|Tensor<[1,12,14,14]>,
Tensor<[1,12,14,14]>,
|ttnn.where|aten::_safe_softmax|4| +|35|Tensor<[1,1,14,14]>,
Tensor<[1,1,14,14]>,
|ttnn.where|aten::where.self|4| +|36|Tensor<[1,12,50,50]>,
Tensor<[1,12,50,50]>,
|ttnn.where|aten::_safe_softmax|4| +|37|Tensor<[2,8,7,7]>,
Tensor<[2,8,7,7]>,
|ttnn.where|aten::_safe_softmax|4| +|38|Tensor<[2,1,7,7]>,
Tensor<[2,1,7,7]>,
|ttnn.where|aten::where.self|4| +|39|Tensor<[196,197]>,
Tensor<[196,197]>,
|ttnn.where|aten::where.self|4| +|40|Tensor<[197,197]>,
Tensor<[197,197]>,
|ttnn.where|aten::where.self|4| diff --git a/docs/ops/stablehlo/stablehlo.sine.md b/docs/ops/stablehlo/stablehlo.sine.md new file mode 100644 index 00000000..1f18f66b --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.sine.md @@ -0,0 +1,10 @@ + +### stablehlo.sine::ttnn.sin + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,32,128]>,
|ttnn.sin|aten::sin|4| +|1|Tensor<[1,23,40,64]>,
|ttnn.sin|aten::sin|4| +|2|Tensor<[1,160]>,
|ttnn.sin|aten::sin|4| +|3|Tensor<[1,7,64]>,
|ttnn.sin|aten::sin|4| diff --git a/docs/ops/stablehlo/stablehlo.slice.md b/docs/ops/stablehlo/stablehlo.slice.md new file mode 100644 index 00000000..80ad752f --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.slice.md @@ -0,0 +1,92 @@ + +### stablehlo.slice::ttnn.slice + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,32,32,128]>,
indices: [0:1, 0:32, 0:32, 0:64]
|ttnn.reshape|aten::slice.Tensor|4| +|1|Tensor<[1,32,32,128]>,
indices: [0:1, 0:32, 0:32, 64:128]
|ttnn.reshape|aten::slice.Tensor|4| +|2|Tensor<[1,7,2304]>,
indices: [0:1, 0:7, 0:768]
|ttnn.reshape|aten::slice.Tensor|4| +|3|Tensor<[1,7,2304]>,
indices: [0:1, 0:7, 768:1536]
|ttnn.reshape|aten::slice.Tensor|4| +|4|Tensor<[1,7,2304]>,
indices: [0:1, 0:7, 1536:2304]
|ttnn.reshape|aten::slice.Tensor|4| +|5|Tensor<[1,185,28,28]>,
indices: [0:1, 0:128, 0:28, 0:28]
|ttnn.reshape|aten::slice.Tensor|4| +|6|Tensor<[1,185,28,28]>,
indices: [0:1, 128:185, 0:28, 0:28]
|ttnn.reshape|aten::slice.Tensor|4| +|7|Tensor<[6,1,100,4]>,
indices: [5:6, 0:1, 0:100, 0:4]
|ttnn.reshape|aten::select.int|4| +|8|Tensor<[6,1,100,92]>,
indices: [5:6, 0:1, 0:100, 0:92]
|ttnn.reshape|aten::select.int|4| +|9|Tensor<[1,23,40]>,
indices: [0:1, 22:23, 0:40]
|ttnn.reshape|aten::slice.Tensor|4| +|10|Tensor<[1,23,40]>,
indices: [0:1, 0:23, 39:40]
|ttnn.reshape|aten::slice.Tensor|4| +|11|Tensor<[1,23,40,128]>,
indices: [0:1, 0:23, 0:40, 0:128:2]
|ttnn.reshape|aten::slice.Tensor|4| +|12|Tensor<[1,23,40,128]>,
indices: [0:1, 0:23, 0:40, 1:128:2]
|ttnn.reshape|aten::slice.Tensor|4| +|13|Tensor<[768,256]>,
indices: [0:256, 0:256]
|ttnn.reshape|aten::slice.Tensor|4| +|14|Tensor<[768,256]>,
indices: [256:512, 0:256]
|ttnn.reshape|aten::slice.Tensor|4| +|15|Tensor<[768,256]>,
indices: [512:768, 0:256]
|ttnn.reshape|aten::slice.Tensor|4| +|16|Tensor<[768]>,
indices: [0:256]
|ttnn.reshape|aten::slice.Tensor|4| +|17|Tensor<[768]>,
indices: [256:512]
|ttnn.reshape|aten::slice.Tensor|4| +|18|Tensor<[768]>,
indices: [512:768]
|ttnn.reshape|aten::slice.Tensor|4| +|19|Tensor<[1,514]>,
indices: [0:1, 0:10]
|ttnn.reshape|aten::slice.Tensor|4| +|20|Tensor<[1,320]>,
indices: [0:1, 160:320]
|ttnn.reshape|aten::slice.Tensor|4| +|21|Tensor<[1,320]>,
indices: [0:1, 0:160]
|ttnn.reshape|aten::slice.Tensor|4| +|22|Tensor<[1,4096,2560]>,
indices: [0:1, 0:4096, 0:1280]
|ttnn.reshape|aten::slice.Tensor|4| +|23|Tensor<[1,4096,2560]>,
indices: [0:1, 0:4096, 1280:2560]
|ttnn.reshape|aten::slice.Tensor|4| +|24|Tensor<[1,1024,5120]>,
indices: [0:1, 0:1024, 0:2560]
|ttnn.reshape|aten::slice.Tensor|4| +|25|Tensor<[1,1024,5120]>,
indices: [0:1, 0:1024, 2560:5120]
|ttnn.reshape|aten::slice.Tensor|4| +|26|Tensor<[1,256,10240]>,
indices: [0:1, 0:256, 0:5120]
|ttnn.reshape|aten::slice.Tensor|4| +|27|Tensor<[1,256,10240]>,
indices: [0:1, 0:256, 5120:10240]
|ttnn.reshape|aten::slice.Tensor|4| +|28|Tensor<[1,64,10240]>,
indices: [0:1, 0:64, 0:5120]
|ttnn.reshape|aten::slice.Tensor|4| +|29|Tensor<[1,64,10240]>,
indices: [0:1, 0:64, 5120:10240]
|ttnn.reshape|aten::slice.Tensor|4| +|30|Tensor<[1,25,768]>,
indices: [0:1, 0:1, 0:768]
|ttnn.reshape|aten::select.int|4| +|31|Tensor<[1,512]>,
indices: [0:1, 0:25]
|ttnn.reshape|aten::slice.Tensor|4| +|32|Tensor<[1,25,2]>,
indices: [0:1, 0:25, 0:1]
|ttnn.reshape|aten::slice.Tensor|4| +|33|Tensor<[1,25,2]>,
indices: [0:1, 0:25, 1:2]
|ttnn.reshape|aten::slice.Tensor|4| +|34|Tensor<[1,4251,192]>,
indices: [0:1, 0:1, 0:192]
|ttnn.reshape|aten::select.int|4| +|35|Tensor<[1,4251,192]>,
indices: [0:1, 4151:4251, 0:192]
|ttnn.reshape|aten::slice.Tensor|4| +|36|Tensor<[1,4251,192]>,
indices: [0:1, 1:4151, 0:192]
|ttnn.reshape|aten::slice.Tensor|4| +|37|Tensor<[1,1445,192]>,
indices: [0:1, 1345:1445, 0:192]
|ttnn.reshape|aten::slice.Tensor|4| +|38|Tensor<[1,8,768]>,
indices: [0:1, 0:1, 0:768]
|ttnn.reshape|aten::select.int|4| +|39|Tensor<[1,512]>,
indices: [0:1, 0:8]
|ttnn.reshape|aten::slice.Tensor|4| +|40|Tensor<[1,16]>,
indices: [0:1, 0:1]
|ttnn.reshape|aten::select.int|4| +|41|Tensor<[1,12]>,
indices: [0:1, 0:1]
|ttnn.reshape|aten::select.int|4| +|42|Tensor<[192,2]>,
indices: [0:192, 0:1]
|ttnn.reshape|aten::select.int|4| +|43|Tensor<[1,201,768]>,
indices: [0:1, 0:1, 0:768]
|ttnn.reshape|aten::select.int|4| +|44|Tensor<[1,40]>,
indices: [0:1, 0:8]
|ttnn.reshape|aten::slice.Tensor|4| +|45|Tensor<[1,145,768]>,
indices: [0:1, 1:145, 0:768]
|ttnn.reshape|aten::slice.Tensor|4| +|46|Tensor<[1,19]>,
indices: [0:1, 18:19]
|ttnn.reshape|aten::select.int|4| +|47|Tensor<[1,19]>,
indices: [0:1, 1:19]
|ttnn.reshape|aten::slice.Tensor|4| +|48|Tensor<[1,19]>,
indices: [0:1, 0:18]
|ttnn.reshape|aten::slice.Tensor|4| +|49|Tensor<[1,32,16,3,96]>,
indices: [0:1, 0:32, 0:16, 0:1, 0:96]
|ttnn.reshape|aten::select.int|4| +|50|Tensor<[1,32,16,3,96]>,
indices: [0:1, 0:32, 0:16, 1:2, 0:96]
|ttnn.reshape|aten::select.int|4| +|51|Tensor<[1,32,16,3,96]>,
indices: [0:1, 0:32, 0:16, 2:3, 0:96]
|ttnn.reshape|aten::select.int|4| +|52|Tensor<[1,512]>,
indices: [0:1, 0:16]
|ttnn.reshape|aten::slice.Tensor|4| +|53|Tensor<[1,2,30,40]>,
indices: [0:1, 0:1, 0:30, 0:40]
|ttnn.reshape|aten::select.int|4| +|54|Tensor<[1,2,30,40]>,
indices: [0:1, 1:2, 0:30, 0:40]
|ttnn.reshape|aten::select.int|4| +|55|Tensor<[1,2,60,80]>,
indices: [0:1, 0:1, 0:60, 0:80]
|ttnn.reshape|aten::select.int|4| +|56|Tensor<[1,2,60,80]>,
indices: [0:1, 1:2, 0:60, 0:80]
|ttnn.reshape|aten::select.int|4| +|57|Tensor<[1,2,120,160]>,
indices: [0:1, 0:1, 0:120, 0:160]
|ttnn.reshape|aten::select.int|4| +|58|Tensor<[1,2,120,160]>,
indices: [0:1, 1:2, 0:120, 0:160]
|ttnn.reshape|aten::select.int|4| +|59|Tensor<[1,197,768]>,
indices: [0:1, 0:1, 0:768]
|ttnn.reshape|aten::select.int|4| +|60|Tensor<[1,7,73,64]>,
indices: [0:1, 0:7, 0:71, 0:64]
|ttnn.reshape|aten::slice.Tensor|4| +|61|Tensor<[1,71,7,64]>,
indices: [0:1, 0:71, 0:7, 0:32]
|ttnn.reshape|aten::slice.Tensor|4| +|62|Tensor<[1,71,7,64]>,
indices: [0:1, 0:71, 0:7, 32:64]
|ttnn.reshape|aten::slice.Tensor|4| +|63|Tensor<[1,1,7,64]>,
indices: [0:1, 0:1, 0:7, 0:32]
|ttnn.reshape|aten::slice.Tensor|4| +|64|Tensor<[1,1,7,64]>,
indices: [0:1, 0:1, 0:7, 32:64]
|ttnn.reshape|aten::slice.Tensor|4| +|65|Tensor<[1,512]>,
indices: [0:1, 0:12]
|ttnn.reshape|aten::slice.Tensor|4| +|66|Tensor<[1,512]>,
indices: [0:1, 0:9]
|ttnn.reshape|aten::slice.Tensor|4| +|67|Tensor<[1,9,768]>,
indices: [0:1, 0:1, 0:768]
|ttnn.reshape|aten::select.int|4| +|68|Tensor<[1,512]>,
indices: [0:1, 0:14]
|ttnn.reshape|aten::slice.Tensor|4| +|69|Tensor<[1,14,2]>,
indices: [0:1, 0:14, 0:1]
|ttnn.reshape|aten::slice.Tensor|4| +|70|Tensor<[1,14,2]>,
indices: [0:1, 0:14, 1:2]
|ttnn.reshape|aten::slice.Tensor|4| +|71|Tensor<[1,50,768]>,
indices: [0:1, 0:1, 0:768]
|ttnn.reshape|aten::select.int|4| +|72|Tensor<[1,77]>,
indices: [0:1, 0:7]
|ttnn.reshape|aten::slice.Tensor|4| +|73|Tensor<[196,196,2]>,
indices: [0:196, 0:196, 0:1]
|ttnn.reshape|aten::select.int|4| +|74|Tensor<[196,196,2]>,
indices: [0:196, 0:196, 1:2]
|ttnn.reshape|aten::select.int|4| +|75|Tensor<[197,197]>,
indices: [0:1, 0:197]
|ttnn.reshape|aten::select.int|4| +|76|Tensor<[197,197]>,
indices: [0:197, 0:1]
|ttnn.reshape|aten::select.int|4| +|77|Tensor<[197]>,
indices: [0:1]
|ttnn.reshape|aten::select.int|4| +|78|Tensor<[732,16]>,
indices: [0:729, 0:16]
|ttnn.reshape|aten::slice.Tensor|4| +|79|Tensor<[732,16]>,
indices: [729:732, 0:16]
|ttnn.reshape|aten::slice.Tensor|4| +|80|Tensor<[197,197]>,
indices: [1:197, 0:197]
|ttnn.reshape|aten::slice.Tensor|4| +|81|Tensor<[196,197]>,
indices: [0:196, 1:197]
|ttnn.reshape|aten::slice.Tensor|4| +|82|Tensor<[1,197,1024]>,
indices: [0:1, 1:197, 0:1024]
|ttnn.reshape|aten::slice.Tensor|4| +|83|Tensor<[732,12]>,
indices: [0:729, 0:12]
|ttnn.reshape|aten::slice.Tensor|4| +|84|Tensor<[732,12]>,
indices: [729:732, 0:12]
|ttnn.reshape|aten::slice.Tensor|4| +|85|Tensor<[1,197,768]>,
indices: [0:1, 1:197, 0:768]
|ttnn.reshape|aten::slice.Tensor|4| diff --git a/docs/ops/stablehlo/stablehlo.sqrt.md b/docs/ops/stablehlo/stablehlo.sqrt.md new file mode 100644 index 00000000..9ef2c16a --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.sqrt.md @@ -0,0 +1,42 @@ + +### stablehlo.sqrt::ttnn.sqrt + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[32]>,
|ttnn.sqrt|aten::sqrt|5| +|1|Tensor<[64]>,
|ttnn.sqrt|aten::sqrt|5| +|2|Tensor<[128]>,
|ttnn.sqrt|aten::sqrt|5| +|3|Tensor<[256]>,
|ttnn.sqrt|aten::sqrt|5| +|4|Tensor<[512]>,
|ttnn.sqrt|aten::sqrt|5| +|5|Tensor<[1024]>,
|ttnn.sqrt|aten::sqrt|5| +|6|Tensor<[2048]>,
|ttnn.sqrt|aten::sqrt|5| +|7|Tensor<[14]>,
|ttnn.sqrt|aten::sqrt|5| +|8|Tensor<[24]>,
|ttnn.sqrt|aten::sqrt|5| +|9|Tensor<[40]>,
|ttnn.sqrt|aten::sqrt|5| +|10|Tensor<[68]>,
|ttnn.sqrt|aten::sqrt|5| +|11|Tensor<[16]>,
|ttnn.sqrt|aten::sqrt|5| +|12|Tensor<[28]>,
|ttnn.sqrt|aten::sqrt|5| +|13|Tensor<[46]>,
|ttnn.sqrt|aten::sqrt|5| +|14|Tensor<[78]>,
|ttnn.sqrt|aten::sqrt|5| +|15|Tensor<[134]>,
|ttnn.sqrt|aten::sqrt|5| +|16|Tensor<[20]>,
|ttnn.sqrt|aten::sqrt|5| +|17|Tensor<[34]>,
|ttnn.sqrt|aten::sqrt|5| +|18|Tensor<[58]>,
|ttnn.sqrt|aten::sqrt|5| +|19|Tensor<[98]>,
|ttnn.sqrt|aten::sqrt|5| +|20|Tensor<[168]>,
|ttnn.sqrt|aten::sqrt|5| +|21|Tensor<[320]>,
|ttnn.sqrt|aten::sqrt|5| +|22|Tensor<[116]>,
|ttnn.sqrt|aten::sqrt|5| +|23|Tensor<[196]>,
|ttnn.sqrt|aten::sqrt|5| +|24|Tensor<[334]>,
|ttnn.sqrt|aten::sqrt|5| +|25|Tensor<[640]>,
|ttnn.sqrt|aten::sqrt|5| +|26|Tensor<[160]>,
|ttnn.sqrt|aten::sqrt|5| +|27|Tensor<[272]>,
|ttnn.sqrt|aten::sqrt|5| +|28|Tensor<[462]>,
|ttnn.sqrt|aten::sqrt|5| +|29|Tensor<[96]>,
|ttnn.sqrt|aten::sqrt|5| +|30|Tensor<[144]>,
|ttnn.sqrt|aten::sqrt|5| +|31|Tensor<[192]>,
|ttnn.sqrt|aten::sqrt|5| +|32|Tensor<[384]>,
|ttnn.sqrt|aten::sqrt|5| +|33|Tensor<[576]>,
|ttnn.sqrt|aten::sqrt|5| +|34|Tensor<[960]>,
|ttnn.sqrt|aten::sqrt|5| +|35|Tensor<[1280]>,
|ttnn.sqrt|aten::sqrt|5| diff --git a/docs/ops/stablehlo/stablehlo.subtract.md b/docs/ops/stablehlo/stablehlo.subtract.md new file mode 100644 index 00000000..e08d054a --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.subtract.md @@ -0,0 +1,234 @@ + +### stablehlo.subtract::ttnn.subtract + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,32,32,32]>,
Tensor<[1,32,32,32]>,
|ttnn.subtract|aten::_safe_softmax|4| +|1|Tensor<[1,12,7,7]>,
Tensor<[1,12,7,7]>,
|ttnn.subtract|aten::_safe_softmax|4| +|2|Tensor<[1,1,7,7]>,
Tensor<[1,1,7,7]>,
|ttnn.subtract|aten::rsub.Scalar|4| +|3|Tensor<[1,7,768]>,
Tensor<[1,7,768]>,
|ttnn.subtract|aten::sub.Tensor|4| +|4|Tensor<[1]>,
Tensor<[1]>,
|ttnn.subtract|aten::sub.Tensor|4| +|5|Tensor<[1,128,28,28]>,
Tensor<[1,128,28,28]>,
|ttnn.subtract|aten::elu|4| +|6|Tensor<[1,32,112,112]>,
Tensor<[1,32,112,112]>,
|ttnn.subtract|aten::sub.Tensor|4| +|7|Tensor<[1,64,112,112]>,
Tensor<[1,64,112,112]>,
|ttnn.subtract|aten::sub.Tensor|4| +|8|Tensor<[1,64,56,56]>,
Tensor<[1,64,56,56]>,
|ttnn.subtract|aten::sub.Tensor|4| +|9|Tensor<[1,128,56,56]>,
Tensor<[1,128,56,56]>,
|ttnn.subtract|aten::sub.Tensor|4| +|10|Tensor<[1,256,28,28]>,
Tensor<[1,256,28,28]>,
|ttnn.subtract|aten::sub.Tensor|4| +|11|Tensor<[1,512,28,28]>,
Tensor<[1,512,28,28]>,
|ttnn.subtract|aten::sub.Tensor|4| +|12|Tensor<[1,256,512]>,
Tensor<[1,256,512]>,
|ttnn.subtract|aten::sub.Tensor|4| +|13|Tensor<[8,920,920]>,
Tensor<[8,920,920]>,
|ttnn.subtract|aten::_softmax|4| +|14|Tensor<[8,100,100]>,
Tensor<[8,100,100]>,
|ttnn.subtract|aten::_softmax|4| +|15|Tensor<[8,100,920]>,
Tensor<[8,100,920]>,
|ttnn.subtract|aten::_softmax|4| +|16|Scalar,
Scalar,
|ttnn.subtract|aten::arange|4| +|17|Tensor<[1,64,1,1]>,
Tensor<[1,64,1,1]>,
|ttnn.subtract|aten::sub.Tensor|5| +|18|Tensor<[1,256,1,1]>,
Tensor<[1,256,1,1]>,
|ttnn.subtract|aten::sub.Tensor|5| +|19|Tensor<[1,128,1,1]>,
Tensor<[1,128,1,1]>,
|ttnn.subtract|aten::sub.Tensor|5| +|20|Tensor<[1,512,1,1]>,
Tensor<[1,512,1,1]>,
|ttnn.subtract|aten::sub.Tensor|5| +|21|Tensor<[1,1024,1,1]>,
Tensor<[1,1024,1,1]>,
|ttnn.subtract|aten::sub.Tensor|5| +|22|Tensor<[1,2048,1,1]>,
Tensor<[1,2048,1,1]>,
|ttnn.subtract|aten::sub.Tensor|5| +|23|Tensor<[920,1,256]>,
Tensor<[920,1,256]>,
|ttnn.subtract|aten::sub.Tensor|4| +|24|Tensor<[100,1,256]>,
Tensor<[100,1,256]>,
|ttnn.subtract|aten::sub.Tensor|4| +|25|Tensor<[1,12,10,10]>,
Tensor<[1,12,10,10]>,
|ttnn.subtract|aten::_safe_softmax|4| +|26|Tensor<[1,1,10,10]>,
Tensor<[1,1,10,10]>,
|ttnn.subtract|aten::rsub.Scalar|4| +|27|Tensor<[1,10,768]>,
Tensor<[1,10,768]>,
|ttnn.subtract|aten::sub.Tensor|4| +|28|Tensor<[1,8,4096,4096]>,
Tensor<[1,8,4096,4096]>,
|ttnn.subtract|aten::_safe_softmax|4| +|29|Tensor<[1,8,4096,9]>,
Tensor<[1,8,4096,9]>,
|ttnn.subtract|aten::_safe_softmax|4| +|30|Tensor<[1,8,1024,1024]>,
Tensor<[1,8,1024,1024]>,
|ttnn.subtract|aten::_safe_softmax|4| +|31|Tensor<[1,8,1024,9]>,
Tensor<[1,8,1024,9]>,
|ttnn.subtract|aten::_safe_softmax|4| +|32|Tensor<[1,8,256,256]>,
Tensor<[1,8,256,256]>,
|ttnn.subtract|aten::_safe_softmax|4| +|33|Tensor<[1,8,256,9]>,
Tensor<[1,8,256,9]>,
|ttnn.subtract|aten::_safe_softmax|4| +|34|Tensor<[1,8,64,64]>,
Tensor<[1,8,64,64]>,
|ttnn.subtract|aten::_safe_softmax|4| +|35|Tensor<[1,8,64,9]>,
Tensor<[1,8,64,9]>,
|ttnn.subtract|aten::_safe_softmax|4| +|36|Tensor<[1,32,10,4096]>,
Tensor<[1,32,10,4096]>,
|ttnn.subtract|aten::sub.Tensor|4| +|37|Tensor<[1,4096,320]>,
Tensor<[1,4096,320]>,
|ttnn.subtract|aten::sub.Tensor|4| +|38|Tensor<[1,32,10,1024]>,
Tensor<[1,32,10,1024]>,
|ttnn.subtract|aten::sub.Tensor|4| +|39|Tensor<[1,32,20,1024]>,
Tensor<[1,32,20,1024]>,
|ttnn.subtract|aten::sub.Tensor|4| +|40|Tensor<[1,1024,640]>,
Tensor<[1,1024,640]>,
|ttnn.subtract|aten::sub.Tensor|4| +|41|Tensor<[1,32,20,256]>,
Tensor<[1,32,20,256]>,
|ttnn.subtract|aten::sub.Tensor|4| +|42|Tensor<[1,32,40,256]>,
Tensor<[1,32,40,256]>,
|ttnn.subtract|aten::sub.Tensor|4| +|43|Tensor<[1,256,1280]>,
Tensor<[1,256,1280]>,
|ttnn.subtract|aten::sub.Tensor|4| +|44|Tensor<[1,32,40,64]>,
Tensor<[1,32,40,64]>,
|ttnn.subtract|aten::sub.Tensor|4| +|45|Tensor<[1,64,1280]>,
Tensor<[1,64,1280]>,
|ttnn.subtract|aten::sub.Tensor|4| +|46|Tensor<[1,32,80,64]>,
Tensor<[1,32,80,64]>,
|ttnn.subtract|aten::sub.Tensor|4| +|47|Tensor<[1,32,80,256]>,
Tensor<[1,32,80,256]>,
|ttnn.subtract|aten::sub.Tensor|4| +|48|Tensor<[1,32,60,256]>,
Tensor<[1,32,60,256]>,
|ttnn.subtract|aten::sub.Tensor|4| +|49|Tensor<[1,32,60,1024]>,
Tensor<[1,32,60,1024]>,
|ttnn.subtract|aten::sub.Tensor|4| +|50|Tensor<[1,32,40,1024]>,
Tensor<[1,32,40,1024]>,
|ttnn.subtract|aten::sub.Tensor|4| +|51|Tensor<[1,32,30,1024]>,
Tensor<[1,32,30,1024]>,
|ttnn.subtract|aten::sub.Tensor|4| +|52|Tensor<[1,32,30,4096]>,
Tensor<[1,32,30,4096]>,
|ttnn.subtract|aten::sub.Tensor|4| +|53|Tensor<[1,32,20,4096]>,
Tensor<[1,32,20,4096]>,
|ttnn.subtract|aten::sub.Tensor|4| +|54|Tensor<[1,12,25,25]>,
Tensor<[1,12,25,25]>,
|ttnn.subtract|aten::_safe_softmax|4| +|55|Tensor<[1,1,25,25]>,
Tensor<[1,1,25,25]>,
|ttnn.subtract|aten::rsub.Scalar|4| +|56|Tensor<[1,25,768]>,
Tensor<[1,25,768]>,
|ttnn.subtract|aten::sub.Tensor|4| +|57|Tensor<[1,3,1445,1445]>,
Tensor<[1,3,1445,1445]>,
|ttnn.subtract|aten::_safe_softmax|4| +|58|Tensor<[1,1445,192]>,
Tensor<[1,1445,192]>,
|ttnn.subtract|aten::sub.Tensor|4| +|59|Tensor<[1,256,14,14]>,
Tensor<[1,256,14,14]>,
|ttnn.subtract|aten::sub.Tensor|4| +|60|Tensor<[1,512,7,7]>,
Tensor<[1,512,7,7]>,
|ttnn.subtract|aten::sub.Tensor|4| +|61|Tensor<[1,12,8,8]>,
Tensor<[1,12,8,8]>,
|ttnn.subtract|aten::_softmax|4| +|62|Tensor<[1,1,1,8]>,
Tensor<[1,1,1,8]>,
|ttnn.subtract|aten::rsub.Scalar|4| +|63|Tensor<[1,8,768]>,
Tensor<[1,8,768]>,
|ttnn.subtract|aten::sub.Tensor|4| +|64|Tensor<[1,8,256,2048]>,
Tensor<[1,8,256,2048]>,
|ttnn.subtract|aten::_softmax|4| +|65|Tensor<[1,8,2048,256]>,
Tensor<[1,8,2048,256]>,
|ttnn.subtract|aten::_softmax|4| +|66|Tensor<[1,1,1,2048]>,
Tensor<[1,1,1,2048]>,
|ttnn.subtract|aten::rsub.Scalar|4| +|67|Tensor<[1,2048,768]>,
Tensor<[1,2048,768]>,
|ttnn.subtract|aten::sub.Tensor|4| +|68|Tensor<[1,256,56,56]>,
Tensor<[1,256,56,56]>,
|ttnn.subtract|aten::sub.Tensor|4| +|69|Tensor<[1,1024,14,14]>,
Tensor<[1,1024,14,14]>,
|ttnn.subtract|aten::sub.Tensor|4| +|70|Tensor<[1,512,14,14]>,
Tensor<[1,512,14,14]>,
|ttnn.subtract|aten::sub.Tensor|4| +|71|Tensor<[1,2048,7,7]>,
Tensor<[1,2048,7,7]>,
|ttnn.subtract|aten::sub.Tensor|4| +|72|Tensor<[1,12,201,201]>,
Tensor<[1,12,201,201]>,
|ttnn.subtract|aten::_softmax|4| +|73|Tensor<[1,192]>,
Tensor<[1,192]>,
|ttnn.subtract|aten::rsub.Scalar|4| +|74|Tensor<[1,1,1,201]>,
Tensor<[1,1,1,201]>,
|ttnn.subtract|aten::rsub.Scalar|4| +|75|Tensor<[1,201,768]>,
Tensor<[1,201,768]>,
|ttnn.subtract|aten::sub.Tensor|4| +|76|Tensor<[1,1536]>,
Tensor<[1,1536]>,
|ttnn.subtract|aten::sub.Tensor|4| +|77|Tensor<[1,10]>,
Tensor<[1,10]>,
|ttnn.subtract|aten::sub.Tensor|4| +|78|Tensor<[16,19,19]>,
Tensor<[16,19,19]>,
|ttnn.subtract|aten::_softmax|4| +|79|Tensor<[1,1,19,19]>,
Tensor<[1,1,19,19]>,
|ttnn.subtract|aten::rsub.Scalar|4| +|80|Tensor<[1,19,1024]>,
Tensor<[1,19,1024]>,
|ttnn.subtract|aten::sub.Tensor|4| +|81|Tensor<[19]>,
Tensor<[19]>,
|ttnn.subtract|aten::sub.Tensor|4| +|82|Tensor<[19,256008]>,
Tensor<[19,256008]>,
|ttnn.subtract|aten::sub.Tensor|4| +|83|Tensor<[1,14,56,56]>,
Tensor<[1,14,56,56]>,
|ttnn.subtract|aten::sub.Tensor|4| +|84|Tensor<[1,24,56,56]>,
Tensor<[1,24,56,56]>,
|ttnn.subtract|aten::sub.Tensor|4| +|85|Tensor<[1,40,56,56]>,
Tensor<[1,40,56,56]>,
|ttnn.subtract|aten::sub.Tensor|4| +|86|Tensor<[1,68,56,56]>,
Tensor<[1,68,56,56]>,
|ttnn.subtract|aten::sub.Tensor|4| +|87|Tensor<[1,16,28,28]>,
Tensor<[1,16,28,28]>,
|ttnn.subtract|aten::sub.Tensor|4| +|88|Tensor<[1,28,28,28]>,
Tensor<[1,28,28,28]>,
|ttnn.subtract|aten::sub.Tensor|4| +|89|Tensor<[1,46,28,28]>,
Tensor<[1,46,28,28]>,
|ttnn.subtract|aten::sub.Tensor|4| +|90|Tensor<[1,78,28,28]>,
Tensor<[1,78,28,28]>,
|ttnn.subtract|aten::sub.Tensor|4| +|91|Tensor<[1,134,28,28]>,
Tensor<[1,134,28,28]>,
|ttnn.subtract|aten::sub.Tensor|4| +|92|Tensor<[1,20,28,28]>,
Tensor<[1,20,28,28]>,
|ttnn.subtract|aten::sub.Tensor|4| +|93|Tensor<[1,34,28,28]>,
Tensor<[1,34,28,28]>,
|ttnn.subtract|aten::sub.Tensor|4| +|94|Tensor<[1,58,28,28]>,
Tensor<[1,58,28,28]>,
|ttnn.subtract|aten::sub.Tensor|4| +|95|Tensor<[1,98,28,28]>,
Tensor<[1,98,28,28]>,
|ttnn.subtract|aten::sub.Tensor|4| +|96|Tensor<[1,168,28,28]>,
Tensor<[1,168,28,28]>,
|ttnn.subtract|aten::sub.Tensor|4| +|97|Tensor<[1,320,28,28]>,
Tensor<[1,320,28,28]>,
|ttnn.subtract|aten::sub.Tensor|4| +|98|Tensor<[1,40,14,14]>,
Tensor<[1,40,14,14]>,
|ttnn.subtract|aten::sub.Tensor|4| +|99|Tensor<[1,68,14,14]>,
Tensor<[1,68,14,14]>,
|ttnn.subtract|aten::sub.Tensor|4| +|100|Tensor<[1,116,14,14]>,
Tensor<[1,116,14,14]>,
|ttnn.subtract|aten::sub.Tensor|4| +|101|Tensor<[1,196,14,14]>,
Tensor<[1,196,14,14]>,
|ttnn.subtract|aten::sub.Tensor|4| +|102|Tensor<[1,334,14,14]>,
Tensor<[1,334,14,14]>,
|ttnn.subtract|aten::sub.Tensor|4| +|103|Tensor<[1,640,14,14]>,
Tensor<[1,640,14,14]>,
|ttnn.subtract|aten::sub.Tensor|4| +|104|Tensor<[1,160,7,7]>,
Tensor<[1,160,7,7]>,
|ttnn.subtract|aten::sub.Tensor|4| +|105|Tensor<[1,272,7,7]>,
Tensor<[1,272,7,7]>,
|ttnn.subtract|aten::sub.Tensor|4| +|106|Tensor<[1,462,7,7]>,
Tensor<[1,462,7,7]>,
|ttnn.subtract|aten::sub.Tensor|4| +|107|Tensor<[1,1024,7,7]>,
Tensor<[1,1024,7,7]>,
|ttnn.subtract|aten::sub.Tensor|4| +|108|Tensor<[1,32,512,512]>,
Tensor<[1,32,512,512]>,
|ttnn.subtract|aten::sub.Tensor|4| +|109|Tensor<[1,64,256,256]>,
Tensor<[1,64,256,256]>,
|ttnn.subtract|aten::sub.Tensor|4| +|110|Tensor<[1,32,256,256]>,
Tensor<[1,32,256,256]>,
|ttnn.subtract|aten::sub.Tensor|4| +|111|Tensor<[1,128,128,128]>,
Tensor<[1,128,128,128]>,
|ttnn.subtract|aten::sub.Tensor|4| +|112|Tensor<[1,64,128,128]>,
Tensor<[1,64,128,128]>,
|ttnn.subtract|aten::sub.Tensor|4| +|113|Tensor<[1,256,64,64]>,
Tensor<[1,256,64,64]>,
|ttnn.subtract|aten::sub.Tensor|4| +|114|Tensor<[1,128,64,64]>,
Tensor<[1,128,64,64]>,
|ttnn.subtract|aten::sub.Tensor|4| +|115|Tensor<[1,512,32,32]>,
Tensor<[1,512,32,32]>,
|ttnn.subtract|aten::sub.Tensor|4| +|116|Tensor<[1,256,32,32]>,
Tensor<[1,256,32,32]>,
|ttnn.subtract|aten::sub.Tensor|4| +|117|Tensor<[1,1024,16,16]>,
Tensor<[1,1024,16,16]>,
|ttnn.subtract|aten::sub.Tensor|4| +|118|Tensor<[1,512,16,16]>,
Tensor<[1,512,16,16]>,
|ttnn.subtract|aten::sub.Tensor|4| +|119|Tensor<[1,256,16,16]>,
Tensor<[1,256,16,16]>,
|ttnn.subtract|aten::sub.Tensor|4| +|120|Tensor<[1,128,32,32]>,
Tensor<[1,128,32,32]>,
|ttnn.subtract|aten::sub.Tensor|4| +|121|Tensor<[1,16,32,32]>,
Tensor<[1,16,32,32]>,
|ttnn.subtract|aten::_softmax|4| +|122|Tensor<[1,32,1536]>,
Tensor<[1,32,1536]>,
|ttnn.subtract|aten::sub.Tensor|4| +|123|Tensor<[1,32]>,
Tensor<[1,32]>,
|ttnn.subtract|aten::sub.Tensor|4| +|124|Tensor<[1,12,16,16]>,
Tensor<[1,12,16,16]>,
|ttnn.subtract|aten::_safe_softmax|4| +|125|Tensor<[1,1,16,16]>,
Tensor<[1,1,16,16]>,
|ttnn.subtract|aten::rsub.Scalar|4| +|126|Tensor<[1,16,768]>,
Tensor<[1,16,768]>,
|ttnn.subtract|aten::sub.Tensor|4| +|127|Tensor<[1,64,224,224]>,
Tensor<[1,64,224,224]>,
|ttnn.subtract|aten::sub.Tensor|4| +|128|Tensor<[1,128,112,112]>,
Tensor<[1,128,112,112]>,
|ttnn.subtract|aten::sub.Tensor|4| +|129|Tensor<[1,1,19200,300]>,
Tensor<[1,1,19200,300]>,
|ttnn.subtract|aten::_softmax|4| +|130|Tensor<[1,2,4800,300]>,
Tensor<[1,2,4800,300]>,
|ttnn.subtract|aten::_softmax|4| +|131|Tensor<[1,5,1200,300]>,
Tensor<[1,5,1200,300]>,
|ttnn.subtract|aten::_softmax|4| +|132|Tensor<[1,8,300,300]>,
Tensor<[1,8,300,300]>,
|ttnn.subtract|aten::_softmax|4| +|133|Tensor<[1,19200,64]>,
Tensor<[1,19200,64]>,
|ttnn.subtract|aten::sub.Tensor|4| +|134|Tensor<[1,300,64]>,
Tensor<[1,300,64]>,
|ttnn.subtract|aten::sub.Tensor|4| +|135|Tensor<[1,4800,128]>,
Tensor<[1,4800,128]>,
|ttnn.subtract|aten::sub.Tensor|4| +|136|Tensor<[1,300,128]>,
Tensor<[1,300,128]>,
|ttnn.subtract|aten::sub.Tensor|4| +|137|Tensor<[1,1200,320]>,
Tensor<[1,1200,320]>,
|ttnn.subtract|aten::sub.Tensor|4| +|138|Tensor<[1,300,320]>,
Tensor<[1,300,320]>,
|ttnn.subtract|aten::sub.Tensor|4| +|139|Tensor<[1,300,512]>,
Tensor<[1,300,512]>,
|ttnn.subtract|aten::sub.Tensor|4| +|140|Tensor<[30]>,
Tensor<[30]>,
|ttnn.subtract|aten::sub.Tensor|4| +|141|Tensor<[40]>,
Tensor<[40]>,
|ttnn.subtract|aten::sub.Tensor|4| +|142|Tensor<[1,64,30,40]>,
Tensor<[1,64,30,40]>,
|ttnn.subtract|aten::sub.Tensor|5| +|143|Tensor<[30,1]>,
Tensor<[30,1]>,
|ttnn.subtract|aten::sub.Tensor|4| +|144|Tensor<[1,32,30,40]>,
Tensor<[1,32,30,40]>,
|ttnn.subtract|aten::sub.Tensor|4| +|145|Tensor<[60]>,
Tensor<[60]>,
|ttnn.subtract|aten::sub.Tensor|4| +|146|Tensor<[80]>,
Tensor<[80]>,
|ttnn.subtract|aten::sub.Tensor|4| +|147|Tensor<[1,64,60,80]>,
Tensor<[1,64,60,80]>,
|ttnn.subtract|aten::sub.Tensor|5| +|148|Tensor<[60,1]>,
Tensor<[60,1]>,
|ttnn.subtract|aten::sub.Tensor|4| +|149|Tensor<[1,32,60,80]>,
Tensor<[1,32,60,80]>,
|ttnn.subtract|aten::sub.Tensor|4| +|150|Tensor<[120]>,
Tensor<[120]>,
|ttnn.subtract|aten::sub.Tensor|4| +|151|Tensor<[160]>,
Tensor<[160]>,
|ttnn.subtract|aten::sub.Tensor|4| +|152|Tensor<[1,64,120,160]>,
Tensor<[1,64,120,160]>,
|ttnn.subtract|aten::sub.Tensor|5| +|153|Tensor<[120,1]>,
Tensor<[120,1]>,
|ttnn.subtract|aten::sub.Tensor|4| +|154|Tensor<[1,32,120,160]>,
Tensor<[1,32,120,160]>,
|ttnn.subtract|aten::sub.Tensor|4| +|155|Tensor<[240]>,
Tensor<[240]>,
|ttnn.subtract|aten::sub.Tensor|4| +|156|Tensor<[320]>,
Tensor<[320]>,
|ttnn.subtract|aten::sub.Tensor|4| +|157|Tensor<[1,64,240,320]>,
Tensor<[1,64,240,320]>,
|ttnn.subtract|aten::sub.Tensor|5| +|158|Tensor<[240,1]>,
Tensor<[240,1]>,
|ttnn.subtract|aten::sub.Tensor|4| +|159|Tensor<[480]>,
Tensor<[480]>,
|ttnn.subtract|aten::sub.Tensor|4| +|160|Tensor<[640]>,
Tensor<[640]>,
|ttnn.subtract|aten::sub.Tensor|4| +|161|Tensor<[1,64,480,640]>,
Tensor<[1,64,480,640]>,
|ttnn.subtract|aten::sub.Tensor|5| +|162|Tensor<[480,1]>,
Tensor<[480,1]>,
|ttnn.subtract|aten::sub.Tensor|4| +|163|Tensor<[1,12,197,197]>,
Tensor<[1,12,197,197]>,
|ttnn.subtract|aten::_safe_softmax|4| +|164|Tensor<[1,197,768]>,
Tensor<[1,197,768]>,
|ttnn.subtract|aten::sub.Tensor|4| +|165|Tensor<[1,1,16384,256]>,
Tensor<[1,1,16384,256]>,
|ttnn.subtract|aten::_softmax|4| +|166|Tensor<[1,2,4096,256]>,
Tensor<[1,2,4096,256]>,
|ttnn.subtract|aten::_softmax|4| +|167|Tensor<[1,5,1024,256]>,
Tensor<[1,5,1024,256]>,
|ttnn.subtract|aten::_softmax|4| +|168|Tensor<[1,16384,32]>,
Tensor<[1,16384,32]>,
|ttnn.subtract|aten::sub.Tensor|4| +|169|Tensor<[1,256,32]>,
Tensor<[1,256,32]>,
|ttnn.subtract|aten::sub.Tensor|4| +|170|Tensor<[1,4096,64]>,
Tensor<[1,4096,64]>,
|ttnn.subtract|aten::sub.Tensor|4| +|171|Tensor<[1,256,64]>,
Tensor<[1,256,64]>,
|ttnn.subtract|aten::sub.Tensor|4| +|172|Tensor<[1,1024,160]>,
Tensor<[1,1024,160]>,
|ttnn.subtract|aten::sub.Tensor|4| +|173|Tensor<[1,256,160]>,
Tensor<[1,256,160]>,
|ttnn.subtract|aten::sub.Tensor|4| +|174|Tensor<[1,256,256]>,
Tensor<[1,256,256]>,
|ttnn.subtract|aten::sub.Tensor|4| +|175|Tensor<[128]>,
Tensor<[128]>,
|ttnn.subtract|aten::sub.Tensor|4| +|176|Tensor<[1,256,128,128]>,
Tensor<[1,256,128,128]>,
|ttnn.subtract|aten::sub.Tensor|5| +|177|Tensor<[128,1]>,
Tensor<[128,1]>,
|ttnn.subtract|aten::sub.Tensor|4| +|178|Tensor<[1,71,7,7]>,
Tensor<[1,71,7,7]>,
|ttnn.subtract|aten::_safe_softmax|4| +|179|Tensor<[1,7,4544]>,
Tensor<[1,7,4544]>,
|ttnn.subtract|aten::sub.Tensor|4| +|180|Tensor<[1,16,112,112]>,
Tensor<[1,16,112,112]>,
|ttnn.subtract|aten::sub.Tensor|4| +|181|Tensor<[1,96,112,112]>,
Tensor<[1,96,112,112]>,
|ttnn.subtract|aten::sub.Tensor|4| +|182|Tensor<[1,96,56,56]>,
Tensor<[1,96,56,56]>,
|ttnn.subtract|aten::sub.Tensor|4| +|183|Tensor<[1,144,56,56]>,
Tensor<[1,144,56,56]>,
|ttnn.subtract|aten::sub.Tensor|4| +|184|Tensor<[1,144,28,28]>,
Tensor<[1,144,28,28]>,
|ttnn.subtract|aten::sub.Tensor|4| +|185|Tensor<[1,32,28,28]>,
Tensor<[1,32,28,28]>,
|ttnn.subtract|aten::sub.Tensor|4| +|186|Tensor<[1,192,28,28]>,
Tensor<[1,192,28,28]>,
|ttnn.subtract|aten::sub.Tensor|4| +|187|Tensor<[1,192,14,14]>,
Tensor<[1,192,14,14]>,
|ttnn.subtract|aten::sub.Tensor|4| +|188|Tensor<[1,64,14,14]>,
Tensor<[1,64,14,14]>,
|ttnn.subtract|aten::sub.Tensor|4| +|189|Tensor<[1,384,14,14]>,
Tensor<[1,384,14,14]>,
|ttnn.subtract|aten::sub.Tensor|4| +|190|Tensor<[1,96,14,14]>,
Tensor<[1,96,14,14]>,
|ttnn.subtract|aten::sub.Tensor|4| +|191|Tensor<[1,576,14,14]>,
Tensor<[1,576,14,14]>,
|ttnn.subtract|aten::sub.Tensor|4| +|192|Tensor<[1,576,7,7]>,
Tensor<[1,576,7,7]>,
|ttnn.subtract|aten::sub.Tensor|4| +|193|Tensor<[1,960,7,7]>,
Tensor<[1,960,7,7]>,
|ttnn.subtract|aten::sub.Tensor|4| +|194|Tensor<[1,320,7,7]>,
Tensor<[1,320,7,7]>,
|ttnn.subtract|aten::sub.Tensor|4| +|195|Tensor<[1,1280,7,7]>,
Tensor<[1,1280,7,7]>,
|ttnn.subtract|aten::sub.Tensor|4| +|196|Tensor<[1,12,12,12]>,
Tensor<[1,12,12,12]>,
|ttnn.subtract|aten::_safe_softmax|4| +|197|Tensor<[1,1,12,12]>,
Tensor<[1,1,12,12]>,
|ttnn.subtract|aten::rsub.Scalar|4| +|198|Tensor<[1,12,128]>,
Tensor<[1,12,128]>,
|ttnn.subtract|aten::sub.Tensor|4| +|199|Tensor<[1,12,768]>,
Tensor<[1,12,768]>,
|ttnn.subtract|aten::sub.Tensor|4| +|200|Tensor<[1,12,9,9]>,
Tensor<[1,12,9,9]>,
|ttnn.subtract|aten::_safe_softmax|4| +|201|Tensor<[1,1,9,9]>,
Tensor<[1,1,9,9]>,
|ttnn.subtract|aten::rsub.Scalar|4| +|202|Tensor<[1,9,128]>,
Tensor<[1,9,128]>,
|ttnn.subtract|aten::sub.Tensor|4| +|203|Tensor<[1,9,768]>,
Tensor<[1,9,768]>,
|ttnn.subtract|aten::sub.Tensor|4| +|204|Tensor<[1,16,9,9]>,
Tensor<[1,16,9,9]>,
|ttnn.subtract|aten::_safe_softmax|4| +|205|Tensor<[1,9,2048]>,
Tensor<[1,9,2048]>,
|ttnn.subtract|aten::sub.Tensor|4| +|206|Tensor<[1,9,1024]>,
Tensor<[1,9,1024]>,
|ttnn.subtract|aten::sub.Tensor|4| +|207|Tensor<[1,64,9,9]>,
Tensor<[1,64,9,9]>,
|ttnn.subtract|aten::_safe_softmax|4| +|208|Tensor<[1,9,4096]>,
Tensor<[1,9,4096]>,
|ttnn.subtract|aten::sub.Tensor|4| +|209|Tensor<[1,12,14,14]>,
Tensor<[1,12,14,14]>,
|ttnn.subtract|aten::_safe_softmax|4| +|210|Tensor<[1,1,14,14]>,
Tensor<[1,1,14,14]>,
|ttnn.subtract|aten::rsub.Scalar|4| +|211|Tensor<[1,14,128]>,
Tensor<[1,14,128]>,
|ttnn.subtract|aten::sub.Tensor|4| +|212|Tensor<[1,14,768]>,
Tensor<[1,14,768]>,
|ttnn.subtract|aten::sub.Tensor|4| +|213|Tensor<[1,12,50,50]>,
Tensor<[1,12,50,50]>,
|ttnn.subtract|aten::_safe_softmax|4| +|214|Tensor<[2,8,7,7]>,
Tensor<[2,8,7,7]>,
|ttnn.subtract|aten::_safe_softmax|4| +|215|Tensor<[2,1,7,7]>,
Tensor<[2,1,7,7]>,
|ttnn.subtract|aten::rsub.Scalar|4| +|216|Tensor<[1,50,768]>,
Tensor<[1,50,768]>,
|ttnn.subtract|aten::sub.Tensor|4| +|217|Tensor<[1,768]>,
Tensor<[1,768]>,
|ttnn.subtract|aten::sub.Tensor|4| +|218|Tensor<[2,7,512]>,
Tensor<[2,7,512]>,
|ttnn.subtract|aten::sub.Tensor|4| +|219|Tensor<[1,16,197,197]>,
Tensor<[1,16,197,197]>,
|ttnn.subtract|aten::_softmax|4| +|220|Tensor<[1,197,1024]>,
Tensor<[1,197,1024]>,
|ttnn.subtract|aten::sub.Tensor|4| +|221|Tensor<[27]>,
Tensor<[27]>,
|ttnn.subtract|aten::sub.Tensor|4| +|222|Tensor<[1,16,27,27]>,
Tensor<[1,16,27,27]>,
|ttnn.subtract|aten::sub.Tensor|5| +|223|Tensor<[27,1]>,
Tensor<[27,1]>,
|ttnn.subtract|aten::sub.Tensor|4| +|224|Tensor<[2,196,196]>,
Tensor<[2,196,196]>,
|ttnn.subtract|aten::sub.Tensor|4| +|225|Tensor<[197]>,
Tensor<[197]>,
|ttnn.subtract|aten::sub.Tensor|4| +|226|Tensor<[1,1024]>,
Tensor<[1,1024]>,
|ttnn.subtract|aten::sub.Tensor|4| +|227|Tensor<[1,12,27,27]>,
Tensor<[1,12,27,27]>,
|ttnn.subtract|aten::sub.Tensor|5| diff --git a/docs/ops/stablehlo/stablehlo.tanh.md b/docs/ops/stablehlo/stablehlo.tanh.md new file mode 100644 index 00000000..deb15deb --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.tanh.md @@ -0,0 +1,16 @@ + +### stablehlo.tanh::ttnn.tanh + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,7,3072]>,
|ttnn.tanh|aten::tanh|4| +|1|Tensor<[1,768]>,
|ttnn.tanh|aten::tanh|4| +|2|Tensor<[1,32,6144]>,
|ttnn.tanh|aten::tanh|4| +|3|Tensor<[1,12,3072]>,
|ttnn.tanh|aten::tanh|4| +|4|Tensor<[1,9,3072]>,
|ttnn.tanh|aten::tanh|4| +|5|Tensor<[1,9,128]>,
|ttnn.tanh|aten::tanh|4| +|6|Tensor<[1,9,8192]>,
|ttnn.tanh|aten::tanh|4| +|7|Tensor<[1,9,4096]>,
|ttnn.tanh|aten::tanh|4| +|8|Tensor<[1,9,16384]>,
|ttnn.tanh|aten::tanh|4| +|9|Tensor<[1,14,3072]>,
|ttnn.tanh|aten::tanh|4| diff --git a/docs/ops/stablehlo/stablehlo.transpose.md b/docs/ops/stablehlo/stablehlo.transpose.md new file mode 100644 index 00000000..206cb242 --- /dev/null +++ b/docs/ops/stablehlo/stablehlo.transpose.md @@ -0,0 +1,317 @@ + +### stablehlo.transpose::ttnn.permute + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[1,64,32]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|1|Tensor<[4096,4096]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|2|Tensor<[1,32,32,128]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|3|Tensor<[1,32,32,128]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|4|Tensor<[11008,4096]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|5|Tensor<[4096,11008]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|6|Tensor<[32000,4096]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|7|Tensor<[1,7,12,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|8|Tensor<[1,12,7,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|9|Tensor<[1,12,7,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|10|Tensor<[2,768]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|11|Tensor<[1,3,16,16,16,16]>,
dims: [0, 2, 4, 3, 5, 1]
|ttnn.permute|aten::permute|4| +|12|Tensor<[1,256,512]>,
dims: [0, 2, 1]
|ttnn.permute|aten::permute|4| +|13|Tensor<[512,768]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|14|Tensor<[256,512]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|15|Tensor<[512,256]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|16|Tensor<[1000,512]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|17|Tensor<[1,23,40,256]>,
dims: [0, 3, 1, 2]
|ttnn.permute|aten::permute|4| +|18|Tensor<[1,256,920]>,
dims: [2, 0, 1]
|ttnn.permute|aten::permute|4| +|19|Tensor<[256,256]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|20|Tensor<[920,8,32]>,
dims: [1, 0, 2]
|ttnn.permute|aten::transpose.int|4| +|21|Tensor<[8,920,32]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|22|Tensor<[8,920,32]>,
dims: [1, 0, 2]
|ttnn.permute|aten::transpose.int|4| +|23|Tensor<[2048,256]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|24|Tensor<[256,2048]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|25|Tensor<[100,8,32]>,
dims: [1, 0, 2]
|ttnn.permute|aten::transpose.int|4| +|26|Tensor<[8,100,32]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|27|Tensor<[8,100,32]>,
dims: [1, 0, 2]
|ttnn.permute|aten::transpose.int|4| +|28|Tensor<[6,100,1,256]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|29|Tensor<[92,256]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|30|Tensor<[4,256]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|31|Tensor<[1,10,12,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|32|Tensor<[768,768]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|33|Tensor<[1,12,10,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|34|Tensor<[1,12,10,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|35|Tensor<[3072,768]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|36|Tensor<[768,3072]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|37|Tensor<[250002,768]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|38|Tensor<[1,320,64,64]>,
dims: [0, 2, 3, 1]
|ttnn.permute|aten::permute|4| +|39|Tensor<[1,64,64,320]>,
dims: [0, 3, 1, 2]
|ttnn.permute|aten::permute|4| +|40|Tensor<[1,640,32,32]>,
dims: [0, 2, 3, 1]
|ttnn.permute|aten::permute|4| +|41|Tensor<[1,32,32,640]>,
dims: [0, 3, 1, 2]
|ttnn.permute|aten::permute|4| +|42|Tensor<[1,1280,16,16]>,
dims: [0, 2, 3, 1]
|ttnn.permute|aten::permute|4| +|43|Tensor<[1,16,16,1280]>,
dims: [0, 3, 1, 2]
|ttnn.permute|aten::permute|4| +|44|Tensor<[1,1280,8,8]>,
dims: [0, 2, 3, 1]
|ttnn.permute|aten::permute|4| +|45|Tensor<[1,8,8,1280]>,
dims: [0, 3, 1, 2]
|ttnn.permute|aten::permute|4| +|46|Tensor<[1280,320]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|47|Tensor<[1280,1280]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|48|Tensor<[320,1280]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|49|Tensor<[320,320]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|50|Tensor<[1,4096,8,40]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|51|Tensor<[1,8,4096,40]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|52|Tensor<[1,8,4096,40]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|53|Tensor<[320,768]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|54|Tensor<[1,9,8,40]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|55|Tensor<[1,8,9,40]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|56|Tensor<[2560,320]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|57|Tensor<[640,1280]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|58|Tensor<[640,640]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|59|Tensor<[1,1024,8,80]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|60|Tensor<[1,8,1024,80]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|61|Tensor<[1,8,1024,80]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|62|Tensor<[640,768]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|63|Tensor<[1,9,8,80]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|64|Tensor<[1,8,9,80]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|65|Tensor<[5120,640]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|66|Tensor<[640,2560]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|67|Tensor<[1,256,8,160]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|68|Tensor<[1,8,256,160]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|69|Tensor<[1,8,256,160]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|70|Tensor<[1280,768]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|71|Tensor<[1,9,8,160]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|72|Tensor<[1,8,9,160]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|73|Tensor<[10240,1280]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|74|Tensor<[1280,5120]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|75|Tensor<[1,64,8,160]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|76|Tensor<[1,8,64,160]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|77|Tensor<[1,8,64,160]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|78|Tensor<[1,25,12,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|79|Tensor<[1,12,25,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|80|Tensor<[1,12,25,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|81|Tensor<[1,768]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|82|Tensor<[1,1445,3,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|83|Tensor<[1,3,1445,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|84|Tensor<[1,192,1344]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|85|Tensor<[1,4150,192]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|86|Tensor<[192,192]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|87|Tensor<[1,3,1445,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|88|Tensor<[768,192]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|89|Tensor<[192,768]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|90|Tensor<[92,192]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|91|Tensor<[4,192]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|92|Tensor<[1,8,768]>,
dims: [0, 2, 1]
|ttnn.permute|aten::permute|4| +|93|Tensor<[1,12,64,8]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::permute|4| +|94|Tensor<[1,12,8,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::permute|4| +|95|Tensor<[1,768,8]>,
dims: [0, 2, 1]
|ttnn.permute|aten::permute|4| +|96|Tensor<[3,768]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|97|Tensor<[1,256,8,32]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|98|Tensor<[1,2048,8,32]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|99|Tensor<[1,2048,8,160]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|100|Tensor<[1,256,8,96]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|101|Tensor<[1,8,2048,96]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|102|Tensor<[256,1280]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|103|Tensor<[256,768]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|104|Tensor<[1,8,2048,32]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|105|Tensor<[1,8,256,32]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|106|Tensor<[768,1280]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|107|Tensor<[262,768]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|108|Tensor<[1000,2048]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|109|Tensor<[1,201,12,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|110|Tensor<[1,12,201,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|111|Tensor<[1,144,768]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|112|Tensor<[1,768,192]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|113|Tensor<[1,12,201,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|114|Tensor<[1536,768]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|115|Tensor<[3129,1536]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|116|Tensor<[128,9216]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|117|Tensor<[10,128]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|118|Tensor<[1024,1024]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|119|Tensor<[1,19,16,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|120|Tensor<[16,19,64]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|121|Tensor<[1,16,19,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|122|Tensor<[4096,1024]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|123|Tensor<[1024,4096]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|124|Tensor<[256008,1024]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|125|Tensor<[1000,1024]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|126|Tensor<[512,256,2,2]>,
dims: [2, 3, 1, 0]
|ttnn.permute|aten::convolution|4| +|127|Tensor<[256,128,2,2]>,
dims: [2, 3, 1, 0]
|ttnn.permute|aten::convolution|4| +|128|Tensor<[128,64,2,2]>,
dims: [2, 3, 1, 0]
|ttnn.permute|aten::convolution|4| +|129|Tensor<[64,32,2,2]>,
dims: [2, 3, 1, 0]
|ttnn.permute|aten::convolution|4| +|130|Tensor<[4,16,2,2]>,
dims: [2, 3, 1, 0]
|ttnn.permute|aten::convolution|4| +|131|Tensor<[16,1,2,2]>,
dims: [2, 3, 1, 0]
|ttnn.permute|aten::convolution|4| +|132|Tensor<[1,16,32,96]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|133|Tensor<[4608,1536]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|134|Tensor<[1,32,16,96]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|135|Tensor<[16,32,96]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|136|Tensor<[1536,1536]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|137|Tensor<[6144,1536]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|138|Tensor<[1536,6144]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|139|Tensor<[250880,1536]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|140|Tensor<[1,16,12,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|141|Tensor<[1,12,16,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|142|Tensor<[1,12,16,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|143|Tensor<[1024,512,2,2]>,
dims: [2, 3, 1, 0]
|ttnn.permute|aten::convolution|4| +|144|Tensor<[1,19200,1,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|145|Tensor<[1,19200,64]>,
dims: [0, 2, 1]
|ttnn.permute|aten::permute|4| +|146|Tensor<[1,64,300]>,
dims: [0, 2, 1]
|ttnn.permute|aten::permute|4| +|147|Tensor<[1,300,1,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|148|Tensor<[1,1,19200,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|149|Tensor<[1,120,160,64]>,
dims: [0, 3, 1, 2]
|ttnn.permute|aten::permute|4| +|150|Tensor<[1,4800,2,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|151|Tensor<[1,4800,128]>,
dims: [0, 2, 1]
|ttnn.permute|aten::permute|4| +|152|Tensor<[1,128,300]>,
dims: [0, 2, 1]
|ttnn.permute|aten::permute|4| +|153|Tensor<[1,300,2,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|154|Tensor<[1,2,4800,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|155|Tensor<[1,60,80,128]>,
dims: [0, 3, 1, 2]
|ttnn.permute|aten::permute|4| +|156|Tensor<[1,1200,5,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|157|Tensor<[1,1200,320]>,
dims: [0, 2, 1]
|ttnn.permute|aten::permute|4| +|158|Tensor<[1,320,300]>,
dims: [0, 2, 1]
|ttnn.permute|aten::permute|4| +|159|Tensor<[1,300,5,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|160|Tensor<[1,5,1200,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|161|Tensor<[1,30,40,320]>,
dims: [0, 3, 1, 2]
|ttnn.permute|aten::permute|4| +|162|Tensor<[1,300,8,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|163|Tensor<[1,8,300,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|164|Tensor<[1,15,20,512]>,
dims: [0, 3, 1, 2]
|ttnn.permute|aten::permute|4| +|165|Tensor<[1,64,19200]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|166|Tensor<[64,64]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|167|Tensor<[1,1,300,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|168|Tensor<[256,64]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|169|Tensor<[1,19200,256]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|170|Tensor<[1,256,19200]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|171|Tensor<[64,256]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|172|Tensor<[1,128,4800]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|173|Tensor<[128,128]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|174|Tensor<[1,2,300,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|175|Tensor<[512,128]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|176|Tensor<[1,4800,512]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|177|Tensor<[1,512,4800]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|178|Tensor<[128,512]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|179|Tensor<[1,320,1200]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|180|Tensor<[1,5,300,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|181|Tensor<[1,1200,1280]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|182|Tensor<[1,1280,1200]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|183|Tensor<[1,512,300]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|184|Tensor<[512,512]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|185|Tensor<[1,8,300,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|186|Tensor<[2048,512]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|187|Tensor<[1,300,2048]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|188|Tensor<[1,2048,300]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|189|Tensor<[512,2048]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|190|Tensor<[1,197,12,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|191|Tensor<[1,12,197,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|192|Tensor<[1,768,196]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|193|Tensor<[1,12,197,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|194|Tensor<[1000,768]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|195|Tensor<[1,16384,1,32]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|196|Tensor<[1,16384,32]>,
dims: [0, 2, 1]
|ttnn.permute|aten::permute|4| +|197|Tensor<[1,32,256]>,
dims: [0, 2, 1]
|ttnn.permute|aten::permute|4| +|198|Tensor<[1,256,1,32]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|199|Tensor<[1,1,16384,32]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|200|Tensor<[1,128,128,32]>,
dims: [0, 3, 1, 2]
|ttnn.permute|aten::permute|4| +|201|Tensor<[1,4096,2,32]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|202|Tensor<[1,4096,64]>,
dims: [0, 2, 1]
|ttnn.permute|aten::permute|4| +|203|Tensor<[1,64,256]>,
dims: [0, 2, 1]
|ttnn.permute|aten::permute|4| +|204|Tensor<[1,256,2,32]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|205|Tensor<[1,2,4096,32]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|206|Tensor<[1,64,64,64]>,
dims: [0, 3, 1, 2]
|ttnn.permute|aten::permute|4| +|207|Tensor<[1,1024,5,32]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|208|Tensor<[1,1024,160]>,
dims: [0, 2, 1]
|ttnn.permute|aten::permute|4| +|209|Tensor<[1,160,256]>,
dims: [0, 2, 1]
|ttnn.permute|aten::permute|4| +|210|Tensor<[1,256,5,32]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|211|Tensor<[1,5,1024,32]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|212|Tensor<[1,32,32,160]>,
dims: [0, 3, 1, 2]
|ttnn.permute|aten::permute|4| +|213|Tensor<[1,8,256,32]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|214|Tensor<[1,16,16,256]>,
dims: [0, 3, 1, 2]
|ttnn.permute|aten::permute|4| +|215|Tensor<[1,16384,256]>,
dims: [0, 2, 1]
|ttnn.permute|aten::permute|4| +|216|Tensor<[1,4096,256]>,
dims: [0, 2, 1]
|ttnn.permute|aten::permute|4| +|217|Tensor<[1,1024,256]>,
dims: [0, 2, 1]
|ttnn.permute|aten::permute|4| +|218|Tensor<[1,256,256]>,
dims: [0, 2, 1]
|ttnn.permute|aten::permute|4| +|219|Tensor<[1,32,16384]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|220|Tensor<[32,32]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|221|Tensor<[1,1,256,32]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|222|Tensor<[128,32]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|223|Tensor<[1,16384,128]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|224|Tensor<[1,128,16384]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|225|Tensor<[32,128]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|226|Tensor<[1,64,4096]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|227|Tensor<[1,2,256,32]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|228|Tensor<[1,256,4096]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|229|Tensor<[1,160,1024]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|230|Tensor<[160,160]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|231|Tensor<[1,5,256,32]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|232|Tensor<[640,160]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|233|Tensor<[1,1024,640]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|234|Tensor<[1,640,1024]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|235|Tensor<[160,640]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|236|Tensor<[1024,256]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|237|Tensor<[1,256,1024]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|238|Tensor<[256,1024]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|239|Tensor<[256,32]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|240|Tensor<[256,160]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|241|Tensor<[4672,4544]>,
dims: [1, 0]
|ttnn.permute|aten::permute|5| +|242|Tensor<[1,71,7,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|243|Tensor<[4544,4544]>,
dims: [1, 0]
|ttnn.permute|aten::permute|5| +|244|Tensor<[18176,4544]>,
dims: [1, 0]
|ttnn.permute|aten::permute|5| +|245|Tensor<[4544,18176]>,
dims: [1, 0]
|ttnn.permute|aten::permute|5| +|246|Tensor<[1,32,7]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|247|Tensor<[1,7,71,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|248|Tensor<[1,7,1,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|249|Tensor<[1,1,7,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|250|Tensor<[65024,4544]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|251|Tensor<[1000,1280]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|252|Tensor<[1,12,12,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|253|Tensor<[768,128]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|254|Tensor<[1,12,12,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|255|Tensor<[1,9,12,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|256|Tensor<[1,12,9,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|257|Tensor<[1,12,9,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|258|Tensor<[128,768]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|259|Tensor<[30000,128]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|260|Tensor<[1,9,16,128]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|261|Tensor<[2048,128]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|262|Tensor<[2048,2048]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|263|Tensor<[1,16,9,128]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|264|Tensor<[1,16,9,128]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|265|Tensor<[8192,2048]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|266|Tensor<[2048,8192]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|267|Tensor<[128,2048]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|268|Tensor<[1,9,16,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|269|Tensor<[1024,128]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|270|Tensor<[1,16,9,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|271|Tensor<[1,16,9,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|272|Tensor<[128,1024]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|273|Tensor<[1,9,64,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|274|Tensor<[4096,128]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|275|Tensor<[1,64,9,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|276|Tensor<[1,64,9,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|277|Tensor<[16384,4096]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|278|Tensor<[4096,16384]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|279|Tensor<[128,4096]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|280|Tensor<[1,14,12,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|281|Tensor<[1,12,14,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|282|Tensor<[1,12,14,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|283|Tensor<[1,768,49]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|284|Tensor<[1,50,12,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|285|Tensor<[1,12,50,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|286|Tensor<[1,12,50,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|287|Tensor<[2,7,8,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|288|Tensor<[2,8,7,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|289|Tensor<[2,8,7,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::transpose.int|4| +|290|Tensor<[1,512]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|291|Tensor<[2,1]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|292|Tensor<[1,197,16,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|293|Tensor<[1,27,27,16]>,
dims: [0, 3, 1, 2]
|ttnn.permute|aten::permute|4| +|294|Tensor<[1,16,27,27]>,
dims: [0, 2, 3, 1]
|ttnn.permute|aten::permute|4| +|295|Tensor<[2,196,196]>,
dims: [1, 2, 0]
|ttnn.permute|aten::permute|4| +|296|Tensor<[197,197,16]>,
dims: [2, 0, 1]
|ttnn.permute|aten::permute|4| +|297|Tensor<[1,16,197,64]>,
dims: [0, 2, 1, 3]
|ttnn.permute|aten::permute|4| +|298|Tensor<[1,1024,196]>,
dims: [0, 2, 1]
|ttnn.permute|aten::transpose.int|4| +|299|Tensor<[1,16,197,64]>,
dims: [0, 1, 3, 2]
|ttnn.permute|aten::transpose.int|4| +|300|Tensor<[1,27,27,12]>,
dims: [0, 3, 1, 2]
|ttnn.permute|aten::permute|4| +|301|Tensor<[1,12,27,27]>,
dims: [0, 2, 3, 1]
|ttnn.permute|aten::permute|4| +|302|Tensor<[197,197,12]>,
dims: [2, 0, 1]
|ttnn.permute|aten::permute|4| +|303|Tensor<[128,784]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|304|Tensor<[64,128]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|305|Tensor<[12,64]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|306|Tensor<[3,12]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|307|Tensor<[12,3]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|308|Tensor<[64,12]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|309|Tensor<[128,64]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| +|310|Tensor<[784,128]>,
dims: [1, 0]
|ttnn.permute|aten::transpose.int|5| diff --git a/docs/ops/stablehlo/tensor.empty.md b/docs/ops/stablehlo/tensor.empty.md new file mode 100644 index 00000000..27f6db85 --- /dev/null +++ b/docs/ops/stablehlo/tensor.empty.md @@ -0,0 +1,9 @@ + +### tensor.empty + + +||STABLE HLO Input Variations|ttnn op|Torch Name|Status| +| :--- | :--- | :--- | :--- | :--- | +|0|Tensor<[32,32]>,
||aten::empty_strided|4| +|1|Tensor<[7,7]>,
||aten::empty_strided|4| +|2|Tensor<[19,19]>,
||aten::empty_strided|4| diff --git a/results/parse_op_by_op_results.py b/results/parse_op_by_op_results.py index 33303d2a..8e4341cd 100644 --- a/results/parse_op_by_op_results.py +++ b/results/parse_op_by_op_results.py @@ -6,6 +6,7 @@ import json import csv import xlsxwriter +from mdutils.mdutils import MdUtils # Script to parse the results of the unique ops json files and combine them into a spreadsheet # This script parses models compiled into stable hlo / TTIR op by op @@ -35,11 +36,23 @@ def append_shape(shape): return "x".join(shape_strs) +def extract_shapes_md(shape_list): + shape_str = "" + for shape in shape_list: + if len(shape): + shape_str += f"Tensor<[{','.join([str(dim) for dim in shape])}]>,
" + else: + shape_str += "Scalar,
" + + return shape_str + + def process_json_files(): json_files = find_json_files() ops_per_model = {} stable_hlo_ops_per_model = {} + stable_hlo_ops = {} models_per_op = {} stable_hlo_models_per_op = {} model_names = [] @@ -127,6 +140,14 @@ def process_json_files(): worksheet.set_row(row, None, None, {"hidden": True}) row += 1 for shlo_op in ops: + if shlo_op[1] not in stable_hlo_ops: + stable_hlo_ops[shlo_op[1]] = [] + op = shlo_op + op.append(torch_name) + op.append(input_shapes) + op.append(output_shapes) + op.append(status) + stable_hlo_ops[shlo_op[1]].append(op) stable_hlo_ops_per_model[model_name].add(shlo_op[1]) stable_hlo_ops_per_torch_op[torch_name].add(shlo_op[1]) row_data = ["", "", "", "", "", shlo_op[-1]] @@ -271,6 +292,287 @@ def process_json_files(): workbook.close() + op_mappings = { + # "arith.constant": "ttnn.", + "stablehlo.abs": "ttnn.abs", + "stablehlo.add": "ttnn.add", + "stablehlo.and": "ttnn.and", + # "stablehlo.broadcast_in_dim": "ttnn.", + "stablehlo.ceil": "ttnn.ceil", + "stablehlo.clamp": "ttnn.clamp", + "stablehlo.compare": "ttnn.?", + "stablehlo.concatenate": "ttnn.concat", + # "stablehlo.constant": "ttnn.", + # "stablehlo.convert": "ttnn.", + "stablehlo.convolution": "ttnn.conv2d", + "stablehlo.cosine": "ttnn.cos", + "stablehlo.divide": "ttnn.div", + "stablehlo.dot_general": "ttnn.matmul", + "stablehlo.dynamic_iota": "ttnn.arange", + "stablehlo.exponential": "ttnn.exp", + "stablehlo.floor": "ttnn.floor", + "stablehlo.gather": "ttnn.embedding", + "stablehlo.iota": "ttnn.arange", + "stablehlo.log": "ttnn.log", + "stablehlo.logistic": "ttnn.sigmoig", + "stablehlo.maximum": "ttnn.maximum", + "stablehlo.minimum": "ttnn.minimum", + "stablehlo.multiply": "ttnn.multiply", + "stablehlo.negate": "ttnn.neg", + "stablehlo.not": "ttnn.not", + "stablehlo.or": "ttnn.or", + "stablehlo.power": "ttnn.pow", + "stablehlo.reduce_stablehlo.add": "ttnn.sum", + "stablehlo.reduce_stablehlo.and": "ttnn.?", + "stablehlo.reduce_stablehlo.maximum": "ttnn.max", + "stablehlo.reduce_stablehlo.or": "ttnn.?", + "stablehlo.reduce_window_stablehlo.add": "ttnn.avg_pool2d", + "stablehlo.remainder": "ttnn.remainder", + "stablehlo.reshape": "ttnn.reshape", + "stablehlo.reverse": "ttnn.?", + "stablehlo.rsqrt": "ttnn.rsqrt", + "stablehlo.scatter": "ttnn.scatter", + "stablehlo.select": "ttnn.where", + "stablehlo.sine": "ttnn.sin", + "stablehlo.slice": "ttnn.slice", + "stablehlo.sqrt": "ttnn.sqrt", + "stablehlo.subtract": "ttnn.subtract", + "stablehlo.tanh": "ttnn.tanh", + "stablehlo.transpose": "ttnn.permute", + # "tensor.empty": "ttnn.", + } + + def process_compare(shlo_op, md_data): + inputs_and_attr = extract_shapes_md(shlo_op[4]) + md_data.append(inputs_and_attr) + op = shlo_op[2][0] + if op == "EQ": + md_data.append("ttnn.eq") + elif op == "GT": + md_data.append("ttnn.gt") + elif op == "LT": + md_data.append("ttnn.lt") + elif op == "NE": + md_data.append("ttnn.ne") + elif op == "GE": + md_data.append("ttnn.ge") + + md_data.append(shlo_op[7]) + md_data.append(f"{shlo_op[10]}") + return + + def process_gather(shlo_op, md_data): + # '%0 = "stablehlo.gather"(%arg0, %arg1) <{dimension_numbers = #stablehlo.gather, indices_are_sorted = false, slice_sizes = array}> : (tensor<32000x4096xbf16>, tensor<1x32xi64>) -> tensor<1x32x4096xbf16>' + shlo_attr = [ + "offset_dims", + "collapsed_slice_dims", + "start_index_map", + "index_vector_dim", + "indices_are_sorted", + "slice_sizes", + ] + terminators = [",", "}", ">", ":"] + inputs_and_attr = extract_shapes_md(shlo_op[4]) + shlo = shlo_op[6] + for attr in shlo_attr: + if attr in shlo: + res = shlo.split(f"{attr} = ")[1] + open_brackets = 0 + for idx, char in enumerate(res): + if char == "[": + open_brackets += 1 + elif char == "]": + open_brackets -= 1 + if open_brackets == 0 and char in terminators: + break + res = res[:idx] + inputs_and_attr += f"{attr}: {res}
" + + md_data.append(inputs_and_attr) + md_data.append("ttnn.embedding") + + md_data.append(shlo_op[7]) + md_data.append(f"{shlo_op[10]}") + + return + + def process_convolution(shlo_op, md_data): + # {'(%arg0, %arg1) dim_numbers': '[b, f, 0, 1]x[o, i, 0, 1]->[b, f, 0, 1]', 'window': '{stride = [2, 2], pad = [[1, 1], [1, 1]], rhs_dilate = [1, 1]} {batch_group_count = 1 : i64, feature_group_count = 1 : i64}'} + shlo_attr = [ + "stride", + "pad", + "rhs_dilate", + "lhs_dilate", + "batch_group_count", + "feature_group_count", + ] + terminators = [",", "}", ":"] + inputs_and_attr = extract_shapes_md(shlo_op[4]) + shlo = shlo_op[6] + for attr in shlo_attr: + if attr in shlo: + res = shlo.split(f"{attr} = ")[1] + open_brackets = 0 + for idx, char in enumerate(res): + if char == "[": + open_brackets += 1 + elif char == "]": + open_brackets -= 1 + if open_brackets == 0 and char in terminators: + break + res = res[:idx] + inputs_and_attr += f"{attr}: {res}
" + + md_data.append(inputs_and_attr) + md_data.append("ttnn.conv2d") + md_data.append(shlo_op[7]) + md_data.append(f"{shlo_op[10]}") + + return + + def process_reshape(shlo_op, md_data): + # %1 = stablehlo.reshape %0 : (tensor<1x32x32xf32>) -> tensor<1x32x32x1xf32> + inputs_and_attr = extract_shapes_md(shlo_op[4]) + inputs_and_attr += extract_shapes_md(shlo_op[5]) + + md_data.append(inputs_and_attr) + md_data.append("ttnn.reshape") + md_data.append(shlo_op[7]) + md_data.append(f"{shlo_op[10]}") + + return + + def process_scatter(shlo_op, md_data): + # %1 = "stablehlo.scatter"(%arg0, %c, %0) <{indices_are_sorted = false, scatter_dimension_numbers = #stablehlo.scatter, unique_indices = false}> ({^bb0(%arg2: tensor, %arg3: tensor):stablehlo.return %arg3 : tensor}) : (tensor<1x3x720x1280xbf16>, tensor<1x1xi64>, tensor<1x3x720x1280xbf16>) -> tensor<1x3x720x1280xbf16> + shlo_attr = [ + "update_window_dims", + "inserted_window_dims", + "scatter_dims_to_operand_dims", + "index_vector_dim", + ] + terminators = [",", "}", ":"] + inputs_and_attr = extract_shapes_md(shlo_op[4]) + shlo = shlo_op[6] + for attr in shlo_attr: + if attr in shlo: + res = shlo.split(f"{attr} = ")[1] + open_brackets = 0 + for idx, char in enumerate(res): + if char == "[": + open_brackets += 1 + elif char == "]": + open_brackets -= 1 + if open_brackets == 0 and char in terminators: + break + res = res[:idx] + inputs_and_attr += f"{attr}: {res}
" + + md_data.append(inputs_and_attr) + md_data.append("ttnn.scatter") + md_data.append(shlo_op[7]) + md_data.append(f"{shlo_op[10]}") + + return + + def process_slice(shlo_op, md_data): + #% 0 = stablehlo.slice %arg0 [0:1, 0:32, 0:32, 0:64] : (tensor<1x32x32x128xbf16>) -> tensor<1x32x32x64xbf16> + inputs_and_attr = extract_shapes_md(shlo_op[4]) + indices = shlo_op[6][shlo_op[6].find("[") : shlo_op[6].find("]") + 1] + inputs_and_attr += f"indices: {indices}
" + + md_data.append(inputs_and_attr) + md_data.append("ttnn.reshape") + md_data.append(shlo_op[7]) + md_data.append(f"{shlo_op[10]}") + + def default(shlo_op, md_data): + if len(shlo_op[4]) == 1 and len(shlo_op[2]) != 1: + shlo_op[4].append(shlo_op[4][0]) + inputs_and_attr = extract_shapes_md(shlo_op[4]) + attrs = shlo_op[3] + for k, v in attrs.items(): + inputs_and_attr += f"{k}: {v}
" + + md_data.append(inputs_and_attr) + if shlo_op[1] in op_mappings: + md_data.append(op_mappings[shlo_op[1]]) + else: + md_data.append("") + + md_data.append(shlo_op[7]) + md_data.append(f"{shlo_op[10]}") + return + + workbook = xlsxwriter.Workbook("results/stable_hlo_ops.xlsx") + keys = sorted(stable_hlo_ops.keys()) + for op in keys: + md_file = MdUtils(file_name="docs/ops/" + op + ".md", title=op) + md_file.create_md_file() + + title = op + if op in op_mappings: + title += "::" + op_mappings[op] + + md_file.new_header(level=3, title=title, add_table_of_contents="n") + worksheet = workbook.add_worksheet(op[:31]) + + row = 0 + header = ( + "Output", + "Op Name", + "Args", + "Attrs", + "Input Shapes", + "Output Shapes", + "HLO", + "TorchName", + "TorchIn", + "TorchOut", + "Status", + ) + worksheet.write_row(row, 0, header, bold) + row += 1 + md_data = ["", "STABLE HLO Input Variations", "ttnn op", "Torch Name", "Status"] + unique_ops = set() + index = 0 + for shlo_op in stable_hlo_ops[op]: + md_data.append(f"{index}") + if op == "stablehlo.compare": + process_compare(shlo_op, md_data) + elif op == "stablehlo.gather": + process_gather(shlo_op, md_data) + elif op == "stablehlo.convolution": + process_convolution(shlo_op, md_data) + elif op == "stablehlo.reshape": + process_reshape(shlo_op, md_data) + elif op == "stablehlo.slice": + process_slice(shlo_op, md_data) + elif op == "stablehlo.scatter": + process_scatter(shlo_op, md_data) + else: + default(shlo_op, md_data) + + if md_data[-4] in unique_ops: + md_data = md_data[:-5] + else: + index += 1 + unique_ops.add(md_data[-4]) + row_data = [elem.__str__() for elem in shlo_op] + worksheet.write_row(row, 0, row_data) + row += 1 + md_file.new_line() + + md_file.new_table( + columns=5, rows=len(md_data) // 5, text=md_data, text_align="left" + ) + md_file.new_line() + text = md_file.file_data_text + with open("docs/ops/" + op + ".md", "w") as f: + f.write(text) + + worksheet.autofit() + workbook.close() + if __name__ == "__main__": process_json_files()