SmallVector<T> Permute(const ArrayRef<T> values,
                       const ArrayRef<int64_t> permutation) {
  SmallVector<T> permuted_values(/*Size=*/values.size(), /*Value=*/T{});
  for (auto [i, permutation_idx] : llvm::enumerate(permutation)) {
    permuted_values[i] = std::move(values[permutation_idx]);
  }
  return permuted_values;
}

}  // namespace mlir::quant

  auto permute_attr = DenseIntElementsAttr::get(permute_type, permutation);
  auto permute_value =
      builder->create<TF::ConstOp>(input.getLoc(), permute_attr);
  return builder->create<TF::TransposeOp>(input.getLoc(), input, permute_value);
}

void HandleConv2DFilter(TF::Conv2DOp conv2d, int64_t block_size) {
  // For example, if filter shape is [7, 7, 3, 64] with block_size 2,