(from_tensor)
| 160 | return op_encoder_result |
| 161 | |
| 162 | def _eff_body(from_tensor): |
| 163 | eff_encoder_result = ft_bert(inputs=from_tensor, |
| 164 | encoder_args=eff_encoder_args, |
| 165 | encoder_vars_dict=encoder_variables_dict, |
| 166 | sequence_length=sequence_length) |
| 167 | return eff_encoder_result |
| 168 | |
| 169 | def _tf_body(from_tensor): |
| 170 | tf_encoder_result = tf_bert(input_tensor=from_tensor, |