code: def _SparseMatrixSoftmaxGrad(op: ops.Operation, grad_softmax):
      """Gradient for sparse_matrix_softmax op."""
      softmax = op.outputs[0]
      return sparse_csr_matrix_ops.sparse_matrix_softmax_grad(
          softmax, grad_softmax, type=op.get_attr("type")
      )
  
file_path: tensorflow-upstream/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 10 for calling a global function with example usage: _SparseMatrixSoftmaxGrad(op, grad_softmax) and returns: sparse_csr_matrix_ops