tf.distribute.experimental.rpc.Server

Базовый класс сервера для приема RPC для зарегистрированных tf.функций.

Функции могут быть зарегистрированы на сервере и доступны через RPC.

Методы

`create`

@staticmethod
create(
    rpc_layer, address
)

Создать сервер TF RPC по заданному адресу.

Аргументы
`rpc_layer`	Слой связи между клиентом и сервером. В настоящее время поддерживается только слой rpc "grpc".
`address`	Адрес, где размещается сервер RPC.

Возвращаемое значение
Экземпляр класса `tf.distribute.experimental.rpc.Server`.

Исключения
ValueError, если используется rpc_layer, отличная от "grpc". В настоящее время поддерживается только GRPC.

Пример использования:

import portpicker
@tf.function(input_signature=[
     tf.TensorSpec([], tf.int32),
     tf.TensorSpec([], tf.int32)])
def remote_fn(a, b):
  return tf.add(a, b)

port = portpicker.pick_unused_port()
address = "localhost:{}".format(port)
server = tf.distribute.experimental.rpc.Server.create("grpc", address)
server.register("addition", remote_fn)
server.start()

`register`

Просмотреть исходный код

register(
    method_name: str,
    func: Union[def_function.Function, tf_function.ConcreteFunction]
)

Метод для регистрации tf.функции на сервере.

Зарегистрированные методы могут быть вызваны удаленно с клиентов.

Аргументы
`method_name`	Имя tf.функции. Клиенты используют это имя_метода для выполнения RPC.
`func`	`tf.function` или ConcreteFunction для регистрации.

`start`

Просмотреть исходный код

start()

Запускает сервер RPC по указанному адресу.

Сервер прослушивает новые запросы от клиента после запуска.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.9/api_docs/python/tf/distribute/experimental/rpc/Server