github/MP-SPDZ
1
1
Fork 0
mirror of https://github.com/data61/MP-SPDZ.git synced 2026-01-10 14:08:09 -05:00
Code Issues Packages Projects Releases Wiki Activity

Missing code for square activation.

Browse Source
This commit is contained in:
Marcel Keller
2021-07-09 20:11:45 +10:00
parent 051b7fd1dc
commit a94c2d7613
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
Compiler/ml.py
View File

@@ -910,6 +910,12 @@ class ElementWiseLayer(NoVariableLayer):
self.nabla_Y = Tensor(backward_shape, sfix)
self.inputs = inputs
def f_part(self, base, size):
return self.f(self.X.get_part_vector(base, size))
def f_prime_part(self, base, size):
return self.f_prime(self.Y.get_part_vector(base, size))
def _forward(self, batch=[0]):
n_per_item = reduce(operator.mul, self.X.sizes[1:])
@multithread(self.n_threads, len(batch), max(1, 1000 // n_per_item))