Searched refs:get_masked_weights (Results 1 – 3 of 3) sorted by relevance
51 self.assertEqual(len(pruning.get_masked_weights()), expected_num_masks)75 self.assertEqual(len(pruning.get_masked_weights()), expected_num_masks)
29 from tensorflow.contrib.model_pruning.python.pruning import get_masked_weights
115 def get_masked_weights(): function