On my host machine I am using Keras with Tensorflow as the backend to develop neural networks. On the host machine I am saving the network as a .uff for use on the Jetson TX2. I’ve modified the SampleUffMNIST.cpp code to load my own .uff file “foo.uff” but when I execute the code on the TX2 I get the following error:
ERROR: UFFParser: Validator error: flatten_1/Reshape: Unsupported operation Flatten
Which seems weird to me because according to the TensorRT documentation, Flatten is a supported layer.
Can anyone provide any insight into why I’m getting this error?
I installed TensorRT on the TX2 via the Jetpack 3.2 developer’s preview so it should support Tensorflow models.