Fix some seq_length refs rather than relying on tensor shapes
Fix some seq_length refs rather than relying on tensor shapes

stanfordnlp/spinn,nyu-mll/spinn,nyu-mll/spinn,stanfordnlp/spinn,nyu-mll/spinn,stanfordnlp/spinn