def gen_conv(lhs, rhs, preferred_element_type: Optional[DType]):
    out = tfxla.conv(lhs, rhs, window_strides, padding, lhs_dilation, rhs_dilation, dnums_proto, feature_group_count=feature_group_count, precision_config=precision_config_proto, preferred_element_type=preferred_element_type)
    out.set_shape(out_tf_shape)
    return out
def gen_conv(lhs, rhs, preferred_element_type: Optional[DType]):
    out = tfxla.conv(lhs, rhs, window_strides, padding, lhs_dilation, rhs_dilation, dnums_proto, feature_group_count=feature_group_count, precision_config=precision_config_proto, preferred_element_type=preferred_element_type)
    out.set_shape(out_tf_shape)
    return out