From e1f8727302b3d11a540ff9479cebe1720ad86d23 Mon Sep 17 00:00:00 2001 From: Nilaykumar Patel Date: Thu, 26 Dec 2024 08:41:47 +0000 Subject: [PATCH] Add ttforge-sweep in github workflow. Signed-off-by: Nilaykumar Patel --- .github/workflows/ttnn-run-sweeps.yaml | 1 + .../sweep_utils/conv2d_common.py | 4 +- .../sweeps/conv2d/short/conv2d_short_sweep.py | 7187 ---------------- .../conv2d/short/conv2d_ttforge_sweep.py | 7233 +++++++++++++++++ 4 files changed, 7235 insertions(+), 7190 deletions(-) create mode 100644 tests/sweep_framework/sweeps/conv2d/short/conv2d_ttforge_sweep.py diff --git a/.github/workflows/ttnn-run-sweeps.yaml b/.github/workflows/ttnn-run-sweeps.yaml index 11bcc7acde0..f75e7bbfd6d 100644 --- a/.github/workflows/ttnn-run-sweeps.yaml +++ b/.github/workflows/ttnn-run-sweeps.yaml @@ -388,6 +388,7 @@ on: - conv2d.full.conv2d_sharding - conv2d.full.conv2d_sliding_window - conv2d.short.conv2d_short_sweep + - conv2d.short.conv2d_ttforge_sweep - pooling.global_avg_pool2d - pooling.max_pool2d - max_pool2d.short.max_pool2d_short_sweep diff --git a/tests/sweep_framework/sweep_utils/conv2d_common.py b/tests/sweep_framework/sweep_utils/conv2d_common.py index 5be1e1e31d4..c6f2abba90d 100644 --- a/tests/sweep_framework/sweep_utils/conv2d_common.py +++ b/tests/sweep_framework/sweep_utils/conv2d_common.py @@ -259,9 +259,7 @@ def run_conv2d_short_sweep( if is_forge_suite: input_layout = ttnn.Layout(input_layout) input_dtype = ttnn.DataType(input_dtype) - input_memory_config = ( - ttnn.DRAM_MEMORY_CONFIG if input_buffer_type == int(ttnn.BufferType.DRAM) else ttnn.L1_MEMORY_CONFIG - ) + input_memory_config = ttnn.DRAM_MEMORY_CONFIG if input_buffer_type == "dram" else ttnn.L1_MEMORY_CONFIG tt_input_tensor = ttnn.from_torch( torch_input_tensor, dtype=input_dtype, layout=input_layout, device=device, memory_config=input_memory_config ) diff --git a/tests/sweep_framework/sweeps/conv2d/short/conv2d_short_sweep.py b/tests/sweep_framework/sweeps/conv2d/short/conv2d_short_sweep.py index 16b1abf8c81..91752c80ec1 100644 --- a/tests/sweep_framework/sweeps/conv2d/short/conv2d_short_sweep.py +++ b/tests/sweep_framework/sweeps/conv2d/short/conv2d_short_sweep.py @@ -1583,7193 +1583,6 @@ ], "is_conv1d": [True], }, - "tt-forge_sweep_conv2d": { - "input_specs": [ - # Contains following params - # [batch_size, output_channels, input_channels, input_height, input_width, kernel_height, kernel_width, stride_h, stride_w, pad_h, pad_w, groups, dilation_h, dilation_w, bias, [input_layout, input_memory_config, input_datatype], [weight_layout, weight_memory_config, weight_datatype], [output_layout, output_memory_config, output_datatype]] - [ - 1, - 1024, - 1024, - 14, - 14, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 2048, - 1024, - 14, - 14, - 1, - 1, - 2, - 2, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 1024, - 14, - 14, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 1024, - 14, - 14, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1024, - 1024, - 16, - 16, - 3, - 3, - 1, - 1, - 1, - 1, - 1024, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 255, - 1024, - 16, - 16, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 1024, - 16, - 16, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 1024, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 2048, - 1024, - 45, - 80, - 1, - 1, - 2, - 2, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 1024, - 45, - 80, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 1024, - 45, - 80, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 40, - 102, - 56, - 56, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 462, - 1072, - 7, - 7, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 672, - 112, - 20, - 20, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 40, - 116, - 14, - 14, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 34, - 118, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 32, - 120, - 1, - 1, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 480, - 120, - 1, - 1, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 120, - 120, - 40, - 40, - 5, - 5, - 1, - 1, - 2, - 2, - 120, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 40, - 120, - 40, - 40, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 46, - 122, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 124, - 56, - 56, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1280, - 1280, - 16, - 16, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1280, - 1280, - 16, - 16, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1280, - 1280, - 16, - 16, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1280, - 1280, - 30, - 40, - 3, - 3, - 1, - 1, - 1, - 1, - 1280, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1280, - 1280, - 32, - 32, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 640, - 1280, - 32, - 32, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 640, - 1280, - 32, - 32, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1280, - 1280, - 8, - 8, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1280, - 1280, - 8, - 8, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 128, - 112, - 112, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 128, - 120, - 160, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 128, - 128, - 128, - 3, - 3, - 1, - 1, - 1, - 1, - 128, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 128, - 128, - 128, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 128, - 128, - 128, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 128, - 128, - 128, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 128, - 180, - 320, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 128, - 1, - 1, - 3, - 3, - 1, - 1, - 1, - 1, - 128, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 24, - 128, - 1, - 1, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 546, - 128, - 1, - 1, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 64, - 128, - 224, - 224, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 128, - 28, - 28, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 128, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 128, - 28, - 28, - 3, - 3, - 1, - 1, - 2, - 2, - 1, - 2, - 2, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 128, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 128, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 16, - 128, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 19, - 128, - 28, - 28, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 128, - 28, - 28, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 128, - 28, - 28, - 1, - 1, - 2, - 2, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 128, - 28, - 28, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 38, - 128, - 28, - 28, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 128, - 28, - 28, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 128, - 2, - 2, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 64, - 128, - 30, - 40, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 128, - 32, - 32, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 128, - 3, - 3, - 3, - 3, - 2, - 2, - 1, - 1, - 128, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 256, - 128, - 3, - 3, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 128, - 128, - 56, - 56, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 128, - 56, - 56, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 128, - 56, - 56, - 3, - 3, - 1, - 1, - 1, - 1, - 128, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 128, - 56, - 56, - 3, - 3, - 2, - 2, - 1, - 1, - 128, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 128, - 56, - 56, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 128, - 5, - 5, - 3, - 3, - 2, - 2, - 1, - 1, - 128, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 128, - 128, - 60, - 80, - 4, - 4, - 4, - 4, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 320, - 128, - 60, - 80, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 128, - 60, - 80, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 128, - 60, - 80, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 128, - 64, - 64, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 128, - 64, - 64, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 128, - 90, - 160, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 128, - 90, - 160, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 68, - 142, - 56, - 56, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 28, - 144, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 32, - 144, - 28, - 28, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 144, - 144, - 56, - 56, - 3, - 3, - 1, - 1, - 1, - 1, - 144, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 144, - 144, - 56, - 56, - 3, - 3, - 2, - 2, - 1, - 1, - 144, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 24, - 144, - 56, - 56, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 58, - 152, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 68, - 156, - 14, - 14, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 160, - 160, - 32, - 32, - 2, - 2, - 2, - 2, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 160, - 32, - 32, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 960, - 160, - 7, - 7, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 672, - 168, - 1, - 1, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 96, - 16, - 112, - 112, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 4, - 16, - 14, - 14, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 16, - 16, - 160, - 160, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 16, - 16, - 160, - 160, - 3, - 3, - 1, - 1, - 1, - 1, - 16, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 64, - 16, - 160, - 160, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 46, - 172, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 184, - 184, - 20, - 20, - 3, - 3, - 1, - 1, - 1, - 1, - 184, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 80, - 184, - 20, - 20, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 128, - 185, - 28, - 28, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1280, - 1920, - 16, - 16, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1280, - 1920, - 16, - 16, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 640, - 1920, - 32, - 32, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 640, - 1920, - 32, - 32, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 192, - 14, - 14, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 192, - 192, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 192, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 192, - 192, - 28, - 28, - 3, - 3, - 2, - 2, - 1, - 1, - 192, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 32, - 192, - 28, - 28, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 40, - 196, - 14, - 14, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 16, - 1, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 32, - 1, - 28, - 28, - 3, - 3, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 200, - 200, - 20, - 20, - 3, - 3, - 1, - 1, - 1, - 1, - 200, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 80, - 200, - 20, - 20, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 2048, - 2048, - 15, - 20, - 3, - 3, - 1, - 1, - 1, - 1, - 2048, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 2048, - 23, - 40, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 2048, - 23, - 40, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 2048, - 7, - 7, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 78, - 218, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 68, - 236, - 14, - 14, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 80, - 240, - 20, - 20, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 240, - 240, - 40, - 40, - 3, - 3, - 2, - 2, - 1, - 1, - 240, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 72, - 24, - 1, - 1, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 144, - 24, - 56, - 56, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 14, - 24, - 56, - 56, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 72, - 24, - 80, - 80, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 1280, - 2560, - 16, - 16, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1280, - 2560, - 16, - 16, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1280, - 2560, - 8, - 8, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1280, - 2560, - 8, - 8, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 256, - 10, - 10, - 3, - 3, - 2, - 2, - 1, - 1, - 256, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 128, - 256, - 112, - 112, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 256, - 120, - 160, - 3, - 3, - 1, - 1, - 1, - 1, - 256, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 150, - 256, - 128, - 128, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1024, - 256, - 14, - 14, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 256, - 14, - 14, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 256, - 14, - 14, - 1, - 1, - 2, - 2, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 256, - 14, - 14, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 256, - 16, - 16, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 256, - 180, - 320, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 256, - 180, - 320, - 1, - 1, - 2, - 2, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 256, - 180, - 320, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 20, - 256, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 256, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 256, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 256, - 28, - 28, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 256, - 28, - 28, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 256, - 28, - 28, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 256, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 24, - 256, - 2, - 2, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 256, - 256, - 2, - 2, - 3, - 3, - 1, - 1, - 1, - 1, - 256, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 546, - 256, - 2, - 2, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 64, - 256, - 2, - 2, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 128, - 256, - 32, - 32, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 256, - 32, - 32, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 256, - 32, - 32, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 256, - 3, - 3, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 24, - 256, - 3, - 3, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 256, - 256, - 3, - 3, - 3, - 3, - 1, - 1, - 1, - 1, - 256, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 546, - 256, - 3, - 3, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 1024, - 256, - 45, - 80, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 256, - 45, - 80, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 256, - 56, - 56, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 256, - 56, - 56, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 256, - 56, - 56, - 1, - 1, - 2, - 2, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 256, - 56, - 56, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 256, - 5, - 5, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 128, - 256, - 64, - 64, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 256, - 64, - 64, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 255, - 256, - 64, - 64, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 256, - 64, - 64, - 3, - 3, - 1, - 1, - 1, - 1, - 256, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 256, - 64, - 64, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 256, - 90, - 160, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 262, - 28, - 28, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 160, - 272, - 7, - 7, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 34, - 276, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 16, - 28, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 134, - 296, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 116, - 304, - 14, - 14, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 58, - 310, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 40, - 320, - 14, - 14, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 320, - 320, - 30, - 40, - 2, - 2, - 2, - 2, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 320, - 30, - 40, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 320, - 30, - 40, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 640, - 320, - 32, - 32, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 640, - 320, - 32, - 32, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 320, - 320, - 64, - 64, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 320, - 320, - 64, - 64, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 320, - 320, - 64, - 64, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 4, - 320, - 64, - 64, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1280, - 320, - 7, - 7, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 320, - 328, - 28, - 28, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 16, - 32, - 112, - 112, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 32, - 32, - 112, - 112, - 3, - 3, - 1, - 1, - 1, - 1, - 32, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 32, - 112, - 112, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 32, - 112, - 112, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 2, - 32, - 120, - 160, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 32, - 32, - 128, - 128, - 8, - 8, - 8, - 8, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 32, - 128, - 128, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 32, - 128, - 128, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 120, - 32, - 1, - 1, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 1, - 32, - 256, - 256, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 32, - 32, - 256, - 256, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 32, - 256, - 256, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 32, - 26, - 26, - 3, - 3, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 192, - 32, - 28, - 28, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 2, - 32, - 30, - 40, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 32, - 512, - 512, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 2, - 32, - 60, - 80, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 20, - 34, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 68, - 360, - 14, - 14, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 98, - 368, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 384, - 384, - 14, - 14, - 3, - 3, - 1, - 1, - 1, - 1, - 384, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 384, - 14, - 14, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 96, - 384, - 14, - 14, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 384, - 64, - 64, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1024, - 3, - 224, - 224, - 16, - 16, - 16, - 16, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 32, - 3, - 224, - 224, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 3, - 224, - 224, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 3, - 224, - 224, - 7, - 7, - 2, - 2, - 3, - 3, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 768, - 3, - 224, - 224, - 16, - 16, - 16, - 16, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 768, - 3, - 224, - 224, - 32, - 32, - 32, - 32, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 32, - 3, - 256, - 256, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 16, - 3, - 320, - 320, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 768, - 3, - 384, - 512, - 32, - 32, - 32, - 32, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 3, - 480, - 640, - 7, - 7, - 4, - 4, - 3, - 3, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 32, - 3, - 512, - 512, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 32, - 3, - 512, - 512, - 7, - 7, - 4, - 4, - 3, - 3, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 192, - 3, - 512, - 672, - 16, - 16, - 16, - 16, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 3, - 720, - 1280, - 7, - 7, - 2, - 2, - 3, - 3, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 120, - 40, - 40, - 40, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 240, - 40, - 40, - 40, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 14, - 40, - 56, - 56, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 116, - 428, - 14, - 14, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 168, - 466, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 16, - 46, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 24, - 480, - 10, - 10, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 256, - 480, - 10, - 10, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 480, - 480, - 10, - 10, - 3, - 3, - 1, - 1, - 1, - 1, - 480, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 480, - 480, - 10, - 10, - 5, - 5, - 1, - 1, - 2, - 2, - 480, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 546, - 480, - 10, - 10, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 80, - 480, - 10, - 10, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 120, - 480, - 1, - 1, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 112, - 480, - 20, - 20, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 480, - 480, - 20, - 20, - 3, - 3, - 1, - 1, - 1, - 1, - 480, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 320, - 4, - 64, - 64, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1024, - 512, - 14, - 14, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 512, - 14, - 14, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 512, - 15, - 20, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1024, - 512, - 16, - 16, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 512, - 16, - 16, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 512, - 16, - 16, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 2048, - 512, - 23, - 40, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 512, - 23, - 40, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1024, - 512, - 28, - 28, - 1, - 1, - 2, - 2, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 512, - 28, - 28, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 19, - 512, - 28, - 28, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 512, - 28, - 28, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 38, - 512, - 28, - 28, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 512, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 512, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 512, - 28, - 28, - 3, - 3, - 1, - 1, - 2, - 2, - 512, - 2, - 2, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 512, - 28, - 28, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 512, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1024, - 512, - 32, - 32, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 255, - 512, - 32, - 32, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 512, - 32, - 32, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 512, - 32, - 32, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 512, - 45, - 80, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 512, - 56, - 56, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 512, - 5, - 5, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 24, - 512, - 5, - 5, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 512, - 512, - 5, - 5, - 3, - 3, - 1, - 1, - 1, - 1, - 512, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 546, - 512, - 5, - 5, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 512, - 512, - 60, - 80, - 3, - 3, - 1, - 1, - 1, - 1, - 512, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 2048, - 512, - 7, - 7, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 512, - 512, - 7, - 7, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1024, - 512, - 90, - 160, - 1, - 1, - 2, - 2, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 512, - 90, - 160, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 512, - 90, - 160, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 196, - 544, - 14, - 14, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 24, - 54, - 56, - 56, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 576, - 576, - 14, - 14, - 3, - 3, - 1, - 1, - 1, - 1, - 576, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 576, - 576, - 14, - 14, - 3, - 3, - 2, - 2, - 1, - 1, - 576, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 96, - 576, - 14, - 14, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 160, - 576, - 7, - 7, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 20, - 58, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 28, - 62, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1280, - 640, - 16, - 16, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1280, - 640, - 16, - 16, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 640, - 640, - 32, - 32, - 3, - 3, - 1, - 1, - 1, - 1, - 640, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 640, - 640, - 32, - 32, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 640, - 640, - 32, - 32, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 640, - 640, - 32, - 32, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 320, - 640, - 64, - 64, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 320, - 640, - 64, - 64, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 640, - 640, - 64, - 64, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 160, - 640, - 7, - 7, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 64, - 112, - 112, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 64, - 112, - 112, - 3, - 3, - 2, - 2, - 1, - 1, - 64, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 64, - 120, - 160, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 32, - 64, - 120, - 160, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 64, - 120, - 160, - 8, - 8, - 8, - 8, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 64, - 128, - 128, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 64, - 128, - 128, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 384, - 64, - 14, - 14, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 64, - 160, - 160, - 3, - 3, - 2, - 2, - 1, - 1, - 64, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 256, - 64, - 180, - 320, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 64, - 180, - 320, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 64, - 180, - 320, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 64, - 1, - 1, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 1, - 64, - 224, - 224, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 64, - 224, - 224, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 64, - 256, - 256, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 32, - 64, - 256, - 256, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 32, - 64, - 256, - 256, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 64, - 2, - 2, - 3, - 3, - 2, - 2, - 1, - 1, - 64, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 32, - 64, - 30, - 40, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1, - 64, - 480, - 640, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 64, - 480, - 640, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 64, - 56, - 56, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 64, - 56, - 56, - 1, - 1, - 2, - 2, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 64, - 56, - 56, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 14, - 64, - 56, - 56, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 64, - 56, - 56, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 64, - 56, - 56, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 64, - 56, - 56, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 32, - 64, - 60, - 80, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 128, - 64, - 64, - 64, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 160, - 64, - 64, - 64, - 3, - 3, - 2, - 2, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 64, - 64, - 64, - 64, - 4, - 4, - 4, - 4, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 24, - 64, - 80, - 80, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 640, - 654, - 14, - 14, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 80, - 672, - 10, - 10, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 168, - 672, - 1, - 1, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 112, - 672, - 20, - 20, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 24, - 672, - 20, - 20, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 546, - 672, - 20, - 20, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 672, - 672, - 20, - 20, - 3, - 3, - 1, - 1, - 1, - 1, - 672, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 672, - 672, - 20, - 20, - 5, - 5, - 2, - 2, - 2, - 2, - 672, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 40, - 68, - 14, - 14, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 24, - 72, - 1, - 1, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 40, - 72, - 40, - 40, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 24, - 72, - 80, - 80, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 72, - 72, - 80, - 80, - 3, - 3, - 1, - 1, - 1, - 1, - 72, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 72, - 72, - 80, - 80, - 5, - 5, - 2, - 2, - 2, - 2, - 72, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 334, - 740, - 14, - 14, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 256, - 768, - 32, - 32, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 1024, - 782, - 7, - 7, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 16, - 78, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 34, - 78, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 24, - 78, - 56, - 56, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 272, - 800, - 7, - 7, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 480, - 80, - 10, - 10, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 184, - 80, - 20, - 20, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 200, - 80, - 20, - 20, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 480, - 80, - 20, - 20, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.float32)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.float32)], - ], - [ - 1, - 28, - 94, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 640, - 960, - 32, - 32, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 640, - 960, - 32, - 32, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 320, - 960, - 64, - 64, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 320, - 960, - 64, - 64, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 160, - 960, - 7, - 7, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 320, - 960, - 7, - 7, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 960, - 960, - 7, - 7, - 3, - 3, - 1, - 1, - 1, - 1, - 960, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 96, - 96, - 112, - 112, - 3, - 3, - 2, - 2, - 1, - 1, - 96, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 576, - 96, - 14, - 14, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 24, - 96, - 56, - 56, - 1, - 1, - 1, - 1, - 0, - 0, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - [ - 1, - 20, - 98, - 28, - 28, - 3, - 3, - 1, - 1, - 1, - 1, - 1, - 1, - 1, - False, - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - [int(ttnn.ROW_MAJOR_LAYOUT), int(ttnn.BufferType.L1), int(ttnn.bfloat16)], - [int(ttnn.TILE_LAYOUT), int(ttnn.BufferType.DRAM), int(ttnn.bfloat16)], - ], - ], - "is_conv1d": [False], - }, } diff --git a/tests/sweep_framework/sweeps/conv2d/short/conv2d_ttforge_sweep.py b/tests/sweep_framework/sweeps/conv2d/short/conv2d_ttforge_sweep.py new file mode 100644 index 00000000000..9a30494f6bb --- /dev/null +++ b/tests/sweep_framework/sweeps/conv2d/short/conv2d_ttforge_sweep.py @@ -0,0 +1,7233 @@ +# SPDX-FileCopyrightText: © 2024 Tenstorrent Inc. + +# SPDX-License-Identifier: Apache-2.0 + +from typing import Optional, Tuple, List +import os +import itertools +import random +import torch + +import ttnn + +from tests.ttnn.utils_for_testing import check_with_pcc, start_measuring_time, stop_measuring_time +from models.utility_functions import torch_random +from tests.sweep_framework.sweep_utils.conv2d_common import ( + run_conv2d_short_sweep, + mesh_device_fixture, +) + +parameters = { + "ttforge_sweep_conv2d": { + "input_specs": [ + # Contains following params + # [batch_size, output_channels, input_channels, input_height, input_width, kernel_height, kernel_width, stride_h, stride_w, pad_h, pad_w, groups, dilation_h, dilation_w, bias, [input_layout, input_memory_config, input_datatype], [weight_layout, weight_memory_config, weight_datatype], [output_layout, output_memory_config, output_datatype]] + [ + 1, + 1024, + 1024, + 14, + 14, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 2048, + 1024, + 14, + 14, + 1, + 1, + 2, + 2, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 1024, + 14, + 14, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 1024, + 14, + 14, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1024, + 1024, + 16, + 16, + 3, + 3, + 1, + 1, + 1, + 1, + 1024, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 255, + 1024, + 16, + 16, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 1024, + 16, + 16, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 1024, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 2048, + 1024, + 45, + 80, + 1, + 1, + 2, + 2, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 1024, + 45, + 80, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 1024, + 45, + 80, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 40, + 102, + 56, + 56, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 462, + 1072, + 7, + 7, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 672, + 112, + 20, + 20, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 40, + 116, + 14, + 14, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 34, + 118, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 32, + 120, + 1, + 1, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 480, + 120, + 1, + 1, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 120, + 120, + 40, + 40, + 5, + 5, + 1, + 1, + 2, + 2, + 120, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 40, + 120, + 40, + 40, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 46, + 122, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 124, + 56, + 56, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1280, + 1280, + 16, + 16, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1280, + 1280, + 16, + 16, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1280, + 1280, + 16, + 16, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1280, + 1280, + 30, + 40, + 3, + 3, + 1, + 1, + 1, + 1, + 1280, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1280, + 1280, + 32, + 32, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 640, + 1280, + 32, + 32, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 640, + 1280, + 32, + 32, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1280, + 1280, + 8, + 8, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1280, + 1280, + 8, + 8, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 128, + 112, + 112, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 128, + 120, + 160, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 128, + 128, + 128, + 3, + 3, + 1, + 1, + 1, + 1, + 128, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 128, + 128, + 128, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 128, + 128, + 128, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 128, + 128, + 128, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 128, + 180, + 320, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 128, + 1, + 1, + 3, + 3, + 1, + 1, + 1, + 1, + 128, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 24, + 128, + 1, + 1, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 546, + 128, + 1, + 1, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 64, + 128, + 224, + 224, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 128, + 28, + 28, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 128, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 128, + 28, + 28, + 3, + 3, + 1, + 1, + 2, + 2, + 1, + 2, + 2, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 128, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 128, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 16, + 128, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 19, + 128, + 28, + 28, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 128, + 28, + 28, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 128, + 28, + 28, + 1, + 1, + 2, + 2, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 128, + 28, + 28, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 38, + 128, + 28, + 28, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 128, + 28, + 28, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 128, + 2, + 2, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 64, + 128, + 30, + 40, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 128, + 32, + 32, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 128, + 3, + 3, + 3, + 3, + 2, + 2, + 1, + 1, + 128, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 256, + 128, + 3, + 3, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 128, + 128, + 56, + 56, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 128, + 56, + 56, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 128, + 56, + 56, + 3, + 3, + 1, + 1, + 1, + 1, + 128, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 128, + 56, + 56, + 3, + 3, + 2, + 2, + 1, + 1, + 128, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 128, + 56, + 56, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 128, + 5, + 5, + 3, + 3, + 2, + 2, + 1, + 1, + 128, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 128, + 128, + 60, + 80, + 4, + 4, + 4, + 4, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 320, + 128, + 60, + 80, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 128, + 60, + 80, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 128, + 60, + 80, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 128, + 64, + 64, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 128, + 64, + 64, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 128, + 90, + 160, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 128, + 90, + 160, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 68, + 142, + 56, + 56, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 28, + 144, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 32, + 144, + 28, + 28, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 144, + 144, + 56, + 56, + 3, + 3, + 1, + 1, + 1, + 1, + 144, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 144, + 144, + 56, + 56, + 3, + 3, + 2, + 2, + 1, + 1, + 144, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 24, + 144, + 56, + 56, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 58, + 152, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 68, + 156, + 14, + 14, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 160, + 160, + 32, + 32, + 2, + 2, + 2, + 2, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 160, + 32, + 32, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 960, + 160, + 7, + 7, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 672, + 168, + 1, + 1, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 96, + 16, + 112, + 112, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 4, + 16, + 14, + 14, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 16, + 16, + 160, + 160, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 16, + 16, + 160, + 160, + 3, + 3, + 1, + 1, + 1, + 1, + 16, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 64, + 16, + 160, + 160, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 46, + 172, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 184, + 184, + 20, + 20, + 3, + 3, + 1, + 1, + 1, + 1, + 184, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 80, + 184, + 20, + 20, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 128, + 185, + 28, + 28, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1280, + 1920, + 16, + 16, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1280, + 1920, + 16, + 16, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 640, + 1920, + 32, + 32, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 640, + 1920, + 32, + 32, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 192, + 14, + 14, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 192, + 192, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 192, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 192, + 192, + 28, + 28, + 3, + 3, + 2, + 2, + 1, + 1, + 192, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 32, + 192, + 28, + 28, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 40, + 196, + 14, + 14, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 16, + 1, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 32, + 1, + 28, + 28, + 3, + 3, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 200, + 200, + 20, + 20, + 3, + 3, + 1, + 1, + 1, + 1, + 200, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 80, + 200, + 20, + 20, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 2048, + 2048, + 15, + 20, + 3, + 3, + 1, + 1, + 1, + 1, + 2048, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 2048, + 23, + 40, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 2048, + 23, + 40, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 2048, + 7, + 7, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 78, + 218, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 68, + 236, + 14, + 14, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 80, + 240, + 20, + 20, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 240, + 240, + 40, + 40, + 3, + 3, + 2, + 2, + 1, + 1, + 240, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 72, + 24, + 1, + 1, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 144, + 24, + 56, + 56, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 14, + 24, + 56, + 56, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 72, + 24, + 80, + 80, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 1280, + 2560, + 16, + 16, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1280, + 2560, + 16, + 16, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1280, + 2560, + 8, + 8, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1280, + 2560, + 8, + 8, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 256, + 10, + 10, + 3, + 3, + 2, + 2, + 1, + 1, + 256, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 128, + 256, + 112, + 112, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 256, + 120, + 160, + 3, + 3, + 1, + 1, + 1, + 1, + 256, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 150, + 256, + 128, + 128, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1024, + 256, + 14, + 14, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 256, + 14, + 14, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 256, + 14, + 14, + 1, + 1, + 2, + 2, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 256, + 14, + 14, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 256, + 16, + 16, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 256, + 180, + 320, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 256, + 180, + 320, + 1, + 1, + 2, + 2, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 256, + 180, + 320, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 20, + 256, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 256, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 256, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 256, + 28, + 28, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 256, + 28, + 28, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 256, + 28, + 28, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 256, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 24, + 256, + 2, + 2, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 256, + 256, + 2, + 2, + 3, + 3, + 1, + 1, + 1, + 1, + 256, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 546, + 256, + 2, + 2, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 64, + 256, + 2, + 2, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 128, + 256, + 32, + 32, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 256, + 32, + 32, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 256, + 32, + 32, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 256, + 3, + 3, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 24, + 256, + 3, + 3, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 256, + 256, + 3, + 3, + 3, + 3, + 1, + 1, + 1, + 1, + 256, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 546, + 256, + 3, + 3, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 1024, + 256, + 45, + 80, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 256, + 45, + 80, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 256, + 56, + 56, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 256, + 56, + 56, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 256, + 56, + 56, + 1, + 1, + 2, + 2, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 256, + 56, + 56, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 256, + 5, + 5, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 128, + 256, + 64, + 64, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 256, + 64, + 64, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 255, + 256, + 64, + 64, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 256, + 64, + 64, + 3, + 3, + 1, + 1, + 1, + 1, + 256, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 256, + 64, + 64, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 256, + 90, + 160, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 262, + 28, + 28, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 160, + 272, + 7, + 7, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 34, + 276, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 16, + 28, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 134, + 296, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 116, + 304, + 14, + 14, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 58, + 310, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 40, + 320, + 14, + 14, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 320, + 320, + 30, + 40, + 2, + 2, + 2, + 2, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 320, + 30, + 40, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 320, + 30, + 40, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 640, + 320, + 32, + 32, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 640, + 320, + 32, + 32, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 320, + 320, + 64, + 64, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 320, + 320, + 64, + 64, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 320, + 320, + 64, + 64, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 4, + 320, + 64, + 64, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1280, + 320, + 7, + 7, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 320, + 328, + 28, + 28, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 16, + 32, + 112, + 112, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 32, + 32, + 112, + 112, + 3, + 3, + 1, + 1, + 1, + 1, + 32, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 32, + 112, + 112, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 32, + 112, + 112, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 2, + 32, + 120, + 160, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 32, + 32, + 128, + 128, + 8, + 8, + 8, + 8, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 32, + 128, + 128, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 32, + 128, + 128, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 120, + 32, + 1, + 1, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 1, + 32, + 256, + 256, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 32, + 32, + 256, + 256, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 32, + 256, + 256, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 32, + 26, + 26, + 3, + 3, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 192, + 32, + 28, + 28, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 2, + 32, + 30, + 40, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 32, + 512, + 512, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 2, + 32, + 60, + 80, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 20, + 34, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 68, + 360, + 14, + 14, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 98, + 368, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 384, + 384, + 14, + 14, + 3, + 3, + 1, + 1, + 1, + 1, + 384, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 384, + 14, + 14, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 96, + 384, + 14, + 14, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 384, + 64, + 64, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1024, + 3, + 224, + 224, + 16, + 16, + 16, + 16, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 32, + 3, + 224, + 224, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 3, + 224, + 224, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 3, + 224, + 224, + 7, + 7, + 2, + 2, + 3, + 3, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 768, + 3, + 224, + 224, + 16, + 16, + 16, + 16, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 768, + 3, + 224, + 224, + 32, + 32, + 32, + 32, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 32, + 3, + 256, + 256, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 16, + 3, + 320, + 320, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 768, + 3, + 384, + 512, + 32, + 32, + 32, + 32, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 3, + 480, + 640, + 7, + 7, + 4, + 4, + 3, + 3, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 32, + 3, + 512, + 512, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 32, + 3, + 512, + 512, + 7, + 7, + 4, + 4, + 3, + 3, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 192, + 3, + 512, + 672, + 16, + 16, + 16, + 16, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 3, + 720, + 1280, + 7, + 7, + 2, + 2, + 3, + 3, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 120, + 40, + 40, + 40, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 240, + 40, + 40, + 40, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 14, + 40, + 56, + 56, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 116, + 428, + 14, + 14, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 168, + 466, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 16, + 46, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 24, + 480, + 10, + 10, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 256, + 480, + 10, + 10, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 480, + 480, + 10, + 10, + 3, + 3, + 1, + 1, + 1, + 1, + 480, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 480, + 480, + 10, + 10, + 5, + 5, + 1, + 1, + 2, + 2, + 480, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 546, + 480, + 10, + 10, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 80, + 480, + 10, + 10, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 120, + 480, + 1, + 1, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 112, + 480, + 20, + 20, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 480, + 480, + 20, + 20, + 3, + 3, + 1, + 1, + 1, + 1, + 480, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 320, + 4, + 64, + 64, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1024, + 512, + 14, + 14, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 512, + 14, + 14, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 512, + 15, + 20, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1024, + 512, + 16, + 16, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 512, + 16, + 16, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 512, + 16, + 16, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 2048, + 512, + 23, + 40, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 512, + 23, + 40, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1024, + 512, + 28, + 28, + 1, + 1, + 2, + 2, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 512, + 28, + 28, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 19, + 512, + 28, + 28, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 512, + 28, + 28, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 38, + 512, + 28, + 28, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 512, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 512, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 512, + 28, + 28, + 3, + 3, + 1, + 1, + 2, + 2, + 512, + 2, + 2, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 512, + 28, + 28, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 512, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1024, + 512, + 32, + 32, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 255, + 512, + 32, + 32, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 512, + 32, + 32, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 512, + 32, + 32, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 512, + 45, + 80, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 512, + 56, + 56, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 512, + 5, + 5, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 24, + 512, + 5, + 5, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 512, + 512, + 5, + 5, + 3, + 3, + 1, + 1, + 1, + 1, + 512, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 546, + 512, + 5, + 5, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 512, + 512, + 60, + 80, + 3, + 3, + 1, + 1, + 1, + 1, + 512, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 2048, + 512, + 7, + 7, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 512, + 512, + 7, + 7, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1024, + 512, + 90, + 160, + 1, + 1, + 2, + 2, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 512, + 90, + 160, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 512, + 90, + 160, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 196, + 544, + 14, + 14, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 24, + 54, + 56, + 56, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 576, + 576, + 14, + 14, + 3, + 3, + 1, + 1, + 1, + 1, + 576, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 576, + 576, + 14, + 14, + 3, + 3, + 2, + 2, + 1, + 1, + 576, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 96, + 576, + 14, + 14, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 160, + 576, + 7, + 7, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 20, + 58, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 28, + 62, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1280, + 640, + 16, + 16, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1280, + 640, + 16, + 16, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 640, + 640, + 32, + 32, + 3, + 3, + 1, + 1, + 1, + 1, + 640, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 640, + 640, + 32, + 32, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 640, + 640, + 32, + 32, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 640, + 640, + 32, + 32, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 320, + 640, + 64, + 64, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 320, + 640, + 64, + 64, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 640, + 640, + 64, + 64, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 160, + 640, + 7, + 7, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 64, + 112, + 112, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 64, + 112, + 112, + 3, + 3, + 2, + 2, + 1, + 1, + 64, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 64, + 120, + 160, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 32, + 64, + 120, + 160, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 64, + 120, + 160, + 8, + 8, + 8, + 8, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 64, + 128, + 128, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 64, + 128, + 128, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 384, + 64, + 14, + 14, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 64, + 160, + 160, + 3, + 3, + 2, + 2, + 1, + 1, + 64, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 256, + 64, + 180, + 320, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 64, + 180, + 320, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 64, + 180, + 320, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 64, + 1, + 1, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 1, + 64, + 224, + 224, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 64, + 224, + 224, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 64, + 256, + 256, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 32, + 64, + 256, + 256, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 32, + 64, + 256, + 256, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 64, + 2, + 2, + 3, + 3, + 2, + 2, + 1, + 1, + 64, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 32, + 64, + 30, + 40, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1, + 64, + 480, + 640, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 64, + 480, + 640, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 64, + 56, + 56, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 64, + 56, + 56, + 1, + 1, + 2, + 2, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 64, + 56, + 56, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 14, + 64, + 56, + 56, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 64, + 56, + 56, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 64, + 56, + 56, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 64, + 56, + 56, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 32, + 64, + 60, + 80, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 128, + 64, + 64, + 64, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 160, + 64, + 64, + 64, + 3, + 3, + 2, + 2, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 64, + 64, + 64, + 64, + 4, + 4, + 4, + 4, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 24, + 64, + 80, + 80, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 640, + 654, + 14, + 14, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 80, + 672, + 10, + 10, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 168, + 672, + 1, + 1, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 112, + 672, + 20, + 20, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 24, + 672, + 20, + 20, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 546, + 672, + 20, + 20, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 672, + 672, + 20, + 20, + 3, + 3, + 1, + 1, + 1, + 1, + 672, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 672, + 672, + 20, + 20, + 5, + 5, + 2, + 2, + 2, + 2, + 672, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 40, + 68, + 14, + 14, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 24, + 72, + 1, + 1, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 40, + 72, + 40, + 40, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 24, + 72, + 80, + 80, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 72, + 72, + 80, + 80, + 3, + 3, + 1, + 1, + 1, + 1, + 72, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 72, + 72, + 80, + 80, + 5, + 5, + 2, + 2, + 2, + 2, + 72, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 334, + 740, + 14, + 14, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 256, + 768, + 32, + 32, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 1024, + 782, + 7, + 7, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 16, + 78, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 34, + 78, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 24, + 78, + 56, + 56, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 272, + 800, + 7, + 7, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 480, + 80, + 10, + 10, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 184, + 80, + 20, + 20, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 200, + 80, + 20, + 20, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 480, + 80, + 20, + 20, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.float32)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.float32)], + ], + [ + 1, + 28, + 94, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 640, + 960, + 32, + 32, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 640, + 960, + 32, + 32, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 320, + 960, + 64, + 64, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 320, + 960, + 64, + 64, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 160, + 960, + 7, + 7, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 320, + 960, + 7, + 7, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 960, + 960, + 7, + 7, + 3, + 3, + 1, + 1, + 1, + 1, + 960, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 96, + 96, + 112, + 112, + 3, + 3, + 2, + 2, + 1, + 1, + 96, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 576, + 96, + 14, + 14, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 24, + 96, + 56, + 56, + 1, + 1, + 1, + 1, + 0, + 0, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + [ + 1, + 20, + 98, + 28, + 28, + 3, + 3, + 1, + 1, + 1, + 1, + 1, + 1, + 1, + False, + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + [int(ttnn.ROW_MAJOR_LAYOUT), "system_memory", int(ttnn.bfloat16)], + [int(ttnn.TILE_LAYOUT), "dram", int(ttnn.bfloat16)], + ], + ], + "is_conv1d": [False], + }, +} + + +def invalidate_vector(test_vector) -> Tuple[bool, Optional[str]]: + return False, None + + +def run( + input_specs, + is_conv1d=False, + *, + device, +) -> list: + return run_conv2d_short_sweep(input_specs, device) + + +import pytest + + +@pytest.mark.parametrize("input_spec", parameters["ttforge_sweep_conv2d"]["input_specs"]) +@pytest.mark.parametrize("device_params", [{"l1_small_size": 16384}], indirect=True) +def test_conv2d_localrun(device, input_spec): + run_conv2d_short_sweep( + input_spec, + device, + )