inadob opened a new pull request #4704: [Relay][Frontend][TFLite] Add parser support for arg_min_max URL: https://github.com/apache/incubator-tvm/pull/4704 * this implementation supports only the case when the axis is a scalar * tflite 1.13 removes all dims of size 1, Relay doesn't do this * WARNING: every newer version of tflite > 1.13 needs keepdims=TRUE in the corresponding Relay operation. If there is a way to check the tflite version that was used to convert the model passed to the parser, I will be happy to implement that.
---------------------------------------------------------------- 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