diff options
Diffstat (limited to 'nn/runtime/test/specs/V1_1/lsh_projection_2_relaxed.mod.py')
-rw-r--r-- | nn/runtime/test/specs/V1_1/lsh_projection_2_relaxed.mod.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/nn/runtime/test/specs/V1_1/lsh_projection_2_relaxed.mod.py b/nn/runtime/test/specs/V1_1/lsh_projection_2_relaxed.mod.py index 6608b0666..1904f0490 100644 --- a/nn/runtime/test/specs/V1_1/lsh_projection_2_relaxed.mod.py +++ b/nn/runtime/test/specs/V1_1/lsh_projection_2_relaxed.mod.py @@ -20,8 +20,9 @@ num_bits = 2 model = Model() -hhash = Parameter("hash", "TENSOR_FLOAT32", "{%d, %d}" % (num_hash, num_bits), - [0.123, 0.456, -0.321, -0.654, 1.234, 5.678, -4.321, -8.765]) +hhash = Parameter( + "hash", "TENSOR_FLOAT32", "{%d, %d}" % (num_hash, num_bits), + [0.123, 0.456, -0.321, -0.654, 1.234, 5.678, -4.321, -8.765]).ShouldNeverBeInternal() lookup = Input("lookup", "TENSOR_INT32", "{%d, %d}" % (num_input, num_bits)) weight = Input("weight", "TENSOR_FLOAT32", "{%d}" % (num_input)) type_param = Int32Scalar("type_param", 1) # SPARSE |