}

  const bool packed_input = first_partitioned_input.getIsPacked();
  const size_t num_operands_expected = packed_input ? 1 : num_cores_per_replica;
  if (metadata &&
      num_operands_expected != first_partitioned_input.getNumOperands()) {
    return first_partitioned_input->emitOpError()
           << "expects " << num_operands_expected << " operands but found "
           << first_partitioned_input.getNumOperands();
  }