(from_tensor)
| 167 | return eff_encoder_result |
| 168 | |
| 169 | def _tf_body(from_tensor): |
| 170 | tf_encoder_result = tf_bert(input_tensor=from_tensor, |
| 171 | encoder_args=encoder_args, |
| 172 | attention_mask=attention_mask) |
| 173 | return tf_encoder_result |
| 174 | |
| 175 | tf_while_tensor = tf.while_loop(_cond, |
| 176 | _tf_body, |