def METHOD_NAME(self):
    op_types = ["Linear"]
    tg = GraphTrace()
    # absorb_to_layer={'absorb_layer': absorbed_layer}
    absorb_to_layer, no_absorb_layers = tg.get_absorb_to_layer(self.model, self.example_inputs, op_types)
    self.assertTrue(len(no_absorb_layers) == 1)
    absorb_to_layer, no_absorb_layers = tg.get_absorb_to_layer(self.gptj, self.lm_input, op_types)
    self.assertTrue(len(no_absorb_layers) == 11)
    return absorb_to_layer, no_absorb_layers