(rt_input, out_type)
| 161 | |
| 162 | |
| 163 | def _nrows(rt_input, out_type): |
| 164 | if isinstance(rt_input, ragged_tensor.RaggedTensor): |
| 165 | return rt_input.nrows(out_type=out_type) |
| 166 | else: |
| 167 | return array_ops.shape(rt_input, out_type=out_type)[0] |
no test coverage detected