tf-encrypted / moose

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

Fix output expression tracing #1139

Closed jvmncs closed 2 years ago

jvmncs commented 2 years ago

In #1126 there was a bug when tracing explicitly tagged output expressions, which resulted in the OutputOp having HostUnit return type in its signature. Verified the fix with a test here. Also, this change exposes the edsl.output function in the root pymoose namespace.