Searched refs:get_thresholds (Results 1 – 3 of 3) sorted by relevance
52 self.assertEqual(len(pruning.get_thresholds()), expected_num_masks)76 self.assertEqual(len(pruning.get_thresholds()), expected_num_masks)
123 def get_thresholds(): function497 thresholds = get_thresholds()566 thresholds = get_thresholds()
32 from tensorflow.contrib.model_pruning.python.pruning import get_thresholds