Modify distributed_training_utils.py import for TF 2.4 (awslabs#422)

Author: NihalHarish

smdebug/tensorflow/utils.py

@@ -427,6 +427,10 @@ def is_tf_version_greater_than_2_4_x():
     return version.parse("2.4.0") <= TF_VERSION
 
 
+def is_tf_version_greater_than_2_4_x():
+    return version.parse("2.4.0") <= version.parse(tf.__version__)
+
+
 def is_profiler_supported_for_tf_version():
     # Profiler Support Added For TF Versions 2.2.0 And Greater
     return version.parse("2.2.0") <= TF_VERSION