code: def _segment_mm(A, B, out, seglen_A, b_trans=False):
      """Invoke the C API of segment_mm."""
      _CAPI_DGLKernelSEGMENTMM(
          to_dgl_nd(A),
          to_dgl_nd(B),
          to_dgl_nd_for_write(out),
          to_dgl_nd(seglen_A),
          False,
          b_trans,
      )
      return out
  
imports: from __future__ import absolute_import
  from .None import backend, ndarray
  from ._ffi.function import _init_api
  from .base import DGLError
label_desc: Use image node_id 9 for calling a global function with example usage: _segment_mm(A, B, out, seglen_A, b_trans) and returns: out