}
  return type;
}

// Changes the axis of the input per-channel quantized type to match the
// dimension of the target type. Returns nullptr if it fails.
quant::UniformQuantizedPerAxisType ResetAxisAndBroadcast(
    const ArrayRef<int64_t> shape,
    const quant::UniformQuantizedPerAxisType qtype, const Type target,
    const int quant_dim) {
  const auto shaped = dyn_cast<RankedTensorType>(target);