Module: tf.contrib.kfac.layer_collection

Defined in tensorflow/contrib/kfac/python/ops/layer_collection_lib.py.

Registry for layers and their parameters/variables.

This represents the collection of all layers in the approximate Fisher information matrix to which a particular FisherBlock may belong. That is, we might have several layer collections for one TF graph (if we have multiple K-FAC optimizers being used, for example.)

Classes

class LayerCollection: Registry of information about layers and losses.

class LayerParametersDict: An OrderedDict where keys are Tensors or tuples of Tensors.

Functions

get_default_layer_collection(...): Get default LayerCollection.

set_default_layer_collection(...)

Other Members

APPROX_DIAGONAL_NAME

APPROX_FULL_NAME

APPROX_KRONECKER_INDEP_NAME

APPROX_KRONECKER_NAME

APPROX_KRONECKER_SERIES_1_NAME

APPROX_KRONECKER_SERIES_2_NAME

VARIABLE_SCOPE

__cached__

__loader__

__spec__