def METHOD_NAME(
    computation_proto: pb.Computation, arg: Optional[object] = None
) -> object:
  """Runs a TensorFlow computation with argument `arg`.
  Args:
    computation_proto: An instance of `pb.Computation`.
    arg: The argument to invoke the computation with, or None if the computation
      does not specify a parameter type and does not expects one.
  Returns:
    The result of the computation.
  """
  with tf.Graph().as_default() as graph:
    type_signature = type_serialization.deserialize_type(computation_proto.type)
    if type_signature.parameter is not None:  # pytype: disable=attribute-error
      stamped_arg = _stamp_value_into_graph(
          arg,
          type_signature.parameter,  # pytype: disable=attribute-error
          graph,
      )
    else:
      stamped_arg = None
    init_op, result = tensorflow_utils.deserialize_and_call_tf_computation(
        computation_proto, stamped_arg, graph, '', tf.constant('bogus_token')
    )
  with tf.compat.v1.Session(graph=graph) as sess:
    if init_op:
      sess.run(init_op)
    result = tensorflow_utils.fetch_value_in_session(sess, result)
  return result