# shapes from DLRM-TeraByte
16x13:13x512:16x512
16x512:512x256:16x256
16x256:256x128:16x128
16x479:479x1024:16x1024
16x1024:1024x1024:16x1024
16x1024:1024x512:16x512

