code: def forward(self, x, offsets):
      return self.linear(self.emb(x, offsets))
  
file_path: H2O/h2o_flexgen/benchmark/third_party/DeepSpeed/tests/unit/runtime/sparse_tensor/test_sparse_grads.py
imports: import torch
  import deepspeed
  from unit.common import DistributedTest
  import deepspeed.utils.groups
label_desc: Use image node_id 2 for calling the Model obj's underlying member method code with example usage: obj.forward(x, offsets) and returns: self