Lines Matching refs:RunTimeOperandInfo

29                              std::vector<RunTimeOperandInfo>& operands) {  in LSHProjection()
40 bool LSHProjection::Prepare(const Operation& operation, std::vector<RunTimeOperandInfo>& operands, in Prepare()
46 const RunTimeOperandInfo* hash = GetInput(operation, operands, kHashTensor); in Prepare()
51 const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor); in Prepare()
63 RunTimeOperandInfo* weight = GetInput(operation, operands, kWeightTensor); in Prepare()
87 int runningSignBit(const RunTimeOperandInfo* input, const RunTimeOperandInfo* weight, float seed) { in runningSignBit()
116 void SparseLshProjection(LSHProjectionType type, const RunTimeOperandInfo* hash, in SparseLshProjection()
117 const RunTimeOperandInfo* input, const RunTimeOperandInfo* weight, in SparseLshProjection()
137 void DenseLshProjection(const RunTimeOperandInfo* hash, const RunTimeOperandInfo* input, in DenseLshProjection()
138 const RunTimeOperandInfo* weight, int32_t* out_buf) { in DenseLshProjection()
173 template int runningSignBit<float>(const RunTimeOperandInfo* input,
174 const RunTimeOperandInfo* weight, float seed);
175 template int runningSignBit<_Float16>(const RunTimeOperandInfo* input,
176 const RunTimeOperandInfo* weight, float seed);
178 template void SparseLshProjection<float>(LSHProjectionType type, const RunTimeOperandInfo* hash,
179 const RunTimeOperandInfo* input,
180 const RunTimeOperandInfo* weight, int32_t* outBuffer);
181 template void SparseLshProjection<_Float16>(LSHProjectionType type, const RunTimeOperandInfo* hash,
182 const RunTimeOperandInfo* input,
183 const RunTimeOperandInfo* weight, int32_t* outBuffer);
185 template void DenseLshProjection<float>(const RunTimeOperandInfo* hash,
186 const RunTimeOperandInfo* input,
187 const RunTimeOperandInfo* weight, int32_t* outBuffer);
188 template void DenseLshProjection<_Float16>(const RunTimeOperandInfo* hash,
189 const RunTimeOperandInfo* input,
190 const RunTimeOperandInfo* weight, int32_t* outBuffer);