Lines Matching refs:weight
73 RunTimeOperandInfo* weight = GetInput(operation, operands, kWeightTensor); in Prepare() local
75 NN_CHECK_EQ(NumDimensions(weight), 1); in Prepare()
76 NN_CHECK_EQ(SizeOfDimension(weight, 0), SizeOfDimension(input, 0)); in Prepare()
97 int runningSignBit(const RunTimeOperandInfo* input, const RunTimeOperandInfo* weight, float seed) { in runningSignBit() argument
115 if (weight->lifetime == OperandLifeTime::NO_VALUE) { in runningSignBit()
118 score += static_cast<double>(reinterpret_cast<T*>(weight->buffer)[i]) * running_value; in runningSignBit()
127 const RunTimeOperandInfo* input, const RunTimeOperandInfo* weight, in SparseLshProjection() argument
135 int bit = runningSignBit<T>(input, weight, static_cast<float>(seed)); in SparseLshProjection()
148 const RunTimeOperandInfo* weight, int32_t* out_buf) { in DenseLshProjection() argument
154 int bit = runningSignBit<T>(input, weight, static_cast<float>(seed)); in DenseLshProjection()
184 const RunTimeOperandInfo* weight, float seed);
186 const RunTimeOperandInfo* weight, float seed);
190 const RunTimeOperandInfo* weight, int32_t* outBuffer);
193 const RunTimeOperandInfo* weight, int32_t* outBuffer);
197 const RunTimeOperandInfo* weight, int32_t* outBuffer);
200 const RunTimeOperandInfo* weight, int32_t* outBuffer);