7272af90d4f5916f61f8bc5660e5605e730bbd78,examples/securenn/network_a.py,,,#,177

Before Change

  // compute prediction
  w0, b0, w1, b1, w2, b2 = params
  layer0 = x
  layer1 = tfe.relu((tfe.matmul(layer0, w0) + b0))
  layer2 = tfe.relu((tfe.matmul(layer1, w1) + b1))
  logits = tfe.matmul(layer2, w2) + b2

  // send prediction output back to client
  prediction_op = tfe.define_output(
      "prediction-client", [logits, y], prediction_client.receive_output)

After Change

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

  print("Set trained weights")
  model.set_weights(params, sess)

  for _ in range(5):
    print("Predicting")

Instances