Add some special handling for MR shuffle handler network output