aa5f9ddfdf0447c7a53f4e2676a1c866189315cd,kfac/python/ops/curvature_matrix_vector_products.py,CurvatureMatrixVectorProductComputer,_total_loss,#CurvatureMatrixVectorProductComputer#,74

Before Change

  @property
  def _total_loss(self):
    return tf.add_n(tuple(loss.evaluate() for loss in self._losses))

  // Jacobian multiplication functions:
  def _multiply_jacobian(self, vecs):
    Multiply vecs by the Jacobian of losses.

After Change

  @property
  def _total_loss(self):
    vals = []
    for loss in self._losses:
      with tf.colocate_with(self._loss_colocation_ops[loss]):
        vals.append(loss.evaluate())
    return tf.add_n(tuple(vals))

  // Jacobian multiplication functions:
  def _multiply_jacobian(self, vecs):
    Multiply vecs by the Jacobian of losses.

Instances