}

  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();
  }

             << "num cores per replica unavailable";
    }

    const int num_operands_expected = packed_input ? 1 : num_cores_per_replica;
    if (num_cores_per_replica_attr && num_operands_expected != inputs.size()) {
      return partitioned_input->emitOpError()
             << "expects " << num_operands_expected << " operands but found "
             << partitioned_input.getNumOperands();
    }