code: def _mean_squared_error(logits, labels):
      return sum(
          (logit - label) ** 2 for label, logit in zip(labels, logits)
      )
  
imports: import math
  import tensorflow
  from tensorflow_ranking.python import losses
  from tensorflow_ranking.python import losses_impl
label_desc: Use image node_id 9 for calling a global function with example usage: _mean_squared_error(logits, labels) and returns: sum