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)