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

  /external/tensorflow/tensorflow/contrib/kfac/python/ops/
fisher_blocks_lib.py 22 from tensorflow.contrib.kfac.python.ops.fisher_blocks import *
layer_collection.py 34 from tensorflow.contrib.kfac.python.ops import fisher_blocks as fb
128 fisher_blocks: a LayersParamsDict (subclass of OrderedDict) mapping layer
138 self.fisher_blocks = LayerParametersDict()
164 in six.iteritems(self.fisher_blocks))
262 layer_key in self.fisher_blocks):
263 result = self.fisher_blocks[layer_key]
269 if reuse is False and layer_key in self.fisher_blocks:
273 # Insert fisher_block into self.fisher_blocks.
274 if layer_key in self.fisher_blocks:
280 for (params, block) in self.fisher_blocks.items(
    [all...]
estimator.py 205 for params, fb in self._layers.fisher_blocks.items():
  /external/tensorflow/tensorflow/contrib/kfac/python/kernel_tests/
layer_collection_test.py 21 from tensorflow.contrib.kfac.python.ops import fisher_blocks
138 lc.fisher_blocks = {
147 lc.fisher_blocks = {x: '1'}
156 lc.fisher_blocks = {(x, y): '1'}
165 lc.fisher_blocks = {
177 lc.fisher_blocks = {(x, y): '1'}
188 lc.fisher_blocks = {(x, y, z): '1'}
199 lc.fisher_blocks = {x: MockFisherBlock('1'), z: MockFisherBlock('2')}
211 lc.fisher_blocks = {(x, z): '1', (z, w): '2'}
444 self.assertEqual(lc.fisher_blocks[w].num_registered_minibatches, 1
    [all...]
fisher_blocks_test.py 15 """Tests for tf.contrib.kfac.fisher_blocks."""
23 from tensorflow.contrib.kfac.python.ops import fisher_blocks as fb
    [all...]
  /external/tensorflow/tensorflow/contrib/kfac/
__init__.py 24 from tensorflow.contrib.kfac.python.ops import fisher_blocks_lib as fisher_blocks
37 "fisher_blocks",
  /external/tensorflow/tensorflow/contrib/kfac/examples/tests/
convnet_test.py 74 self.assertEqual(len(layer_collection.fisher_blocks), 3)

Completed in 77 milliseconds