18x16x54x64:18x16x64x54
11x16x45x45:11x16x45x64
21x16x41x41:21x16x41x64
16x16x49x49:16x16x49x64
14x16x54x54:14x16x54x64
5x16x38x38:5x16x38x64
24x16x32x32:24x16x32x64
13x16x45x64:13x16x64x45
17x16x41x64:17x16x64x41
21x16x49x64:21x16x64x49

# Broadcast shapes
2x16x384x384:2x1x384x64
1x1x35x64:13x16x64x35
1x16x38x64:5x1x64x38
14x16x54x64:1x1x64x54n"B_full_bcast"
14x6x1x253:1x1x253x1n"dot_prod_w_B_full_bcast"
