code: def _loss_fn(labels, logits, weights=1.0):
      return tf.reduce_mean(
          input_tensor=tf.compat.v1.losses.softmax_cross_entropy(
              onehot_labels=labels, logits=logits, weights=weights
          )
      )
  
file_path: model_search/model_search/meta/distillation_test.py
imports: from absl.testing import parameterized
  from model_search.meta import distillation
  from model_search.proto import distillation_spec_pb2
  import numpy
  import tensorflow.compat.v2
label_desc: Use image node_id 2 for calling a global function with example usage: _loss_fn(labels, logits, weights) and returns: tf