Hi,
Sorry for keeping you waiting.
NVMLibrary, which used for managing GPU device, doesn’t support tx1 and this make it not easy to set GPU configure.
Another way is to release visionworks context before calling tensorflow.
vx_context can be released by vxReleaseContext API.
Could you give it a try?