code: def _apply_pruning(prunable_object):
      """Calculates the masks and updates weights of layers of a wrapped model."""
      assert tf.executing_eagerly()
      for layer in pruning_wrapper.collect_prunable_layers(
          prunable_object
      ):
          layer.pruning_obj.conditional_mask_update()  # Create mask
          layer.pruning_obj.weight_mask_op()
  
file_path: model-optimization/tensorflow_model_optimization/python/core/sparsity/keras/tools/sparsity_tooling.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
  from tensorflow_model_optimization.python.core.keras import metrics
  from tensorflow_model_optimization.python.core.sparsity.keras import prune
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_schedule
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_wrapper
label_desc: Use image node_id 1 for calling a global function with example usage: _apply_pruning(prunable_object) without return types