Lines Matching refs:input
60 const RunTimeOperandInfo* input = GetInput(operation, operands, kInputTensor); in Prepare() local
61 NN_CHECK(NumDimensions(input) >= 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
99 int input_item_bytes = nonExtensionOperandSizeOfData(input->type, input->dimensions) / in runningSignBit()
100 SizeOfDimension(input, 0); in runningSignBit()
101 char* input_ptr = (char*)(input->buffer); in runningSignBit()
107 for (uint32_t i = 0; i < SizeOfDimension(input, 0); ++i) { 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()
147 void DenseLshProjection(const RunTimeOperandInfo* hash, const RunTimeOperandInfo* input, in DenseLshProjection() argument
154 int bit = runningSignBit<T>(input, weight, static_cast<float>(seed)); in DenseLshProjection()
183 template int runningSignBit<float>(const RunTimeOperandInfo* input,
185 template int runningSignBit<_Float16>(const RunTimeOperandInfo* input,
189 const RunTimeOperandInfo* input,
192 const RunTimeOperandInfo* input,
196 const RunTimeOperandInfo* input,
199 const RunTimeOperandInfo* input,