class_docstr: Registry responsible for built-in keras layers.
code: def _get_rnn_cells(cls, rnn_layer):
      if isinstance(rnn_layer.cell, layers.StackedRNNCells):
          return rnn_layer.cell.cells
      else:
          return [rnn_layer.cell]
  
file_path: model-optimization/tensorflow_model_optimization/python/core/sparsity/keras/prune_registry.py
imports: import tensorflow
  from tensorflow_model_optimization.python.core.sparsity.keras import prunable_layer
label_desc: Use image node_id 2 for calling the PruneRegistry obj's underlying member method code with example usage: obj._get_rnn_cells(cls, rnn_layer) and returns: rnn_layer, list