code: def euc_sq_distance(x, y, eval_mode=False):
      """Computes Euclidean squared distance.
  
      Args:
        x: Tensor of size B1 x d
        y: Tensor of size B2 x d
        eval_mode: boolean indicating whether to compute all pairwise distances or
          not. If eval_mode=False, must have B1=B2.
  
      Returns:
        Tensor of size B1 x B2 if eval_mode=True, otherwise Tensor of size B1 x 1.
      """
      x2 = tf.math.reduce_sum(x * x, axis=-1, keepdims=True)
      y2 = tf.math.reduce_sum(y * y, axis=-1, keepdims=True)
      if eval_mode:
          y2 = tf.transpose(y2)
          xy = tf.linalg.matmul(x, y, transpose_b=True)
      else:
          xy = tf.math.reduce_sum(x * y, axis=-1, keepdims=True)
      return x2 + y2 - 2 * xy
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import tensorflow
label_desc: Use image node_id 1 for calling a global function with example usage: euc_sq_distance(x, y, eval_mode) and returns: unknown