(self, inputs)
| 18 | super(MyDense, self).build(input_shape) |
| 19 | |
| 20 | def call(self, inputs): |
| 21 | if self.activation == None: |
| 22 | return sgx.e_matmul(inputs, self.w) |
| 23 | elif self.activation == "relu": |
| 24 | return sgx.e_relu(sgx.e_matmul(inputs, self.w)) |
| 25 | elif self.activation == "sigmoid": |
| 26 | return sgx.e_sigmoid(sgx.e_matmul(inputs, self.w)) |
| 27 | |
| 28 | def compute_output_shape(self, input_shape): |
| 29 | return (input_shape[0], self.units) |
nothing calls this directly
no outgoing calls
no test coverage detected