Searched defs:pruning (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
H A Drnn_cells_test.py21 from tensorflow.contrib.model_pruning.python import pruning namespace
50 self.assertEqual(len(pruning.get_masks()), expected_num_masks)
51 self.assertEqual(len(pruning.get_masked_weights()), expected_num_masks)
52 self.assertEqual(len(pruning.get_thresholds()), expected_num_masks)
53 self.assertEqual(len(pruning.get_weights()), expected_num_masks)
55 for mask in pruning.get_masks():
57 for weight in pruning.get_weights():
74 self.assertEqual(len(pruning.get_masks()), expected_num_masks)
75 self.assertEqual(len(pruning.get_masked_weights()), expected_num_masks)
76 self.assertEqual(len(pruning
[all...]
/external/tensorflow/tensorflow/contrib/model_pruning/examples/cifar10/
H A Dcifar10_train.py50 from tensorflow.contrib.model_pruning.python import pruning namespace
74 # Parse pruning hyperparameters
75 pruning_hparams = pruning.get_pruning_hparams().parse(FLAGS.pruning_hparams)
77 # Create a pruning object using the pruning hyperparameters
78 pruning_obj = pruning.Pruning(pruning_hparams, global_step=global_step)
83 # the pruning spec proto
146 help="""Comma separated list of pruning-related hyperparameters""")
H A Dcifar10_pruning.py15 """Builds the CIFAR-10 network with additional variables to support pruning.
46 from tensorflow.contrib.model_pruning.python import pruning namespace
185 # variables to the layer by calling the pruning.apply_mask() function.
193 images, pruning.apply_mask(kernel, scope), [1, 1, 1, 1], padding='SAME')
215 norm1, pruning.apply_mask(kernel, scope), [1, 1, 1, 1], padding='SAME')
241 tf.matmul(reshape, pruning.apply_mask(weights, scope)) + biases,
251 tf.matmul(local3, pruning.apply_mask(weights, scope)) + biases,
265 tf.matmul(local4, pruning.apply_mask(weights, scope)),
/external/tensorflow/tensorflow/contrib/model_pruning/python/
H A Dpruning_test.py15 """Tests for the key functions in pruning library."""
23 from tensorflow.contrib.model_pruning.python import pruning namespace
50 self.pruning_hparams = pruning.get_pruning_hparams().parse(
54 p = pruning.Pruning(self.pruning_hparams)
64 p = pruning.Pruning(spec=self.pruning_hparams, sparsity=self.sparsity)
71 p = pruning.Pruning(spec=self.pruning_hparams, sparsity=self.sparsity)
72 p_copy = pruning.Pruning(
92 masked_weights = pruning.apply_mask(weights,
103 masked_weights = pruning.apply_mask(weights)
105 p = pruning
[all...]

Completed in 203 milliseconds