data.w_in = op_type.getShape()[3];

  std::vector<int32_t> kernel_size;
  GetI32VectorFromDenseI64CompositeAttr(composite_attrs, "kernel_size",
                                        &kernel_size);
  data.kh = kernel_size[0];
  data.kw = kernel_size[1];

  std::vector<int32_t> padding;
  GetI32VectorFromDenseI64CompositeAttr(composite_attrs, "padding", &padding);
  data.ph = padding[0];
  data.pw = padding[1];

  std::transform(range.begin(), range.end(), ret.begin(),
                 [](int64_t attr) { return static_cast<int32_t>(attr); });
  *out_vec = std::move(ret);
  return true;
}

bool GetI32VectorFromDenseI64CompositeAttr(
    const DictionaryAttr& composite_attrs, const std::string& attr_name,
    std::vector<int32_t>* out_vec) {
  DenseIntElementsAttr attr;
  if (!EnsureAttribute<DenseIntElementsAttr>(composite_attrs, attr_name,

// with the name attr_name. If so, extracts its values and stores as a vector
// of int32_t elements.
// Note: This assumes the DenseIntElementsAttr has its values stored as int64_t.
bool GetI32VectorFromDenseI64CompositeAttr(
    const DictionaryAttr& composite_attrs, const std::string& attr_name,
    std::vector<int32_t>* out_vec);

// Get a DenseIntElementsAttr of type I64 and convert it to an I32 attribute.