MichaelJKlaiber commented on issue #12410: URL: https://github.com/apache/tvm/issues/12410#issuecomment-1215634486
Seems the bug is in the lowering step from relay_prim_func to tir_prim_func. When collecting the input arguments, the order gets messed up. By replacing the function `_get_tensors` here https://github.com/MichaelJKlaiber/tvm/blob/issue/uma-tflite/python/tvm/relay/backend/contrib/uma/api/lower.py#L83 by ``` return list(te_cached_func.inputs) + list(te_cached_func.outputs) ``` everything works perfectly fine for all the test case we provided including the tflite import Does anyone know if it is sufficient to rely on the `.inputs` parameters from te_cached_func ? -- 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. To unsubscribe, e-mail: [email protected] For queries about this service, please contact Infrastructure at: [email protected]
