github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-04-17 03:00:54 -04:00
Code Issues Packages Projects Releases Wiki Activity

feat: remove transpose from layers

Browse Source
This commit is contained in:
jfrery
2021-11-17 12:19:40 +01:00
committed by jfrery
parent b9c52947f6
commit c978107124
3 changed files with 13 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/quantization/test_quantized_layers.py
View File

@@ -30,11 +30,11 @@ def test_quantized_linear(n_examples, n_features, n_neurons, n_bits):
inputs = numpy.random.uniform(size=(n_examples, n_features))
q_inputs = QuantizedArray(n_bits, inputs)
# shape of weights: (n_examples, n_features, n_neurons)
weights = numpy.random.uniform(size=(n_neurons, n_features))
# shape of weights: (n_neurons, n_features)
weights = numpy.random.uniform(size=(n_features, n_neurons))
q_weights = QuantizedArray(n_bits, weights)
bias = numpy.random.uniform(size=(n_neurons))
bias = numpy.random.uniform(size=(1, n_neurons))
q_bias = QuantizedArray(n_bits, bias)
# Define our QuantizedLinear layer