HomeSort by relevance Sort by last modified time
    Searched refs:new_cov (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/contrib/kfac/python/kernel_tests/
fisher_factors_test.py 386 new_cov = sess.run(factor.make_covariance_update_op(.5))
387 self.assertAllClose([[0.75, 0.5], [0.5, 1.5]], new_cov)
416 new_cov = sess.run(factor.make_covariance_update_op(.5))
417 self.assertAllClose([[0.75], [1.5]], new_cov)
439 new_cov = sess.run(cov_update_op)
440 self.assertAllClose(np.array([1., 1., 0., 0., 1.]) / 3., new_cov)
472 new_cov = sess.run(factor.make_covariance_update_op(.5))
473 self.assertAllClose([[3, 3.5, 1], [3.5, 5.5, 1.5], [1, 1.5, 1]], new_cov)
482 new_cov = sess.run(factor.make_covariance_update_op(.5))
483 self.assertAllClose([[3, 3.5], [3.5, 5.5]], new_cov)
    [all...]
  /external/tensorflow/tensorflow/contrib/kfac/python/ops/
fisher_factors.py 311 new_cov = math_ops.add_n(new_cov_contribs)
314 new_cov = utils.cross_replica_mean(new_cov)
316 self._cov, new_cov, ema_decay, zero_debias=ZERO_DEBIAS)
829 new_cov = math_ops.reduce_sum(one_hots, axis=0) # [vocab_size]
830 new_cov /= math_ops.cast(batch_size, new_cov.dtype)
832 return new_cov
899 new_cov = math_ops.matmul(
903 new_cov /= math_ops.cast(self._batch_size, new_cov.dtype
    [all...]