// The unpack & pack should have the same axis & num inputs/outputs.
    if (pack_op.getAxis() != input_unpack_op.getAxis() ||
        pack_op.getValuesCount() != input_unpack_op.getNum())
      return failure();

    const int total_pack_inputs = pack_op.getNumOperands();
    const int num_results = input_unpack_op.getNumResults();
    if (total_pack_inputs != num_results) return failure();