f54c2b1361fb86f55a36064158c6baa658ffffb9,examples/mnist/run.py,,,#,162

Before Change

                                    logits,
                                    prediction_client.receive_output)

  with tfe.Session(target=session_target) as sess:

    sess.run(tf.global_variables_initializer(), tag="init")

    print("Training")
    sess.run(cache_updater, tag="training")

    for _ in range(5):
      print("Predicting")
      sess.run(prediction_op, tag="prediction")

After Change

    // get prediction input from client
    x = tfe.define_private_input(prediction_client.player_name,
                                 prediction_client.provide_input)
    logits = model(x)

  // send prediction output back to client
  prediction_op = tfe.define_output(prediction_client.player_name,
                                    logits,

Instances