def op_bit_sizes(op): sizes = None if not type_has_size(op.output_type): sizes = set(type_sizes(op.output_type)) for input_type in op.input_types: if not type_has_size(input_type): if sizes is None: sizes = set(type_sizes(input_type)) else: sizes = sizes.intersection(set(type_sizes(input_type))) return sorted(list(sizes)) if sizes is not None else None
def op_bit_sizes(op): sizes = None if not type_has_size(op.output_type): sizes = set(type_sizes(op.output_type)) for input_type in op.input_types: if not type_has_size(input_type): if sizes is None: sizes = set(type_sizes(input_type)) else: sizes = sizes.intersection(set(type_sizes(input_type))) return sorted(list(sizes)) if sizes is not None else None
def type_add_size(type_, size): if type_has_size(type_): return type_ return type_ + str(size)
def type_add_size(type_, size): if type_has_size(type_): return type_ return type_ + str(size)