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

Before Change

  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(

After Change

  with tfe.protocol.SecureNN():
    batch_size = PredictionClient.BATCH_SIZE
    flat_dim = ModelTrainer.IN_N
    batch_input_shape = [batch_size, flat_dim]
    // compute prediction
    model = tfe.keras.Sequential()
    model.add(tfe.keras.layers.Dense(ModelTrainer.HIDDEN_N,
                                     batch_input_shape=batch_input_shape))
    model.add(tfe.keras.layers.Activation("relu"))
    model.add(tfe.keras.layers.Dense(ModelTrainer.HIDDEN_N))
    model.add(tfe.keras.layers.Activation("relu"))
    model.add(tfe.keras.layers.Dense(ModelTrainer.OUT_N))

Instances