search

tf-encrypted / moose

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

Improvements for Squeeze Op #1074

Closed rdragos closed 2 years ago

rdragos commented 2 years ago

Closes https://github.com/tf-encrypted/moose/issues/1073

Adds support for squeeze operation for

HostPlacement, * where * means all possible types at this momement.

and

ReplicatedPlacement, Fixed64
ReplicatedPlacement, Fixed128,
ReplicatedPlacement, Boolean
ReplicatedPlacement, Uint64Tensor