Lines Matching refs:hash
55 const RunTimeOperandInfo* hash = GetInput(operation, operands, kHashTensor); in Prepare() local
56 NN_CHECK_EQ(NumDimensions(hash), 2); in Prepare()
58 NN_CHECK(SizeOfDimension(hash, 1) <= 32); in Prepare()
70 outputShape->dimensions = {SizeOfDimension(hash, 0)}; in Prepare()
77 outputShape->dimensions = {SizeOfDimension(hash, 0) * SizeOfDimension(hash, 1)}; in Prepare()
126 void SparseLshProjection(LSHProjectionType type, const RunTimeOperandInfo* hash, in SparseLshProjection() argument
129 int num_hash = SizeOfDimension(hash, 0); in SparseLshProjection()
130 int num_bits = SizeOfDimension(hash, 1); in SparseLshProjection()
134 T seed = reinterpret_cast<T*>(hash->buffer)[i * num_bits + j]; in SparseLshProjection()
147 void DenseLshProjection(const RunTimeOperandInfo* hash, const RunTimeOperandInfo* input, in DenseLshProjection() argument
149 int num_hash = SizeOfDimension(hash, 0); in DenseLshProjection()
150 int num_bits = SizeOfDimension(hash, 1); in DenseLshProjection()
153 T seed = reinterpret_cast<T*>(hash->buffer)[i * num_bits + j]; in DenseLshProjection()
188 template void SparseLshProjection<float>(LSHProjectionType type, const RunTimeOperandInfo* hash,
191 template void SparseLshProjection<_Float16>(LSHProjectionType type, const RunTimeOperandInfo* hash,
195 template void DenseLshProjection<float>(const RunTimeOperandInfo* hash,
198 template void DenseLshProjection<_Float16>(const RunTimeOperandInfo* hash,