K-FAC: Support for tf.AUTO_REUSE when re-using registrations. Multi-tower...
K-FAC: Support for tf.AUTO_REUSE when re-using registrations. Multi-tower support for FullFB, NaiveDiagonalFB. Removal of LayerCollection.generic_registrations. PiperOrigin-RevId: 174092003
Loading
Please sign in to comment