(e)
| 314 | |
| 315 | # Stack TensorArrays for Tensor outputs, and concat RaggedTensor outputs. |
| 316 | def _stack_or_concat(e): |
| 317 | if isinstance(e, _RaggedTensorComponents): |
| 318 | return _concat_ragged_tensor_components(e) |
| 319 | else: |
| 320 | result = e.stack() |
| 321 | return result |
| 322 | |
| 323 | results_flat_components = [ |
| 324 | _stack_or_concat(e) for e in results_as_components |
no test coverage detected