[GitHub] [incubator-tvm] apivovarov opened a new pull request #4320: Fix TFLite RESHAPE assert

[GitBox <gi...@apache.org>]

apivovarov opened a new pull request #4320: Fix TFLite RESHAPE assert
URL: https://github.com/apache/incubator-tvm/pull/4320
 
 
   Recently I found that Reshape op input might have one or two input tensors.
   in TVM code we assert that length is two but we only use `input_tensor[0]`.
   Looks like the second input tensor was reshape shape in the past. But current TFLite schema which we use r.1.13 has ReshapeOptions which contains `new_shape`:
   ```
   table ReshapeOptions {
     new_shape:[int];
   }
   ```
   We do not use the second input tensor in TFLite frontend.
   
   What is more is that I have custom tflite model which has only one input tensor for RESHAPE op.
   Current assert for input tensors length makes the compilation fail.
   I suggest what we allow input tensors length for RESHAPE operator to be 1 or 2.
   
   
   

----------------------------------------------------------------
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:
users@infra.apache.org


With regards,
Apache Git Services