Works and tested on TensorFlow versions 2.3.x and 2.4.x. TensorFlow version 2.6.x doesn't work. Add precisions in the doc about that.