Lines Matching refs:auxInput
141 const T* auxInput = nullptr; in executeTyped() local
147 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
151 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
194 transposeFirstTwoDims(auxInput, auxInputShape, auxInputTransposed.data()); in executeTyped()
200 auxInput = auxInputTransposed.data(); in executeTyped()
237 bwInput = auxInput; in executeTyped()
238 auxInput = nullptr; in executeTyped()
262 auxInputBatchPtr = auxInput + i * batchSize * auxInputSize; in executeTyped()
281 auxInputBatchPtr = auxInput + i * batchSize * auxInputSize; in executeTyped()
378 Shape auxInput = context->getInputShape(kAuxInputTensor); in prepare() local
421 NN_RET_CHECK_EQ(getNumberOfDimensions(auxInput), 3); in prepare()
425 NN_RET_CHECK_EQ(getSizeOfDimension(auxInput, 0), getSizeOfDimension(input, 0)); in prepare()
426 NN_RET_CHECK_EQ(getSizeOfDimension(auxInput, 1), getSizeOfDimension(input, 1)); in prepare()
428 NN_RET_CHECK_EQ(getSizeOfDimension(fwAuxWeights, 1), getSizeOfDimension(auxInput, 2)); in prepare()
430 NN_RET_CHECK_EQ(getSizeOfDimension(bwAuxWeights, 1), getSizeOfDimension(auxInput, 2)); in prepare()
432 NN_RET_CHECK_EQ(getNumberOfDimensions(auxInput), 3); in prepare()
434 NN_RET_CHECK_EQ(getSizeOfDimension(auxInput, 0), getSizeOfDimension(input, 0)); in prepare()
435 NN_RET_CHECK_EQ(getSizeOfDimension(auxInput, 1), getSizeOfDimension(input, 1)); in prepare()
436 NN_RET_CHECK_EQ(getSizeOfDimension(auxInput, 2), getSizeOfDimension(bwWeights, 1)); in prepare()