message: Remove else branch
diff: @@ -1479,13 +1479,10 @@ Status UniqueIdxShapeFn(InferenceContext* c) {
  }
  c->set_output(1, c->Vector(c->Dim(input, axis)));
  return Status::OK();
  - } else {
  + }
  return errors::InvalidArgument(
  "axis does not support input tensors larger than 1 elements.");
  }
  -
  - return Status::OK();
  -}
  } // namespace
  REGISTER_OP("Unique")
  