code: def get_weight_size_in_bytes(weight):
      if weight.dtype == tf.string:
          return tf.reduce_sum(tf.strings.length(weight, unit="BYTE"))
      else:
          return tf.size(weight) * weight.dtype.size
  
imports: import os
  import shutil
  from absl.testing import parameterized
  import tensorflow
  from tensorflow_model_optimization.python.core.common.keras.compression.algorithms import epr
label_desc: Use image node_id 5 for calling a global function with example usage: get_weight_size_in_bytes(weight) and returns: tf, unknown