Convert outputs which are `Tensor`s into `_RaggedTensorComponents`.
(fn_output_flat, output_declared)
| 419 | |
| 420 | |
| 421 | def _convert_declared(fn_output_flat, output_declared): |
| 422 | """Convert outputs which are `Tensor`s into `_RaggedTensorComponents`.""" |
| 423 | for current, declared in zip(fn_output_flat, output_declared): |
| 424 | if isinstance(declared, ragged_tensor.RaggedTensorType): |
| 425 | yield _convert_declared_ragged(current, declared) |
| 426 | else: |
| 427 | yield current |
| 428 | |
| 429 | |
| 430 | def _convert_declared_ragged(current, declared): |
no test coverage detected