apivovarov edited a comment on issue #4298: [TFLite] Support PRelu URL: https://github.com/apache/incubator-tvm/pull/4298#issuecomment-553159880 PRELU docs says that H and W can be "shared". That is what we see in my custom tflite model. PRELU Alpha tensor with index 4 has shape [1,1,16]. It means that H and W are 1 - H and W are shared, alpha parameters only exist for color dimension. Also, Alpha tensor is 3D, not 4D. It is because Batch dimension is probably always shared and simply omitted in Alpha tensor shape. If TVM Relay needs PRELU Alpha tensor dimension size to be the same as main input tensor dimension size then we should add 1 to Alpha tensor shape for batch dimension. `[1,1,16] -> [1,1,1,16]`
---------------------------------------------------------------- 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: [email protected] With regards, Apache Git Services
