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

      tfe_model.set_weights(k_weights)
      y = tfe_model(x)

    with KE.get_session() as sess:
      actual = sess.run(y.reveal())

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

Instances