Searched refs:get_masks (Results 1 – 3 of 3) sorted by relevance
50 self.assertEqual(len(pruning.get_masks()), expected_num_masks)55 for mask in pruning.get_masks():74 self.assertEqual(len(pruning.get_masks()), expected_num_masks)79 for mask in pruning.get_masks():
119 def get_masks(): function140 masks = get_masks()495 masks = get_masks()565 masks = get_masks()
30 from tensorflow.contrib.model_pruning.python.pruning import get_masks