Lines Matching refs:auxInput

110     const T* auxInput = nullptr;  in executeTyped()  local
116 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
120 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
163 transposeFirstTwoDims(auxInput, auxInputShape, auxInputTransposed.data()); in executeTyped()
169 auxInput = auxInputTransposed.data(); in executeTyped()
206 bwInput = auxInput; in executeTyped()
207 auxInput = nullptr; in executeTyped()
231 auxInputBatchPtr = auxInput + i * batchSize * auxInputSize; in executeTyped()
250 auxInputBatchPtr = auxInput + i * batchSize * auxInputSize; in executeTyped()
318 Shape auxInput = context->getInputShape(kAuxInputTensor); in prepare() local
361 NN_RET_CHECK_EQ(getNumberOfDimensions(auxInput), 3u); in prepare()
365 NN_RET_CHECK_EQ(getSizeOfDimension(auxInput, 0), getSizeOfDimension(input, 0)); in prepare()
366 NN_RET_CHECK_EQ(getSizeOfDimension(auxInput, 1), getSizeOfDimension(input, 1)); in prepare()
368 NN_RET_CHECK_EQ(getSizeOfDimension(fwAuxWeights, 1), getSizeOfDimension(auxInput, 2)); in prepare()
370 NN_RET_CHECK_EQ(getSizeOfDimension(bwAuxWeights, 1), getSizeOfDimension(auxInput, 2)); in prepare()
372 NN_RET_CHECK_EQ(getNumberOfDimensions(auxInput), 3u); in prepare()
374 NN_RET_CHECK_EQ(getSizeOfDimension(auxInput, 0), getSizeOfDimension(input, 0)); in prepare()
375 NN_RET_CHECK_EQ(getSizeOfDimension(auxInput, 1), getSizeOfDimension(input, 1)); in prepare()
376 NN_RET_CHECK_EQ(getSizeOfDimension(auxInput, 2), getSizeOfDimension(bwWeights, 1)); in prepare()