code: def test_rpc_ops_non_blocking_convenience_methods(self):
      @eager_def_function.function(
          input_signature=[
              tensor_spec.TensorSpec([], dtypes.int32),
              tensor_spec.TensorSpec([], dtypes.int32),
          ]
      )
      def _remote_fn(a, b):
          return math_ops.multiply(a, b)
  
      port = portpicker.pick_unused_port()
      address = "localhost:{}".format(port)
      server_resource = rpc_ops.GrpcServer(address)
  
      # Register TF function
      server_resource.register("multiply", _remote_fn)
  
      server_resource.start()
      a = variables.Variable(2, dtype=dtypes.int32)
      b = variables.Variable(3, dtype=dtypes.int32)
  
      client = rpc_ops.GrpcClient(address, list_registered_methods=True)
  
      mul_or = client.multiply(a, b)
      self.assertAllEqual(mul_or.is_ok(), True)
      self.assertAllEqual(mul_or.get_value(), 6)
  
      self.assertEqual(
          client.multiply.__doc__,
          "RPC Call for multiply method to server " + address,
      )
  
imports: import threading
  import time
  import numpy
  import portpicker
  from tensorflow.python.distribute.experimental.rpc import rpc_ops
  from tensorflow.python.eager import context
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import config
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import errors
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor_spec
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import data_flow_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import resource_variable_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
  from tensorflow.python.util import nest
label_desc: Use image node_id 5 for calling the RpcOpsTest obj's underlying member method code with example usage: obj.test_rpc_ops_non_blocking_convenience_methods() and returns: math_ops