99a5391722745625de515ff84b2536a7d0947618,tests/test_convert.py,TestConvert,test_sub_convert,#TestConvert#,414

Before Change

        tf.reset_default_graph()

        config = tfe.LocalConfig([
            "server0",
            "server1",
            "crypto_producer",
            "prediction_client",
            "weights_provider"
        ])

        with tfe.protocol.Pond(*config.get_players("server0, server1, crypto_producer")) as prot:
            prot.clear_initializers()

            class PredictionClient(tfe.io.InputProvider):
                def provide_input(self):
                    return tf.constant(np.ones(input_shape))

            input = PredictionClient(config.get_player("prediction_client"))

            converter = Converter(config, prot, config.get_player("weights_provider"))

            x = converter.convert(graph_def, input, register())

            with config.session() as sess:
                tfe.run(sess, prot.initializer, tag="init")

                output = x.reveal().eval(sess, tag="reveal")

After Change

            input = PredictionClient("input-provider")

            converter = Converter(tfe.get_config(), prot, "model-provider")

            x = converter.convert(graph_def, input, register())

            with tfe.Session() as sess:
                sess.run(prot.initializer, tag="init")

                output = sess.run(x.reveal(), tag="reveal")

Instances