def register_pre_forward_operation(self, op):
    key = str(len(self.pre_ops))
    self.pre_ops[key] = op
    return key
def register_pre_forward_operation(self, op):
    key = str(len(self.pre_ops))
    self.pre_ops[key] = op
    return key