mbarrett97 commented on a change in pull request #4543: [FRONTEND][TFLITE] Add support for TFLite_Detection_PostProcess URL: https://github.com/apache/incubator-tvm/pull/4543#discussion_r366410943
########## File path: tests/python/frontend/tflite/test_forward.py ########## @@ -1113,6 +1113,49 @@ def test_forward_fully_connected(): _test_fully_connected([5, 1, 1, 150], [150, 100], [100]) +####################################################################### +# Custom Operators +# ------- + +def test_detection_postprocess(): + tf_model_file = tf_testing.get_workload_official( + "http://download.tensorflow.org/models/object_detection/" + "ssd_mobilenet_v2_quantized_300x300_coco_2019_01_03.tar.gz", Review comment: The test looks non-trivial to add because quite a small difference in the convolutional part of the network can result in significant changes to the ordering of the output tensor (eg. we might see at different detection at the cut off threshold). I'm not sure what the best way is to proceed, do you have any thoughts? ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services