aa5f9ddfdf0447c7a53f4e2676a1c866189315cd,kfac/python/ops/curvature_matrix_vector_products.py,CurvatureMatrixVectorProductComputer,_multiply_loss_hessian_factor,#CurvatureMatrixVectorProductComputer#Any#,123

Before Change

    Multiply loss_inner_vecs by factor of Hessian of total loss.
    return tuple(
        loss.multiply_hessian_factor(loss_vec)
        for loss, loss_vec in zip(self._losses, loss_inner_vecs))

  def _multiply_loss_hessian_factor_transpose(self, loss_vecs):
    Multiply loss_vecs by transpose factor of Hessian of total loss.

After Change

  def _multiply_loss_hessian_factor(self, loss_inner_vecs):
    Multiply loss_inner_vecs by factor of Hessian of total loss.
    mult_func = lambda loss, vec: loss.multiply_hessian_factor(vec)
    return self._multiply_across_losses(mult_func, loss_inner_vecs)

  def _multiply_loss_hessian_factor_transpose(self, loss_vecs):
    Multiply loss_vecs by transpose factor of Hessian of total loss.
    mult_func = lambda loss, vec: loss.multiply_hessian_factor_transpose(vec)

Instances