return op.emitOpError(llvm::formatv(
               "expect 'input' num_elements % {0} == 0, got input type ", z_in))
           << input_type;
  }

  // TODO(jpienaar): Include more shape verification for SHUFFLED4x16INT8
  // format.
  if (op.getWeightsFormat() == "DEFAULT") {
    ShapedType output_type =
        (*op.getOutput().begin()).getType().cast<ShapedType>();
    if (!output_type.hasStaticShape()) {