code: def _size(
      input, out_type=np.int32, name=None
  ):  # pylint: disable=redefined-builtin, unused-argument
      return np.asarray(
          onp.prod(ops.convert_to_tensor(input).shape), dtype=out_type
      )
  
imports: import functools
  import numpy
  import numpy
  from tensorflow_probability.python.internal.backend.numpy import _utils
  from tensorflow_probability.python.internal.backend.numpy import ops
  from tensorflow_probability.python.internal.backend.numpy.linalg_impl import einsum
  from tensorflow_probability.python.internal.backend.numpy.linalg_impl import norm
  from tensorflow_probability.python.internal.backend.numpy.linalg_impl import tensordot
label_desc: Use image node_id 16 for calling a global function with example usage: _size(input, out_type, name) and returns: np