* Enhance bipartite_match_op to support argmax matching after bipartite matching.

* Fix typo error.