(pfor_input)
| 1815 | # different (for compatibility with V1), so we cannot use _convert_identity. |
| 1816 | @RegisterPFor("MatrixDiagV2") |
| 1817 | def _convert_matrix_diag_v2(pfor_input): |
| 1818 | diagonal = pfor_input.stacked_input(0) |
| 1819 | k = pfor_input.unstacked_input(1) |
| 1820 | num_rows = pfor_input.unstacked_input(2) |
| 1821 | num_cols = pfor_input.unstacked_input(3) |
| 1822 | padding_value = pfor_input.unstacked_input(4) |
| 1823 | return wrap( |
| 1824 | array_ops.matrix_diag( |
| 1825 | diagonal, |
| 1826 | k=k, |
| 1827 | num_rows=num_rows, |
| 1828 | num_cols=num_cols, |
| 1829 | padding_value=padding_value), True) |
| 1830 | |
| 1831 | |
| 1832 | # See notes for MatrixDiagV2 |
nothing calls this directly
no test coverage detected