tensorflow.python.framework.errors_impl.InvalidArgumentError: Cannot assign a device for operation model/transformer/position_encoding/embedding_lookup #460