YOLOV8 image segmentation postprocessing with NMS for android TFLite model

Question

After training custom data in YOLOV8 Image segmentation it gives output float32[1,37,8400] and float32[1,160,160,32] where one is prediction and another is detection image edges. But in android there have no way to perform non maximum suppression to get single value for the detected image with confidence. Is there any way to stich NMS with ONNX model when converting YOLOV8 model to ONNX model(If able to add NMS in ONNX then may be can convert it in TFLite). If we can't do that with YOLOV8 then what is the best way to train custom dataset for image segmentation which will support in mobile devices.