def pack(a, b):
    a = _bitcast_convert_type(a, word_dtype)
    b = _bitcast_convert_type(b, word_dtype)
    a = _convert_element_type(a, new_dtype=double_word_dtype)
    b = _convert_element_type(b, new_dtype=double_word_dtype)
    a = tf.bitwise.left_shift(a, const(double_word_dtype, nbits))
    return tf.bitwise.bitwise_or(a, b)
def pack(a, b):
    a = _bitcast_convert_type(a, word_dtype)
    b = _bitcast_convert_type(b, word_dtype)
    a = _convert_element_type(a, new_dtype=double_word_dtype)
    b = _convert_element_type(b, new_dtype=double_word_dtype)
    a = tf.bitwise.left_shift(a, const(double_word_dtype, nbits))
    return tf.bitwise.bitwise_or(a, b)