search

tf-encrypted / moose

Secure distributed dataflow framework for encrypted machine learning and data processing
Apache License 2.0
58 stars 15 forks source link

GrpcRuntime: only send arguments to annotated placements #1132

Open mortendahl opened 2 years ago

mortendahl commented 2 years ago

In the eDSL we have the placement where an argument is coming from:

def simple_computation(
    x: pm.Argument(placement=alice, vtype=pm.TensorType(pm.float64)),
    y: pm.Argument(placement=bob, vtype=pm.TensorType(pm.float64)),
):

which means we might be able to only send x to Alice, and only y to Bob, instead of currently sending everything to everyone.