c56e3db656f3db9a7316b41221571a82ff398a1a,tf_encrypted/keras/models/sequential_test.py,TestSequential,test_conv_model,#TestSequential#,110

Before Change

    with tfe.protocol.SecureNN():
      tfe_model = tfe.keras.models.model_from_config(k_config)
      x = tfe.define_private_variable(input_data)

    with tfe.Session() as sess:
      sess.run(tf.global_variables_initializer())
      tfe_model.set_weights(k_weights)
      y = tfe_model(x)
      actual = sess.run(y.reveal())

After Change

      np.testing.assert_allclose(actual, expected, rtol=1e-2, atol=1e-2)

    KE.clear_session()

def _model_predict_keras(input_data, input_shape):
  with tf.Session():
    model = tf.keras.models.Sequential()
    model.add(tf.keras.layers.Dense(2, batch_input_shape=input_shape))

Instances