auto perm_values = perm_values_attr.getValues<APInt>();
  size_t idx = 0;
  for (; idx < perm_values_attr.size() - 2; ++idx) {
    if (perm_values[idx].getSExtValue() != idx) return false;
  }

  return (perm_values[idx].getSExtValue() == perm_values_attr.size() - 1) &&
         (perm_values[idx + 1].getSExtValue() == idx);
}

  // Transpose and constant-fold the weight if needed.
  if (adj_y.getValue()) {
    SmallVector<int32_t> perm_values(num_batch_dim);
    absl::c_iota(perm_values, 0);
    perm_values.push_back(num_batch_dim + 1);
    perm_values.push_back(num_batch_dim);
    Value perm = Create1DConstValue<int32_t>(builder, loc, perm_values);
    auto transpose_op = builder.create<TF::TransposeOp>(loc, weight, perm);

      return failure();
    }

    auto perm1_value = llvm::dyn_cast<DenseElementsAttr>(perm1.getValue())
                           .getValues<int32_t>();
    auto perm2_value = llvm::dyn_cast<DenseElementsAttr>(perm2.getValue())
                           .getValues<int32_t>();

    llvm::SmallVector<int32_t> perm1_arr(perm1_value.begin(),
                                         perm1_value.end());