return no_op();

    int64_t input_rank = input_type.getRank();
    int64_t first_dim_size = input_type.getDimSize(0);
    if (ShapedType::isDynamic(first_dim_size)) return failure();

    if (first_dim_size <= 1)
      // No shuffling is required, so copy input directly to output.
      return no_op();