ttnn.embedding
This table is a trace for ttnn.embedding op. Traces are generated from nightly tt-torch runs. To see nightly runs: Nightly Runs
| Name | Input Shapes | Input Layouts | Attributes | Output Shapes | Output Layouts | PCC | ATOL |
|---|---|---|---|---|---|---|---|
| ttnn.embedding | tensor<[1000000,ui32]> tensor<[2048000,1,bf16]> | mapping_from: (d0), mapping_to: (0, d0), memory_config: (1, 1000000, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (64000, 1, 'tile<32x32, bf16>', 'dram') | tensor<[1000000,1,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (31250, 1, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[2027520,ui32]> tensor<[168960,1,bf16]> | mapping_from: (d0), mapping_to: (0, d0), memory_config: (1, 2027520, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (5280, 1, 'tile<32x32, bf16>', 'dram') | tensor<[2027520,1,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (63360, 1, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[1200,ui32]> tensor<[33600,1,bf16]> | mapping_from: (d0), mapping_to: (0, d0), memory_config: (1, 1200, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1050, 1, 'tile<32x32, bf16>', 'dram') | tensor<[1200,1,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (38, 1, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[24000,ui32]> tensor<[672000,1,bf16]> | mapping_from: (d0), mapping_to: (0, d0), memory_config: (1, 24000, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (21000, 1, 'tile<32x32, bf16>', 'dram') | tensor<[24000,1,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (750, 1, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[3,ui32]> tensor<[12,1,bf16]> | mapping_from: (d0), mapping_to: (0, d0), memory_config: (1, 3, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 1, 'tile<32x32, bf16>', 'dram') | tensor<[3,1,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1, 1, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[45303552,ui32]> tensor<[50176,1,bf16]> | mapping_from: (d0), mapping_to: (0, d0), memory_config: (1, 45303552, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1568, 1, 'tile<32x32, bf16>', 'dram') | tensor<[45303552,1,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1415736, 1, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[45303552,ui32]> tensor<[200704,1,bf16]> | mapping_from: (d0), mapping_to: (0, d0), memory_config: (1, 45303552, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (6272, 1, 'tile<32x32, bf16>', 'dram') | tensor<[45303552,1,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1415736, 1, 'tile<32x32, bf16>', 'dram') | nan | nan | |
| ttnn.embedding | tensor<[45303552,ui32]> tensor<[12544,1,bf16]> | mapping_from: (d0), mapping_to: (0, d0), memory_config: (1, 45303552, 'ui32', 'dram') mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (392, 1, 'tile<32x32, bf16>', 'dram') | tensor<[45303552,1,bf16]> | mapping_from: (d0, d1), mapping_to: (d0, d1), memory_config: (1415736, 1, 'tile<32x32, bf16>', 'dram') | nan | nan |