tensorplay.ao.pruning.validate_pruning_amount

tensorplay.ao.pruning.validate_pruning_amount(amount: int, tensor_size: int) → None[source]

Check that an absolute pruning count fits the data being pruned.

Parameters:

• amount – absolute number of units to prune.

• tensor_size – number of units available for pruning.

Raises:

ValueError – if amount exceeds tensor_size.