code: def create_auto_shard_compile_args(config):
      def loss_func(y1, y2):
          return tf.abs(y1[0] - y1[1]) + tf.abs(y2[0] - y2[1])
  
      args = {
          "optimizer": tf.keras.optimizers.SGD(lr=0.0),
          "loss": loss_func,
      }
      return args
  
imports: import os
  from unittest import TestCase
  import numpy
  import pytest
  import tensorflow
  from bigdl.orca.learn.tf2 import Estimator
  from bigdl.orca.ray import OrcaRayContext
  from bigdl.orca import OrcaContext
label_desc: Use image node_id 9 for calling a global function with example usage: create_auto_shard_compile_args(config) and returns: args, unknown