Searched refs:auxInput (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceRNN.cpp | 96 const T* auxInput = nullptr; in executeTyped() local
101 auxInput = context->getInputBuffer<T>(kAuxInputTensor); in executeTyped()
143 transposeFirstTwoDims(auxInput, auxInputShape, auxInputTransposed.data()); in executeTyped()
149 auxInput = auxInputTransposed.data(); in executeTyped()
191 auxInputBatchPtr = auxInput + i * batchSize * auxInputSize; in executeTyped()
211 auxInputBatchPtr = auxInput + i * batchSize * auxInputSize; in executeTyped()
301 Shape auxInput = context->getInputShape(kAuxInputTensor); in prepare() local
348 NN_RET_CHECK_EQ(getNumberOfDimensions(auxInput), 3); in prepare()
352 NN_RET_CHECK_EQ(getSizeOfDimension(auxInput, 0), getSizeOfDimension(input, 0)); in prepare()
353 NN_RET_CHECK_EQ(getSizeOfDimension(auxInput, 1), getSizeOfDimension(input, 1)); in prepare()
[all …]
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 1937 ANeuralNetworksOperandType auxInput = floatTensor3D; in lstmBidirectionalSequence() local
2004 auxInput, in lstmBidirectionalSequence()
|