code: def square(x):
      if (
          isfloat_tensor(x)
          and isccontiguous(x)
          and x.size > omp_threshold
      ):
          y = np.empty_like(x)
          _x = ffi.cast("float *", ffi.from_buffer(x))
          _y = ffi.cast("float *", ffi.from_buffer(y))
          NervanaObject.be.mathlib.cmath_square(_x, x.size, _y)
          return y
      else:
          return np.square(x)
  
imports: import sys
  import numpy
  from cffi import FFI
  from neon import NervanaObject
  from neon.transforms import Rectlinclip
label_desc: Use image node_id 12 for calling a global function with example usage: square(x) and returns: y, np