code: def custom_fn(x):
      return x**2
  
imports: import json
  import numpy
  import pytest
  import keras
  from keras import ops
  from keras import testing
  from keras.saving import serialization_lib
label_desc: Use image node_id 1 for calling a global function with example usage: custom_fn(x) and returns: unknown