Just got an official response from NVIDIA regarding a new build of TensorFlow (not sure whether 2.1.0 or 2.1.1) that should fix the API issue. It is scheduled for release this week.
See this post: Tensorflow 2.1.0 for JetPack 4.4 was incorrectly build with V1.x API setting - #6 by AastaLLL