Ejemplo n.º 1
0
 def test_clients_placed(self):
     x = _mock_data_of_type(
         computation_types.at_clients(
             computation_types.SequenceType(tf.int32)))
     val = intrinsics.sequence_sum(x)
     self.assert_value(val, '{int32}@CLIENTS')
Ejemplo n.º 2
0
 def test_server_placed(self):
     x = _mock_data_of_type(
         computation_types.at_server(
             computation_types.SequenceType(tf.int32)))
     val = intrinsics.sequence_sum(x)
     self.assert_value(val, 'int32@SERVER')