ttnn.embedding
| Name | Input Shapes | Input Layouts | Attributes | Output Shapes | Output Layouts | PCC | ATOL |
|---|---|---|---|---|---|---|---|
| ttnn.embedding | tensor<[1,1,ui32]> tensor<[151936,1536,bf16]> tensor<[1,1,1536,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 1, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (4748, 48, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | tensor<[1,1,1536,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,12,ui32]> tensor<[151936,1536,bf16]> tensor<[1,12,1536,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 12, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (4748, 48, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 12 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | tensor<[1,12,1536,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 12 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,13,ui32]> tensor<[152064,3584,bf16]> tensor<[1,13,3584,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 13, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (4752, 112, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 13 + d1, d2), memory_config: (1, 112, 'tile<32x32, bf16>', 'dram') | tensor<[1,13,3584,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 13 + d1, d2), memory_config: (1, 112, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[2,1,ui32]> tensor<[2048,128,bf16]> tensor<[2,1,128,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (2, 1, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (64, 4, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 4, 'tile<32x32, bf16>', 'dram') | tensor<[2,1,128,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 4, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,1,ui32]> tensor<[2050,1024,bf16]> tensor<[1,1,1024,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 1, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (65, 32, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 32, 'tile<32x32, bf16>', 'dram') | tensor<[1,1,1024,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 32, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,6,ui32]> tensor<[2050,1024,bf16]> tensor<[1,6,1024,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 6, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (65, 32, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 6 + d1, d2), memory_config: (1, 32, 'tile<32x32, bf16>', 'dram') | tensor<[1,6,1024,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 6 + d1, d2), memory_config: (1, 32, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,10,ui32]> tensor<[250002,768,bf16]> tensor<[1,10,768,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 10, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (7813, 24, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 10 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | tensor<[1,10,768,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 10 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,32,ui32]> tensor<[250880,1536,bf16]> tensor<[1,32,1536,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 32, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (7840, 48, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 32 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | tensor<[1,32,1536,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 32 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,2048,ui32]> tensor<[262,768,bf16]> tensor<[1,2048,768,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 2048, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (9, 24, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 2048 + d1, d2), memory_config: (64, 24, 'tile<32x32, bf16>', 'dram') | tensor<[1,2048,768,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 2048 + d1, d2), memory_config: (64, 24, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,256,ui32]> tensor<[2,1024,bf16]> tensor<[1,256,1024,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 256, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 32, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 256 + d1, d2), memory_config: (8, 32, 'tile<32x32, bf16>', 'dram') | tensor<[1,256,1024,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 256 + d1, d2), memory_config: (8, 32, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,12,ui32]> tensor<[2,128,bf16]> tensor<[1,12,128,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 12, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 4, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 12 + d1, d2), memory_config: (1, 4, 'tile<32x32, bf16>', 'dram') | tensor<[1,12,128,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 12 + d1, d2), memory_config: (1, 4, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,14,ui32]> tensor<[2,128,bf16]> tensor<[1,14,128,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 14, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 4, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 14 + d1, d2), memory_config: (1, 4, 'tile<32x32, bf16>', 'dram') | tensor<[1,14,128,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 14 + d1, d2), memory_config: (1, 4, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,9,ui32]> tensor<[2,128,bf16]> tensor<[1,9,128,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 9, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 4, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 9 + d1, d2), memory_config: (1, 4, 'tile<32x32, bf16>', 'dram') | tensor<[1,9,128,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 9 + d1, d2), memory_config: (1, 4, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,193,ui32]> tensor<[2,768,bf16]> tensor<[1,193,768,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 193, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 193 + d1, d2), memory_config: (7, 24, 'tile<32x32, bf16>', 'dram') | tensor<[1,193,768,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 193 + d1, d2), memory_config: (7, 24, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,8,ui32]> tensor<[2,768,bf16]> tensor<[1,8,768,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 8, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 8 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | tensor<[1,8,768,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 8 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,12,ui32]> tensor<[30000,128,bf16]> tensor<[1,12,128,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 12, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (938, 4, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 12 + d1, d2), memory_config: (1, 4, 'tile<32x32, bf16>', 'dram') | tensor<[1,12,128,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 12 + d1, d2), memory_config: (1, 4, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,14,ui32]> tensor<[30000,128,bf16]> tensor<[1,14,128,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 14, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (938, 4, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 14 + d1, d2), memory_config: (1, 4, 'tile<32x32, bf16>', 'dram') | tensor<[1,14,128,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 14 + d1, d2), memory_config: (1, 4, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,9,ui32]> tensor<[30000,128,bf16]> tensor<[1,9,128,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 9, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (938, 4, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 9 + d1, d2), memory_config: (1, 4, 'tile<32x32, bf16>', 'dram') | tensor<[1,9,128,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 9 + d1, d2), memory_config: (1, 4, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,256,ui32]> tensor<[30522,1024,bf16]> tensor<[1,256,1024,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 256, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (954, 32, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 256 + d1, d2), memory_config: (8, 32, 'tile<32x32, bf16>', 'dram') | tensor<[1,256,1024,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 256 + d1, d2), memory_config: (8, 32, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,16,ui32]> tensor<[30522,768,bf16]> tensor<[1,16,768,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 16, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (954, 24, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 16 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | tensor<[1,16,768,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 16 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,25,ui32]> tensor<[30522,768,bf16]> tensor<[1,25,768,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 25, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (954, 24, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 25 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | tensor<[1,25,768,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 25 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,8,ui32]> tensor<[30522,768,bf16]> tensor<[1,8,768,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 8, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (954, 24, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 8 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | tensor<[1,8,768,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 8 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,8,ui32]> tensor<[30528,768,bf16]> tensor<[1,8,768,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 8, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (954, 24, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 8 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | tensor<[1,8,768,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 8 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,32,ui32]> tensor<[32000,4096,bf16]> tensor<[1,32,4096,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 32, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1000, 128, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 32 + d1, d2), memory_config: (1, 128, 'tile<32x32, bf16>', 'dram') | tensor<[1,32,4096,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 32 + d1, d2), memory_config: (1, 128, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,1,ui32]> tensor<[32128,1024,bf16]> tensor<[1,1,1024,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 1, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1004, 32, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 32, 'tile<32x32, bf16>', 'dram') | tensor<[1,1,1024,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 32, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,10,ui32]> tensor<[32128,1024,bf16]> tensor<[1,10,1024,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 10, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1004, 32, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 10 + d1, d2), memory_config: (1, 32, 'tile<32x32, bf16>', 'dram') | tensor<[1,10,1024,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 10 + d1, d2), memory_config: (1, 32, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,1,ui32]> tensor<[32128,512,bf16]> tensor<[1,1,512,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 1, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1004, 16, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 16, 'tile<32x32, bf16>', 'dram') | tensor<[1,1,512,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 16, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,10,ui32]> tensor<[32128,512,bf16]> tensor<[1,10,512,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 10, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1004, 16, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 10 + d1, d2), memory_config: (1, 16, 'tile<32x32, bf16>', 'dram') | tensor<[1,10,512,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 10 + d1, d2), memory_config: (1, 16, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,15,ui32]> tensor<[32128,512,bf16]> tensor<[1,15,512,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 15, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1004, 16, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 15 + d1, d2), memory_config: (1, 16, 'tile<32x32, bf16>', 'dram') | tensor<[1,15,512,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 15 + d1, d2), memory_config: (1, 16, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,1,ui32]> tensor<[32128,768,bf16]> tensor<[1,1,768,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 1, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1004, 24, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | tensor<[1,1,768,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,10,ui32]> tensor<[32128,768,bf16]> tensor<[1,10,768,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 10, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1004, 24, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 10 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | tensor<[1,10,768,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 10 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[2,13,ui32]> tensor<[32128,768,bf16]> tensor<[2,13,768,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (2, 13, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1004, 24, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 13 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | tensor<[2,13,768,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 13 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[2,7,ui32]> tensor<[49408,512,bf16]> tensor<[2,7,512,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (2, 7, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1544, 16, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 7 + d1, d2), memory_config: (1, 16, 'tile<32x32, bf16>', 'dram') | tensor<[2,7,512,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 7 + d1, d2), memory_config: (1, 16, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,1,ui32]> tensor<[50257,768,bf16]> tensor<[1,1,768,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 1, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1571, 24, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | tensor<[1,1,768,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,45,ui32]> tensor<[50257,768,bf16]> tensor<[1,45,768,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 45, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1571, 24, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 45 + d1, d2), memory_config: (2, 24, 'tile<32x32, bf16>', 'dram') | tensor<[1,45,768,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 45 + d1, d2), memory_config: (2, 24, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,7,ui32]> tensor<[50257,768,bf16]> tensor<[1,7,768,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 7, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1571, 24, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 7 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | tensor<[1,7,768,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 7 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,1,ui32]> tensor<[50272,512,bf16]> tensor<[1,1,512,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 1, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1571, 16, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 16, 'tile<32x32, bf16>', 'dram') | tensor<[1,1,512,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 16, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,6,ui32]> tensor<[50272,512,bf16]> tensor<[1,6,512,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 6, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1571, 16, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 6 + d1, d2), memory_config: (1, 16, 'tile<32x32, bf16>', 'dram') | tensor<[1,6,512,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 6 + d1, d2), memory_config: (1, 16, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,10,ui32]> tensor<[50280,1536,bf16]> tensor<[1,10,1536,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 10, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1572, 48, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 10 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | tensor<[1,10,1536,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 10 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,11,ui32]> tensor<[50280,1536,bf16]> tensor<[1,11,1536,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 11, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1572, 48, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 11 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | tensor<[1,11,1536,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 11 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,12,ui32]> tensor<[50280,1536,bf16]> tensor<[1,12,1536,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 12, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1572, 48, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 12 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | tensor<[1,12,1536,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 12 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,13,ui32]> tensor<[50280,1536,bf16]> tensor<[1,13,1536,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 13, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1572, 48, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 13 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | tensor<[1,13,1536,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 13 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,14,ui32]> tensor<[50280,1536,bf16]> tensor<[1,14,1536,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 14, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1572, 48, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 14 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | tensor<[1,14,1536,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 14 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,15,ui32]> tensor<[50280,1536,bf16]> tensor<[1,15,1536,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 15, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1572, 48, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 15 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | tensor<[1,15,1536,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 15 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,6,ui32]> tensor<[50280,1536,bf16]> tensor<[1,6,1536,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 6, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1572, 48, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 6 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | tensor<[1,6,1536,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 6 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,7,ui32]> tensor<[50280,1536,bf16]> tensor<[1,7,1536,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 7, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1572, 48, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 7 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | tensor<[1,7,1536,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 7 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,8,ui32]> tensor<[50280,1536,bf16]> tensor<[1,8,1536,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 8, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1572, 48, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 8 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | tensor<[1,8,1536,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 8 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,9,ui32]> tensor<[50280,1536,bf16]> tensor<[1,9,1536,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 9, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1572, 48, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 9 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | tensor<[1,9,1536,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 9 + d1, d2), memory_config: (1, 48, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,1,ui32]> tensor<[51200,1024,bf16]> tensor<[1,1,1024,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 1, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1600, 32, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 32, 'tile<32x32, bf16>', 'dram') | tensor<[1,1,1024,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 32, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,5,ui32]> tensor<[51200,1024,bf16]> tensor<[1,5,1024,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 5, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1600, 32, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 5 + d1, d2), memory_config: (1, 32, 'tile<32x32, bf16>', 'dram') | tensor<[1,5,1024,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 5 + d1, d2), memory_config: (1, 32, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,10,ui32]> tensor<[514,768,bf16]> tensor<[1,10,768,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 10, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (17, 24, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 10 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | tensor<[1,10,768,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 10 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,1,ui32]> tensor<[51865,768,bf16]> tensor<[1,1,768,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 1, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1621, 24, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | tensor<[1,1,768,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 24, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,7,ui32]> tensor<[65024,4544,bf16]> tensor<[1,7,4544,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 7, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (2032, 142, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 7 + d1, d2), memory_config: (1, 142, 'tile<32x32, bf16>', 'dram') | tensor<[1,7,4544,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 7 + d1, d2), memory_config: (1, 142, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,1,ui32]> tensor<[2048,32,bf16]> tensor<[1,1,32,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 1, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (64, 1, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 1, 'tile<32x32, bf16>', 'dram') | tensor<[1,1,32,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 + d1, d2), memory_config: (1, 1, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1,5,ui32]> tensor<[2048,32,bf16]> tensor<[1,5,32,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 5, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (64, 1, 'tile<32x32, bf16>', 'dram') mapping_from: (d0, d1, d2), mapping_to: (d0 * 5 + d1, d2), memory_config: (1, 1, 'tile<32x32, bf16>', 'dram') | tensor<[1,5,32,bf16]> | mapping_from: (d0, d1, d2), mapping_to: (d0 * 5 + d1, d2), memory_config: (1, 1, 'tile<32x32, bf16>', 'dram') | nan | nan |