Lines Matching refs:input
59 const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor); in Prepare() local
60 NN_CHECK(NumDimensions(input) >= 1); in Prepare()
75 NN_CHECK_EQ(SizeOfDimension(weight, 0), SizeOfDimension(input, 0)); in Prepare()
96 int runningSignBit(const RunTimeOperandInfo* input, const RunTimeOperandInfo* weight, float seed) { in runningSignBit() argument
98 int input_item_bytes = nonExtensionOperandSizeOfData(input->type, input->dimensions) / in runningSignBit()
99 SizeOfDimension(input, 0); in runningSignBit()
100 char* input_ptr = (char*)(input->buffer); in runningSignBit()
106 for (uint32_t i = 0; i < SizeOfDimension(input, 0); ++i) { in runningSignBit()
126 const RunTimeOperandInfo* input, const RunTimeOperandInfo* weight, in SparseLshProjection() argument
134 int bit = runningSignBit<T>(input, weight, static_cast<float>(seed)); in SparseLshProjection()
146 void DenseLshProjection(const RunTimeOperandInfo* hash, const RunTimeOperandInfo* input, in DenseLshProjection() argument
153 int bit = runningSignBit<T>(input, weight, static_cast<float>(seed)); in DenseLshProjection()
182 template int runningSignBit<float>(const RunTimeOperandInfo* input,
184 template int runningSignBit<_Float16>(const RunTimeOperandInfo* input,
188 const RunTimeOperandInfo* input,
191 const RunTimeOperandInfo* input,
195 const RunTimeOperandInfo* input,
198 const RunTimeOperandInfo* input,