Fix bug in TE BERT model, create different instances for each layer during init.
Attach a file by drag & drop or click to upload
