def create_client_graph():
inputs = [Input((1, ), "int32")]
x = inputs[0]
x = TcpServerSubgraph(
addr=("localhost", 5678),
graph=create_server_graph(),
)(x)
outputs = [x]
return Model(inputs=inputs, outputs=outputs)
async def _receive_model(self):
line = await self._reader.readline()
self._graph = Model.deserialize(line.decode())
def create_server_graph():
inputs = [Input((1, ), "int")]
outputs = inputs
return Model(inputs=inputs, outputs=outputs)
def create_client_graph():
inputs = [Input((1, ), "int")]
x = inputs[0]
x = IntraprocessServer(graph=create_server_graph())(x)
outputs = [x]
return Model(inputs=inputs, outputs=outputs)
def create_graph():
inputs = [Input((1, ), "int")]
x = inputs[0]
x = Stringify((1, ), "str")(x)
outputs = [x]
return Model(inputs=inputs, outputs=outputs)
def create_server_graph():
inputs = [Input((1, ), "int32")]
outputs = [Square(shape=(1, ), dtype="int32")(inputs[0])]
return Model(inputs=inputs, outputs=outputs)
def create_client_graph():
inputs = [Input((1, ), "int32")]
x = inputs[0]
x = IntraprocessStreamingLoopbackServer((1, ), "int32")(x)
outputs = [x]
return Model(inputs=inputs, outputs=outputs)