import latticex.rosetta as rtt import tensorflow as tf import sys, os import numpy as np np.set_printoptions(suppress=True) tf.set_random_seed(0) protocol = "SecureNN" if "ROSETTA_TEST_PROTOCOL" in os.environ.keys(): print("***** test_cases use ", os.environ["ROSETTA_TEST_PROTOCOL"]) protocol = os.environ["ROSETTA_TEST_PROTOCOL"] else: print("***** test_cases use default helix protocol ") rtt.activate(protocol) print("rtt.get_protocol_name():", rtt.get_protocol_name()) patyid = rtt.get_party_id("") ############################### # 1-d ############################### print('=================1-d=========================') x = [1, 8, 3] print(x) a = tf.Variable(x, dtype=tf.float64) aa = tf.reduce_mean(a) a0 = tf.reduce_mean(a, axis=0) aa_k = tf.reduce_mean(a, keepdims=True) a0_k = tf.reduce_mean(a, axis=0, keepdims=True)
rtt.set_backend_loglevel(2) protocol = "SecureNN" if "ROSETTA_TEST_PROTOCOL" in os.environ.keys(): print("***** test_cases use ", os.environ["ROSETTA_TEST_PROTOCOL"]) protocol = os.environ["ROSETTA_TEST_PROTOCOL"] else: print("***** test_cases use default helix protocol ") rtt.activate(protocol) patyid = rtt.get_party_id() print( "rtt.get_protocol_name: ", rtt.get_protocol_name(), "party: ", ) # float * tf.Variable() num_a = np.ones([10, 10]) num_b = np.ones([10, 10]) * 2 num_neg_a = -num_a # only ALICE could own private data X = tf.Variable(rtt.private_input(0, num_a)) Y = tf.Variable(rtt.private_input(0, num_b)) P = tf.Variable(rtt.private_input(0, num_neg_a)) CX = tf.constant(num_a) CY = tf.constant(num_b)