Searched refs:_check_removal_of_pruning_vars (Results 1 – 1 of 1) sorted by relevance
170 def _check_removal_of_pruning_vars(self, number_masked_layers): member in StripPruningVarsTest191 self._check_removal_of_pruning_vars(number_masked_conv_layers)208 self._check_removal_of_pruning_vars(number_masked_fc_layers)225 self._check_removal_of_pruning_vars(number_masked_lstm_layers)