code: def image_shape(batch_size):
      if data_format() == "channels_first":
          return [batch_size, 3, 224, 224]
      return [batch_size, 224, 224, 3]
  
file_path: tensorflow-upstream/tensorflow/python/eager/benchmarks/resnet50/resnet50_graph_test.py
imports: import time
  import numpy
  import tensorflow.compat.v1
  from tensorflow.python.eager.benchmarks.resnet50 import resnet50
label_desc: Use image node_id 2 for calling a global function with example usage: image_shape(batch_size) and returns: list, list