Commit 370e5217 authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Adds a fisher block for fully connected recurrent layers.

`FullyConnectedSeriesFB` uses an approximation to the Fisher information
matrix designed for RNNs. This CL also adds support for dtypes
other than `float32` to `fisher_factors`.

PiperOrigin-RevId: 177558080
parent e747fc91
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment