message: Be less strict about matching bias apply reduction
  Summary:
  The reduction takes place over any dimensions, so don't bother to
  check the parameters
  Test Plan: None
  Reviewers: georgep
diff: @@ -154,15 +154,10 @@ bool IsBiasReduce(const HloInstruction *inst) {
  if (ShapeUtil::Rank(inst->shape()) != 1) return false;
  - if (ShapeUtil::Rank(inst->operand(0)->shape()) != 4) return false;
  -
  const std::vector<int64> &dims(inst->dimensions());
  if (dims.size() != ShapeUtil::Rank(inst->operand(0)->shape()) - 1) {
  return false;
  }
  - for (int64 d : dims) {
  - if (d == ShapeUtil::Rank(inst->operand(0)->shape()) - 1) return false;
  - }
  return true;
  }
  