The node you are looking for is at XREFoptimize_diagonal_matrix.