srkreddy1238 commented on code in PR #13408:
URL: https://github.com/apache/tvm/pull/13408#discussion_r1024208179
##########
python/tvm/relay/frontend/tflite.py:
##########
@@ -1549,7 +1549,7 @@ def convert_gather(self, op):
assert axis < data_dim, "Axis out of bounds"
if self.has_expr(indices.tensor_idx):
- indices_expr = self.get_expr(indices.tensor_idx)
+ indices_expr = _op.cast(self.get_expr(indices.tensor_idx), "int32")
Review Comment:
That was a bug I encountered while importing other tflite models.
Those models not added here as they are time consuming, but planning to add
them to benchmarking scripts.
--
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: commits-unsubscr...@tvm.apache.org
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
