code: def disable_auto_shard(options):
      if version.parse(tf.__version__) < version.parse("2.1.0"):
          options.experimental_distribute.auto_shard = False
      else:
          options.experimental_distribute.auto_shard_policy = (
              tf.data.experimental.AutoShardPolicy.OFF
          )
  
imports: import tensorflow
  from packaging import version
label_desc: Use image node_id 2 for calling a global function with example usage: disable_auto_shard(options) without return types