def METHOD_NAME(data_shape):
    valid_counts_shape = output_tensor((1,), "int64")
    out_tensor_shape = output_tensor((data_shape.shape[0],), "int64")
    out_indices_shape = output_tensor((2,), "int64")
    valid_counts_shape[0] = data_shape[0]
    for i in const_range(data_shape.shape[0]):
        out_tensor_shape[i] = data_shape[i]
    out_indices_shape[0] = data_shape[0]
    out_indices_shape[1] = data_shape[1]
    return valid_counts_shape, out_tensor_shape, out_indices_shape