Lines Matching refs:weight
39 // weight of 1.0 Tensor[1].Dim[0] == Tensor[2].Dim[0]
89 TfLiteTensor* weight = GetInput(context, node, 2);
90 TF_LITE_ENSURE_EQ(context, NumDimensions(weight), 1);
91 TF_LITE_ENSURE_EQ(context, SizeOfDimension(weight, 0),
110 // Compute sign bit of dot product of hash(seed, input) and weight.
115 int RunningSignBit(const TfLiteTensor* input, const TfLiteTensor* weight,
133 if (weight == nullptr) {
136 score += weight->data.f[i] * running_value;
144 const TfLiteTensor* weight, int32_t* out_buf) {
151 int bit = RunningSignBit(input, weight, seed);
159 const TfLiteTensor* weight, int32_t* out_buf) {
165 int bit = RunningSignBit(input, weight, seed);
178 TfLiteTensor* weight =
183 DenseLshProjection(hash, input, weight, out_buf);
186 SparseLshProjection(hash, input, weight, out_buf);