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

  /external/tensorflow/tensorflow/contrib/kfac/python/ops/
fisher_blocks.py 530 instance members _input_factor and _output_factor.
537 self._output_factor.get_cov(),
541 self._output_factor.register_damped_inverse(self._output_damping)
557 reshaped_out = self._output_factor.right_multiply_inverse(
569 reshaped_out = self._output_factor.right_multiply(
588 right_factor = self._output_factor.get_cov()
637 self._output_factor = self._layer_collection.make_or_get_factor( #
703 self._output_factor = self._layer_collection.make_or_get_factor( #
791 self._output_factor = self._layer_collection.make_or_get_factor(
900 self._output_factor = self._layer_collection.make_or_get_factor
    [all...]
  /external/tensorflow/tensorflow/contrib/kfac/python/kernel_tests/
fisher_blocks_test.py 485 sess.run(block._output_factor.make_inverse_update_ops())
511 sess.run(block._output_factor.make_inverse_update_ops())
533 sess.run(state_ops.assign(block._output_factor._cov, _make_psd(2)))
535 sess.run(block._output_factor.make_inverse_update_ops())
767 sess.run(block._output_factor.make_inverse_update_ops())
794 sess.run(block._output_factor.make_inverse_update_ops())
    [all...]

Completed in 71 milliseconds