code: def test_dc_hybridblock_graph_partition():
      class MyBlock(mx.gluon.HybridBlock):
          def __init__(self):
              super().__init__()
              self.dense = mx.gluon.nn.Dense(units=4)
  
          def forward(self, x, idx):
              mask = mx.nd.np._internal.boolean_mask(self.dense(x), idx)
              return mx.np.sum(mask)
  
      def setup(*, nd):
          x = mx.np.array([[0, 1], [2, 3], [4, 5], [6, 7]])
          idx = mx.np.array([1, 1, 1, 1])
          return [x, idx]
  
      net = MyBlock()
      net.initialize()
      _assert_dc_gluon(setup, net, numpy=True, autograd=False)
  
imports: import functools
  import operator
  import numpy
  import mxnet
  import mxnet._deferred_compute
  from mxnet.base import MXNetError
  from mxnet.util import TemporaryDirectory
  import pytest
label_desc: Use image node_id 36 for calling a global function with example usage: test_dc_hybridblock_graph_partition() and returns: list, mx