1# These problems are ported from corresponding inner product forward 2# ip mb=128 3128x768:768x2304n"BERT:0*1" 4128x768:768x768n"BERT:1*1" 5128x768:768x3072n"BERT:2*1" 6128x3072:3072x768n"BERT:3*1" 7