code: def _PruneCSRMatrix(unpruned, pruned_pattern):
      """TODO(tabakg): Consider re-writing in C++."""
      _, dtype = sparse_csr_matrix_ops.dense_shape_and_type(
          pruned_pattern
      )
      coo_unpruned = sparse_tensor.SparseTensor(
          *sparse_csr_matrix_ops.csr_sparse_matrix_to_sparse_tensor(
              unpruned, type=dtype
          )
      )
      coo_pruned_pattern = sparse_tensor.SparseTensor(
          *sparse_csr_matrix_ops.csr_sparse_matrix_to_sparse_tensor(
              pruned_pattern, type=dtype
          )
      )
      return sparse_csr_matrix_ops.sparse_tensor_to_csr_sparse_matrix(
          *_PruneSparseTensor(coo_unpruned, coo_pruned_pattern)
      )
  
file_path: tensorflow/tensorflow/python/ops/linalg/sparse/sparse_csr_matrix_grad.py
imports: from tensorflow.python.framework import ops
  from tensorflow.python.framework import sparse_tensor
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import array_ops_stack
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import sparse_ops
  from tensorflow.python.ops.linalg.sparse import sparse_csr_matrix_ops
label_desc: Use image node_id 6 for calling a global function with example usage: _PruneCSRMatrix(unpruned, pruned_pattern) and returns: sparse_csr_matrix_ops