code: def argmin_grad(x, y):
      abs_diff = tf.abs(tf.subtract(x, y))
      argmin = tf.cast(tf.argmin(abs_diff), tf.float32)
  
      def grad(dy):
          """Let gradients pass through."""
          return dy, None
  
      return argmin, grad
  
imports: import tensorflow.compat.v1
label_desc: Use image node_id 1 for calling a global function with example usage: argmin_grad(x, y) and returns: argmin,  grad, dy,  None