TransformerML: make positionsTensor created on consistent device (#467)
TransformerML: make positionsTensor created on consistent device (#467)


tensorflow/swift-models,tensorflow/swift-models,tensorflow/swift-models,tensorflow/swift-models