Gather operator dimension check error about armnn

[morgolock]

Hi @zxros10

Could you please share the model ? Is it an .onnx file?

[zxros10]

The attach is the test model I construct for this case. you can download and change filename .jpg to .onnx

[MikeJKelly]

Hi @zxros10

That Gather operator is actually a GatherND (it has both axis and batch_dims) but the Onnx Parser doesn't support GatherND so it's interpreting it as a Gather operator and so it's calculating the Output shape incorrectly. I'll see if we can add support for GatherND to the Onnx Parser.

Best regards,
Mike

[zxros10]

Thanks
I think this is not only problem of armnn onnx parser. I convert this onnx model to tflite, and run in armnn gpu, and has this error.

[zxros10]

I have to resolve this problem in my project. The input model is tflite, and execute ExecuteNetwork with parameter --tflite-executor parser.
Please give me some suggestion for modify the armnn code. Thanks

[tracyn-arm]

Hi. The link for gather_dim_test above is broken for me. Can you please repost this file and provide the exact ExecuteNetwork line you are using? Thanks.

[zxros10]

gather_dim_test.zip
This is the test model

[tracyn-arm]

I ran the test with a very current version of Arm NN: ./ExecuteNetwork --compute GpuAcc --model-path ./gather_dim_test.onnx --tflite-executor parser
I am getting the following error: Fatal: Unsupported operation Slice for node 'Slice_13' at function LoadGraph

This thread has been about Gather so wondering if Slice is now the issue you are asking about? Thanks.

[tracyn-arm]

Hi. Can you please close this issue and re-open another one with the right .tflite file and command line to reproduce? The zip file provided 5 days ago had only the .onnx model which has additional issues in the parsing. Thanks.

[zxros10]

Ok, I will re-open another one and provide the models