code: def define_trial_run_flags():
      flags.DEFINE_string(
          name="resource", default="", help="resource specification"
      )
      flags.DEFINE_integer(
          name="trial_steps",
          default=100,
          help="number of steps for trial",
      )
      flags.DEFINE_enum(
          name="optimizer",
          default="adam",
          enum_values=["adam", "lazyadam", "sgd"],
          case_sensitive=False,
          help=flags_core.help_wrap("optimizer to use."),
      )
      flags.DEFINE_bool(
          name="dense_gradient",
          default="True",
          help="whether to use dense gradient",
      )
      flags.DEFINE_string(
          name="simulation_strategy_id",
          default=None,
          help="strategy id to simulate",
      )
      flags.DEFINE_string(
          name="simulation_folder",
          default=None,
          help="folder to store simulation result",
      )
      flags.DEFINE_string(
          name="autodist_strategy",
          default="PS",
          help="the autodist strategy",
      )
      flags.DEFINE_boolean(
          name="autodist_patch_tf",
          default=True,
          help="AUTODIST_PATCH_TF",
      )
      flags.DEFINE_boolean(name="proxy", default=True, help="proxy")
      flags.DEFINE_string(
          name="default_data_dir",
          default="~/movielens",
          help="the default data directory",
      )
      flags.DEFINE_integer(
          name="num_epochs", default=3, help="number of training epochs"
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import json
  import yaml
  import os
  import sys
  import tensorflow
  from absl import app
  from absl import flags
  from absl import logging
  from os.path import expanduser
  from tensorflow_addons.optimizers import LazyAdam
  from utils.recommendation import constants
  from utils.recommendation import movielens
  from utils.recommendation import ncf_common
  from utils.recommendation import ncf_input_pipeline
  from utils.recommendation import neumf_model
  from utils.flags import core
  from utils.logs import logger
  from utils.logs import mlperf_helper
  from utils.misc import keras_utils
  from utils.misc import model_helpers
  from autodist import AutoDist
  from autodist.strategy.ps_strategy import PS
  from autodist.strategy.ps_lb_strategy import PSLoadBalancing
  from autodist.strategy.partitioned_ps_strategy import PartitionedPS
  from autodist.strategy.all_reduce_strategy import AllReduce
  from autodist.strategy.parallax_strategy import Parallax
label_desc: Use image node_id 5 for calling a global function with example usage: define_trial_run_flags() without return types