code: def get_uncompiled_model2(dataset_size=32, output_size=4):
      inputs = tf.keras.Input(shape=(dataset_size,), name="inputs")
      x = tf.keras.layers.Dense(16, activation="relu", name="dense_1")(
          inputs
      )
      x = tf.keras.layers.BatchNormalization()(x)
      x = tf.keras.layers.Dense(32, activation="relu", name="dense_2")(
          x
      )
      outputs = tf.keras.layers.Dense(
          output_size, activation="sigmoid", name="predictions"
      )(x)
      model = tf.keras.Model(inputs=inputs, outputs=outputs)
      return model
  
file_path: tinygrad/examples/compile_tensorflow.py
imports: import os, sys
  import numpy
  import subprocess
  import tensorflow
  import tf2onnx
  from extra.onnx import get_run_onnx
  from tinygrad.tensor import Tensor
  from extra.export_model import export_model_clang, compile_net, jit_model
label_desc: Use image node_id 1 for calling a global function with example usage: get_uncompiled_model2(dataset_size, output_size) and returns: model