code: def get_hw(features, dtype=tf.int32):
      """Return the height, width of image as float32 tf.Tensors."""
      if isinstance(features, dict):
          sz = tf.shape(features["inputs"])
      elif isinstance(features, tf.Tensor):
          sz = tf.shape(features)
      else:
          raise ValueError(f"Unknown type of object: {features}")
  
      h = tf.cast(sz[0], dtype=dtype)
      w = tf.cast(sz[1], dtype=dtype)
      return h, w
  
imports: from typing import Any, Dict
  import tensorflow
label_desc: Use image node_id 3 for calling a global function with example usage: get_hw(features, dtype) and returns: h,  w