139     const T* auxInput = nullptr;  in executeTyped()  local
145         auxInput = context->getInputBuffer<T>(kAuxInputTensor);  in executeTyped()
149         auxInput = context->getInputBuffer<T>(kAuxInputTensor);  in executeTyped()
192             transposeFirstTwoDims(auxInput, auxInputShape, auxInputTransposed.data());  in executeTyped()
198             auxInput = auxInputTransposed.data();  in executeTyped()
235         bwInput = auxInput;  in executeTyped()
236         auxInput = nullptr;  in executeTyped()
260             auxInputBatchPtr = auxInput + i * batchSize * auxInputSize;  in executeTyped()
279             auxInputBatchPtr = auxInput + i * batchSize * auxInputSize;  in executeTyped()
377     Shape auxInput = context->getInputShape(kAuxInputTensor);  in prepare()  local
420         NN_RET_CHECK_EQ(getNumberOfDimensions(auxInput), 3);  in prepare()
424         NN_RET_CHECK_EQ(getSizeOfDimension(auxInput, 0), getSizeOfDimension(input, 0));  in prepare()
425         NN_RET_CHECK_EQ(getSizeOfDimension(auxInput, 1), getSizeOfDimension(input, 1));  in prepare()
427         NN_RET_CHECK_EQ(getSizeOfDimension(fwAuxWeights, 1), getSizeOfDimension(auxInput, 2));  in prepare()
429         NN_RET_CHECK_EQ(getSizeOfDimension(bwAuxWeights, 1), getSizeOfDimension(auxInput, 2));  in prepare()
431         NN_RET_CHECK_EQ(getNumberOfDimensions(auxInput), 3);  in prepare()
433         NN_RET_CHECK_EQ(getSizeOfDimension(auxInput, 0), getSizeOfDimension(input, 0));  in prepare()
434         NN_RET_CHECK_EQ(getSizeOfDimension(auxInput, 1), getSizeOfDimension(input, 1));  in prepare()
435         NN_RET_CHECK_EQ(getSizeOfDimension(auxInput, 2), getSizeOfDimension(bwWeights, 1));  in prepare()