absl::flat_hash_map<int32_t, int32_t> permutation_map;

  for (size_t before_dim_idx = 0, after_dim_idx = 0;
       before_dim_idx < input_shape.size(); ++before_dim_idx) {
    if (input_shape[before_dim_idx] == 1) {
      continue;
    }
    permutation_map.insert({before_dim_idx, after_dim_idx++});
  }

  SmallVector<int32_t> squeezed_permutation;
  DenseElementsAttr input_perm_const;