Design tweaks of Fisher Factor classes. Now they don't declare ops in their construc...
authorA. Unique TensorFlower <gardener@tensorflow.org>
Mon, 31 Dec 2018 17:53:18 +0000 (09:53 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Sun, 31 Dec 2017 17:57:05 +0000 (09:57 -0800)
commit13fc601fa38c05d9384dbf657d0ec0555c03e140
tree7abcaba50f189123a939423c13155544b9de86a7
parent2cdbdee8f4059f65e3dd7f96bb328f790df547f6
Design tweaks of Fisher Factor classes.  Now they don't declare ops in their constructors, except possibly for making the cov variables (is this considered an op?).  This should allow for easier control over device placement of various ops in the future.

Fixed some problems with how colocations were done for ops computed in the base class and in the RNN class.

Colocations now controlled with global configuration variable (similar to the rest of the configuration of the FisherFactor classes).

PiperOrigin-RevId: 180441903
tensorflow/contrib/kfac/python/kernel_tests/fisher_factors_test.py
tensorflow/contrib/kfac/python/ops/fisher_factors.py
tensorflow/contrib/kfac/python/ops/layer_collection.py
tensorflow/contrib/kfac/python/ops/utils.py
tensorflow/contrib/kfac/python/ops/utils_lib.py