def __init__(self):
    self.arg_shape_dict = None
    self.out_shape_dict = None
    self.aux_shape_dict = None
    self.sym = None
def __init__(self):
    self.arg_shape_dict = None
    self.out_shape_dict = None
    self.aux_shape_dict = None
    self.sym = None