code: def get_flops():
      run_meta = tf.RunMetadata()
      opts = tf.profiler.ProfileOptionBuilder.float_operation()
  
      # We use the Keras session graph in the call to the profiler.
      flops = tf.profiler.profile(
          graph=keras.backend.get_session().graph,
          run_meta=run_meta,
          cmd="op",
          options=opts,
      )
  
      return flops.total_float_ops
  
file_path: auptimizer/Examples/early_stopping/mnist_keras/mnist.py
imports: import aup
  import tensorflow
  import numpy
  from tensorflow import keras
  from tensorflow.keras import layers
  from math import log
  import sys
label_desc: Use image node_id 1 for calling a global function with example usage: get_flops() and returns: flops