code: def _int_square(value):
      """Avoids the TF op `Square(T=...)` for ints as sampling can happen on clients."""
      return (value - 1) * (value + 1) + 1
  
imports: import tensorflow
  import tensorflow_probability
label_desc: Use image node_id 4 for calling a global function with example usage: _int_square(value) and returns: unknown