Searched defs:pruning (Results 1 - 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/ |
H A D | rnn_cells_test.py | 21 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 D | cifar10_train.py | 50 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 D | cifar10_pruning.py | 15 """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 D | pruning_test.py | 15 """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