def register_post_forward_operation(self, op):
    key = str(len(self.post_ops))
    self.post_ops[key] = op
    return key
def register_post_forward_operation(self, op):
    key = str(len(self.post_ops))
    self.post_ops[key] = op
    return key