Searched refs:new_diag (Results 1 – 3 of 3) sorted by relevance
258 new_diag = self._diag + x_diag259 return array_ops.matrix_set_diag(x, new_diag)
411 new_diag = 1 + mat_diag412 return array_ops.matrix_set_diag(mat, new_diag)752 new_diag = multiplier_vector + mat_diag754 return array_ops.matrix_set_diag(mat, new_diag)
99 auto new_diag = in TestSetMatrixDiagonal() local106 ComputeAndCompareR2<T>(&builder, kv.second, {a_data.get(), new_diag.get()}); in TestSetMatrixDiagonal()