Skip to content

Commit

Permalink
#5079: add some comments
Browse files Browse the repository at this point in the history
  • Loading branch information
dvartaniansTT committed Mar 20, 2024
1 parent f8baef3 commit 79eb088
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/ttnn/unit_tests/operations/test_conv2d.py
Original file line number Diff line number Diff line change
Expand Up @@ -1078,7 +1078,7 @@ def test_unet_conv_wh(
"batch_size, output_channels, input_channels, input_height, input_width, filter_height, filter_width, stride_h, stride_w, pad_h, pad_w, use_1d_systolic_array, config_override, use_shallow_conv_variant",
(
# yolov4 convs with batch size 1
# unique convs in yolov4 (complete list) # groups: number
# unique convs in yolov4 (complete list) # groups: 1 groups > 1 are commented out for now until they are supported.
(1, 32, 3, 480, 640, 3, 3, 1, 1, 1, 1, True, None, False), # groups: 1
(1, 64, 32, 480, 640, 3, 3, 1, 1, 1, 1, True, None, False), # groups: 1
(1, 32, 64, 480, 640, 1, 1, 1, 1, 0, 0, True, None, False), # groups: 1
Expand Down

0 comments on commit 79eb088

Please sign in to comment.