Lines Matching refs:RunTimeOperandInfo

33 LSHProjection::LSHProjection(const Operation& operation, RunTimeOperandInfo* operands) {  in LSHProjection()
44 bool LSHProjection::Prepare(const Operation& operation, RunTimeOperandInfo* operands,  in Prepare()
54     const RunTimeOperandInfo* hash = GetInput(operation, operands, kHashTensor);  in Prepare()
59     const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor);  in Prepare()
72             RunTimeOperandInfo* weight = GetInput(operation, operands, kWeightTensor);  in Prepare()
96 int runningSignBit(const RunTimeOperandInfo* input, const RunTimeOperandInfo* weight, float seed) {  in runningSignBit()
125 void SparseLshProjection(LSHProjectionType type, const RunTimeOperandInfo* hash,  in SparseLshProjection()
126                          const RunTimeOperandInfo* input, const RunTimeOperandInfo* weight,  in SparseLshProjection()
146 void DenseLshProjection(const RunTimeOperandInfo* hash, const RunTimeOperandInfo* input,  in DenseLshProjection()
147                         const RunTimeOperandInfo* weight, int32_t* out_buf) {  in DenseLshProjection()
182 template int runningSignBit<float>(const RunTimeOperandInfo* input,
183                                    const RunTimeOperandInfo* weight, float seed);
184 template int runningSignBit<_Float16>(const RunTimeOperandInfo* input,
185                                       const RunTimeOperandInfo* weight, float seed);
187 template void SparseLshProjection<float>(LSHProjectionType type, const RunTimeOperandInfo* hash,
188                                          const RunTimeOperandInfo* input,
189                                          const RunTimeOperandInfo* weight, int32_t* outBuffer);
190 template void SparseLshProjection<_Float16>(LSHProjectionType type, const RunTimeOperandInfo* hash,
191                                             const RunTimeOperandInfo* input,
192                                             const RunTimeOperandInfo* weight, int32_t* outBuffer);
194 template void DenseLshProjection<float>(const RunTimeOperandInfo* hash,
195                                         const RunTimeOperandInfo* input,
196                                         const RunTimeOperandInfo* weight, int32_t* outBuffer);
197 template void DenseLshProjection<_Float16>(const RunTimeOperandInfo* hash,
198                                            const RunTimeOperandInfo* input,
199                                            const RunTimeOperandInfo* weight, int32_t* outBuffer);